shunting314andGitHub 8b141ed8c3 full cudagraph for flex-attn (#36298) ...

Signed-off-by: shunting314 <shunting@meta.com>

2026-04-02 21:15:01 -07:00

..

__init__.py

…

test_basic_correctness.py

replace cuda_device_count_stateless() to current_platform.device_count() (#37841)

2026-03-31 22:32:54 +08:00

test_full_cudagraph.py

full cudagraph for flex-attn (#36298)

2026-04-02 21:15:01 -07:00

test_full_graph.py

…

test_multimodal_compile.py

…

test_multiple_graphs.py

…

test_simple.py

[torch.compile] Stop doing unnecessary FakeTensorProp in PiecewiseCompileInterpreter (#34093)

2026-02-10 19:15:40 -08:00

test_toy_llama.py

…