Charlie FuandGitHub a44b1c951d [Feature][ROCm] Add full graph capture support for TritonAttentionBackend (#19158) ...

Signed-off-by: charlifu <charlifu@amd.com>

2025-06-17 17:03:06 -04:00

..

__init__.py

[torch.compile] rework compile control with piecewise cudagraph (#9715)

2024-10-29 23:03:49 -07:00

test_full_cudagraph.py

[Feature][ROCm] Add full graph capture support for TritonAttentionBackend (#19158)

2025-06-17 17:03:06 -04:00

test_simple.py

[CUDA] Enable full cudagraph for FlashMLA (#18581)

2025-06-13 18:12:26 +00:00

test_toy_llama.py

[CUDA] Enable full cudagraph for FlashMLA (#18581)

2025-06-13 18:12:26 +00:00