Rahul SteigerandRahul Steiger 4934741df9 [Bugfix] Add TRTLLM attention custom op and execution contract (#7214) ...

Signed-off-by: Rahul Steiger <rsteiger@aws-cmh-slurm-1-vscode-04.cm.cluster>
Signed-off-by: Rahul Steiger <rsteiger@nvidia.com>
Co-authored-by: Rahul Steiger <rsteiger@aws-cmh-slurm-1-vscode-04.cm.cluster>

2026-10-04 09:33:51 +00:00

..

ring

[Model][MiniMax-H3] Multi-GPU serving fixes and layer-wise offload component selection (#7047)

2026-09-14 10:02:30 -07:00

utils

[Core] Prototype diffusion attention execution contracts with FA4 (#7379)

2026-10-02 17:58:47 +08:00

__init__.py

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

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

abstract.py

[Core] Prototype diffusion attention execution contracts with FA4 (#7379)

2026-10-02 17:58:47 +08:00

cudnn_attn.py

[CI] Select mirror_hardwares presets via MIRROR_HW (H100/B200) (#5543)

2026-09-02 15:39:49 +00:00

fastvideo_vsa.py

[Attention] [P1] Keep H3 VSA policy in the model and share sparse primitives (#7535)

2026-09-16 15:47:31 +08:00

flash_attn_hub.py

[Bugfix] Resolve #6931 hub FA3 on torch 2.13 via kernels 0.16.1 (#7185)

2026-09-09 10:20:34 +00:00

flash_attn.py

[Core] Prototype diffusion attention execution contracts with FA4 (#7379)

2026-10-02 17:58:47 +08:00

flashinfer_attn.py

[Kernel] Improve 2 backends importing FlashInfer: FLASHINFER_ATTN and TRTLLM_ATTN (#7431)

2026-09-28 21:21:58 +00:00

rainfusion_attn.py

[Bugfix][RAINFUSION_ATTN] Fix padding-mask rejection on packed [real, pad] layouts and reuse query tail as output padding (#7235)

2026-09-20 10:56:51 +08:00

registry.py

[FEAT]Add FastVideo VSA backend for Wan2.2 (#4820)

2026-08-24 14:48:31 +08:00

ring_flash_attn.py

[Model][MiniMax-H3] Multi-GPU serving fixes and layer-wise offload component selection (#7047)

2026-09-14 10:02:30 -07:00

ring_pytorch_attn.py

[Model][MiniMax-H3] Multi-GPU serving fixes and layer-wise offload component selection (#7047)

2026-09-14 10:02:30 -07:00

sage_attn3.py

[CI] Select mirror_hardwares presets via MIRROR_HW (H100/B200) (#5543)

2026-09-02 15:39:49 +00:00

sage_attn.py

[CI] Select mirror_hardwares presets via MIRROR_HW (H100/B200) (#5543)

2026-09-02 15:39:49 +00:00

sdpa.py

[CI] Select mirror_hardwares presets via MIRROR_HW (H100/B200) (#5543)

2026-09-02 15:39:49 +00:00

trtllm_attn.py

[Bugfix] Add TRTLLM attention custom op and execution contract (#7214)

2026-10-04 09:33:51 +00:00

trtllm_calibration.py

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

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