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

..

cpu

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

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

__init__.py

Tuning script and configs for Triton Mamba SSU kernel (#43083)

2026-05-24 20:12:44 +03:00

test_causal_conv1d.py

[Attention][Spec Decode] Enable varlen Mamba2 decode for adaptive verification (#53036)

2026-10-16 11:23:59 -07:00

test_cpu_short_conv.py

[Core] Remove deprecated mamba_cache_mode "all" (#58997)

2026-10-01 02:59:37 -07:00

test_gdn_forward_core_split.py

Disable mypy arg-type and assignment checks in tests (#59428)

2026-09-30 15:18:50 +00:00

test_gdn_fused_mtp.py

[Perf] Keep non-speculative GDN decode on the standard path (#59735)

2026-10-02 13:44:53 +00:00

test_gdn_output_projection.py

[ROCm][Perf] Avoid extra reshape kernel in Qwen GDN output norm (#47842)

2026-09-22 14:38:16 +08:00

test_gdn_prefill_cutedsl.py

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

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

test_gdn_prefill_flashinfer.py

[Perf][GDN] Internal prefill checkpoints for the FlashInfer GDN prefill backend (#60050)

2026-10-14 14:49:46 +08:00

test_gdn_rocm_layout_dispatch.py

[ROCm][Perf] Make AITER FlyDSL GDN prefill default on gfx942/gfx950 (#60645)

2026-10-14 10:11:57 +00:00

test_mamba2_checkpoint_state.py

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

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

test_mamba_mixer2.py

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

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

test_mamba_ssm_configs.py

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

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

test_mamba_ssm_ssd.py

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

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

test_mamba_ssm.py

Add pydocstyle to the ruff rules (#52136)

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

test_memcpy_u64_tiled.py

[Perf][Hybrid] 3D-grid tiling of the state-copy Triton kernels (#49436)

2026-08-10 17:02:49 +02:00

test_precopy_mamba_align.py

[Bugfix] Index Mamba align state copies by request slot in MRV2 (#60210)

2026-10-07 02:32:46 +00:00

test_replayssm_prefill_decode_equivalence_mamba2.py

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

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

test_replayssm_standard_decode_mamba2.py

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

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

test_ssu_dispatch.py

[Bugfix] Fix ReplaySSM ring shape assertion when TP does not divide Mamba2 n_groups (#60704)

2026-10-13 11:31:53 +00:00

utils.py

Add pydocstyle to the ruff rules (#52136)

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