R0CKSTARandCanlin Guo 52f17bf292 [Refactor] Extract has_flash_attn_pkg in OmniPlatform (#3068) ...

Signed-off-by: Xiaodong Ye <yeahdongcn@gmail.com>
Co-authored-by: Canlin Guo <canlinguosdu@gmail.com>

2026-04-27 04:07:34 +00:00

..

ring

[Cleanup] Replace bare print() with logger and use specific exception types (#2228)

2026-04-03 23:59:07 +08:00

utils

[Refactor] Extract has_flash_attn_pkg in OmniPlatform (#3068)

2026-04-27 04:07:34 +00:00

__init__.py

[Diffusion] Init Attention Backends and Selector for Diffusion (#115)

2025-12-05 10:46:20 +00:00

abstract.py

[Feat] Add MUSA platform support for Moore Threads GPUs (#2337)

2026-04-01 17:55:00 +08:00

flash_attn.py

[MUSA][Feat] Upgrade MATE to match CUDA/ROCm behavior on FA (#2766)

2026-04-23 03:07:02 +00:00

registry.py

[Hardware] Support platforms and plugin system (#774)

2026-01-28 22:12:52 +08:00

ring_flash_attn.py

[Bugfix] Raise ValueError when joint_strategy='rear' and causal=True in Ring Attention (#767)

2026-01-13 17:13:31 +08:00

ring_pytorch_attn.py

[CI] Refactor test_sequence_parallel.py and add a warmup run for more accurate performance stat (#1165)

2026-02-03 12:23:15 +08:00

sage_attn.py

[Hardware] Support platforms and plugin system (#774)

2026-01-28 22:12:52 +08:00

sdpa.py

[Bugfix] Fix SDPA attention mask dtype and shape (Fix #857) (#1349)

2026-02-26 11:54:28 +08:00