Lucas Wilkinson 978b45f399

[Kernel] Flash Attention 3 Support (#12093) ...

Signed-off-by: Lucas Wilkinson <lwilkinson@neuralmagic.com>

2025-01-23 06:45:48 -08:00

..

backends

[Kernel] Flash Attention 3 Support (#12093)

2025-01-23 06:45:48 -08:00

ops

[Bugfix] Fix chunked prefill with model dtype float32 on Turing Devices (#9850)

2024-11-25 12:23:32 -05:00

__init__.py

[Core] Add AttentionState abstraction (#7663)

2024-08-20 18:50:45 +00:00

layer.py

[Misc] Pass attention to impl backend (#12218)

2025-01-20 23:25:28 +08:00

selector.py

[Platform] Do not raise error if _Backend is not found (#12023)

2025-01-15 10:14:15 +00:00