3 people aed894c190 [6/N][warmup][DSv4] Migrate sampling, and DFlash JIT kernels (#56323) ...

Signed-off-by: Lucas Wilkinson <lwilkins@redhat.com>
Signed-off-by: LopezCastroRoberto <rocastro@redhat.com>
Co-authored-by: Lucas Wilkinson <lwilkins@redhat.com>
Co-authored-by: OpenAI Codex <codex@openai.com>

2026-09-12 17:24:24 -07:00

..

gdn

[Perf][GDN] Enable the FlashInfer GDN prefill kernel on SM12x (#55715)

2026-09-08 12:26:03 +00:00

linear

…

ops

[6/N][warmup][DSv4] Migrate sampling, and DFlash JIT kernels (#56323)

2026-09-12 17:24:24 -07:00

__init__.py

…

abstract.py

…

mamba_mixer2.py

[XPU] Add forward_xpu to Mixer2RMSNormGated and FusedRMSNormGated (#54968)

2026-09-11 07:58:28 +08:00

mamba_mixer.py

…

mamba_utils.py

[Perf] Integrate FlashInfer KDA kernels (#55364)

2026-09-07 20:29:30 -07:00

short_conv.py

…