3 people 102ba716bf [Feat] Add native Mooncake KV transfer from AR to DiT (#7166) ...

Signed-off-by: asukaqaq-s <1311722138@qq.com>
Signed-off-by: Acerak01-fy <wfy2003324@163.com>
Co-authored-by: Acerak01-fy <wfy2003324@163.com>
Co-authored-by: amy-why-3459 <wuhaiyan17@huawei.com>

2026-09-20 14:44:55 +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

[Model] Add native MAGI-2 Preview diffusion support (#5918)

2026-09-05 15:32:30 +08:00

__init__.py

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

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

abstract.py

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

2026-09-14 10:02:30 -07: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

[Feat] Add native Mooncake KV transfer from AR to DiT (#7166)

2026-09-20 14:44:55 +00:00

flashinfer_attn.py

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

2026-09-02 15:39:49 +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

[perf] Add mask-free TRTLLM packed-padding path (#6542)

2026-08-25 07:45:21 +00:00

trtllm_calibration.py

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

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