TJian 46e96e18e3 [ROCm] [Perf] Add AITER Flash Attention (#941) ...

Signed-off-by: tjtanaa <tunjian.tan@embeddedllm.com>

2026-01-27 20:14:49 +08:00

..

ring

[Diffusion]: Diffusion Ring Attention support (#273)

2025-12-31 14:26:30 +00:00

utils

[ROCm] [Perf] Add AITER Flash Attention (#941)

2026-01-27 20:14:49 +08:00

__init__.py

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

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

abstract.py

[Bugfix] Solve Ulysses-SP sequence length not divisible by SP degree (using padding and attention mask) (#672)

2026-01-08 00:02:22 +08:00

ascend_attn.py

[Model] Support Qwen-Image series models on NPU (#463)

2025-12-27 10:46:10 +08:00

flash_attn.py

[ROCm] [Perf] Add AITER Flash Attention (#941)

2026-01-27 20:14:49 +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

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

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

sage_attn.py

[Diffusion][Attention] sage attention backend (#243)

2025-12-13 08:17:41 +08:00

sdpa.py

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

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