SYLARandlishunyang12 7762d664d8 [Attention] Add trtllm diffusion attention backend with Skip-Softmax (#5283) ...

Signed-off-by: lishunyang12 <lishunyang03@gmail.com>
Co-authored-by: lishunyang12 <lishunyang03@gmail.com>

2026-07-27 22:48:11 +08:00

..

ring

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

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

utils

[Feature] Add AllGather-KV sequence-parallel attention str… (#4968)

2026-07-24 14:45:26 +00:00

__init__.py

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

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

abstract.py

[Feature] Add AllGather-KV sequence-parallel attention str… (#4968)

2026-07-24 14:45:26 +00:00

cudnn_attn.py

[Feature] Add AllGather-KV sequence-parallel attention str… (#4968)

2026-07-24 14:45:26 +00:00

flash_attn_hub.py

[Feature] Add AllGather-KV sequence-parallel attention str… (#4968)

2026-07-24 14:45:26 +00:00

flash_attn.py

[Feature] Add AllGather-KV sequence-parallel attention str… (#4968)

2026-07-24 14:45:26 +00:00

flashinfer_attn.py

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

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

registry.py

[Attention] Add trtllm diffusion attention backend with Skip-Softmax (#5283)

2026-07-27 22:48:11 +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_attn3.py

[Blackwell][1/N] Add SageAttention3 diffusion backend on blackwell(GB200/B200/RTX5090/PRO6000/DGX Spark available) (#3015)

2026-05-28 14:08:45 +00:00

sage_attn.py

[XPU] Add sage_attn backend (#3785)

2026-06-09 19:28:18 +08:00

sdpa.py

[Feature] Add AllGather-KV sequence-parallel attention str… (#4968)

2026-07-24 14:45:26 +00:00

trtllm_attn.py

[Attention] Add trtllm diffusion attention backend with Skip-Softmax (#5283)

2026-07-27 22:48:11 +08:00

trtllm_calibration.py

[Attention] Add trtllm diffusion attention backend with Skip-Softmax (#5283)

2026-07-27 22:48:11 +08:00