3 people fbe8a157fb [ROCm][Perf] Enable the AITER GDN decode fast path for flat qkvz layouts (#53623) ...

Signed-off-by: Mikko Tukiainen <Mikko.Tukiainen@amd.com>
Co-authored-by: Cursor <cursoragent@cursor.com>
Co-authored-by: TJian <tunjian.tan@embeddedllm.com>

2026-09-19 06:05:04 -07:00

..

cpu

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

2026-09-16 10:57:24 -07: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

Add pydocstyle to the ruff rules (#52136)

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

test_cpu_short_conv.py

[CPU][Bugfix] Fix flaky ShortConv prefill test on ARM (uninitialized weights) (#47848)

2026-07-07 18:20:09 -07:00

test_gdn_forward_core_split.py

[MyPy][1/N] Fix mypy errors in some tests/ directories and enforce follow-imports=silent (#49570)

2026-07-30 05:02:51 -07:00

test_gdn_fused_mtp.py

[Model] Support Qwen3.8-Flash-Next (#53896)

2026-08-31 05:57:56 +00: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

[Model] add GLM-5.3-Flash support (#53906)

2026-09-03 09:40:35 -07:00

test_gdn_rocm_layout_dispatch.py

[ROCm][Perf] Enable the AITER GDN decode fast path for flat qkvz layouts (#53623)

2026-09-19 06:05:04 -07:00

test_mamba_mixer2.py

[Hardware] Replace torch.cuda.device_count/current_device/set_device API (#36145)

2026-03-12 07:57:47 -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

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00: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

[Model] Support Qwen3.8-Flash-Next (#53896)

2026-08-31 05:57:56 +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

[Mamba] Add FlashInfer ReplaySSM backend (#52506)

2026-09-01 10:26:43 -04:00

utils.py

Add pydocstyle to the ruff rules (#52136)

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