XU MingshiandHongsheng Liu 5b8ade8446 [Diffusion][Bugfix] Fix the flash_attn backends selection logic (#983) ...

Signed-off-by: mxuax <mxuax@connect.ust.hk>
Signed-off-by: XU Mingshi <91017482+mxuax@users.noreply.github.com>
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>

2026-01-29 16:44:58 +08:00

..

ring

[Diffusion][Bugfix] Fix the flash_attn backends selection logic (#983)

2026-01-29 16:44:58 +08:00

utils

[Diffusion][Bugfix] Fix the flash_attn backends selection logic (#983)

2026-01-29 16:44:58 +08:00

__init__.py

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

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

abstract.py

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

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

flash_attn.py

[Diffusion][Bugfix] Fix the flash_attn backends selection logic (#983)

2026-01-29 16:44:58 +08: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

[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

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

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

sdpa.py

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

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