3 people 373d3c7476 [Blackwell] Add CUDNN_ATTN and FLASHINFER_ATTN backends for diffusion (auto-route) (#3079) ...

Signed-off-by: lishunyang <lishunyang12@163.com>
Signed-off-by: Zhang <jianmusings@gmail.com>
Signed-off-by: Zhang Jian <jianmusings@gmail.com>
Co-authored-by: frauttauteffasu <247219357+frauttauteffasu@users.noreply.github.com>
Co-authored-by: Zhang Jian <jianmusings@gmail.com>

2026-05-28 11:09:25 +08:00

..

__init__.py

[Diffusion] Qwen image support (#82)

2025-11-28 14:22:11 +08:00

adalayernorm.py

[PERF] Wan2.2 support rmsnorm fused op (#2583)

2026-04-16 03:21:34 -07:00

custom_op.py

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

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

fourier.py

[Diffusion] [Model] Support AudioX (#2077)

2026-05-04 10:24:35 -04:00

norm.py

[Fix] Fix RMSNorm inductor KeyError under HSDP + torch.compile (#3460)

2026-05-11 22:56:54 +08:00

rope.py

[Blackwell] Add CUDNN_ATTN and FLASHINFER_ATTN backends for diffusion (auto-route) (#3079)

2026-05-28 11:09:25 +08:00