Matthew Bonanni 620db1fc58

[Attention] FlashAttention MLA cudagraph support (#23958) ...

Signed-off-by: Matthew Bonanni <mbonanni001@gmail.com>
Co-authored-by: Robert Shaw <114415538+robertgshaw2-redhat@users.noreply.github.com>

2025-09-08 22:05:26 +00:00

..

__init__.py

[Core] Allow full cudagraph with separate attention routines and orthogonal to compilation, add support for FA2 and FlashInfer (#20059)

2025-08-15 10:01:39 -04:00

test_cudagraph_dispatch.py

[Core] Allow full cudagraph with separate attention routines and orthogonal to compilation, add support for FA2 and FlashInfer (#20059)

2025-08-15 10:01:39 -04:00

test_cudagraph_mode.py

[Attention] FlashAttention MLA cudagraph support (#23958)

2025-09-08 22:05:26 +00:00