Vadim GimpelsonandClaude 783380af39 [Bugfix] Fix NaN output of the Mamba2 chunk scan kernel on sm90 (#61718) ...

Signed-off-by: Vadim Gimpelson <vadim.gimpelson@gmail.com>
Co-authored-by: Claude <noreply@anthropic.com>

2026-10-16 17:24:51 -07:00

..

configs/selective_state_update

[XPU] Add tuned Mamba SSU configs for Intel Arc Pro B60 (#56765)

2026-10-09 14:29:29 +08:00

cpu

[CPU] Conv1d optimised kernel for aarch64 (#54093)

2026-09-30 20:23:42 +08:00

gdn_chunk_cutedsl

[Bugfix][GDN] Fix CuteDSL BF16 KKT inversion divergence (#53864)

2026-09-17 08:49:56 -07:00

__init__.py

[Kernel/Model] Migrate mamba_ssm and causal_conv1d kernels to vLLM (#7651)

2024-08-28 15:06:52 -07:00

causal_conv1d.py

[Core][Mamba] Remove leftovers of mamba_cache_mode="all" (#60043)

2026-10-13 00:58:59 -07:00

gather_initial_states.py

[XPU] Enable Kimi K3 KDA kernel tests on XPU (#51809)

2026-08-17 17:21:34 +08:00

layernorm_gated.py

replace with torch.cuda.device with with torch.accelerator.device_index (#36144)

2026-03-11 23:12:57 -07:00

mamba_ssm.py

[CI] Ignore ruff D209, rejoin the docstrings it split, and silence incompatible-rule warnings (#57212)

2026-09-16 10:57:24 -07:00

replayssm_config.py

[Refactor] Remove dead kernel code (#56845)

2026-09-15 13:17:53 -07:00

scatter_states.py

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

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

selective_state_update_replayssm_output_only.py

[ROCm] Use backend-default dot precision for ReplaySSM (#49909)

2026-07-27 18:11:06 +08:00

ssd_bmm.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

ssd_checkpoint.py

[Bugfix][Mamba] Store conv_kernel - 1 inputs in Mamba2 prefill checkpoints (#60797)

2026-10-13 10:09:17 +00:00

ssd_chunk_scan.py

[Bugfix] Fix NaN output of the Mamba2 chunk scan kernel on sm90 (#61718)

2026-10-16 17:24:51 -07:00

ssd_chunk_state.py

[Perf] Add do_not_specialize to Mamba SSD chunk kernels (#43803)

2026-05-28 15:40:02 +00:00

ssd_combined.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

ssd_state_passing.py

replace with torch.cuda.device with with torch.accelerator.device_index (#36144)

2026-03-11 23:12:57 -07:00

ssu_dispatch.py

[Mamba] Add FlashInfer ReplaySSM support for MTP (#52928)

2026-09-30 21:41:17 +00:00

triton_helpers.py

[Kernel][Mamba] Optimize Mamba2 SSD prefill Triton kernels (#35397)

2026-03-04 19:47:17 +01:00