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

Signed-off-by: Rahul Vishwakarma <Rahul.Vishwakarma2@ibm.com>

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

..

cpu

[CPU] Enable chunked prefill and prefix caching for qwen3.5 (#46202)

2026-06-25 03:49:21 +00: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

[Kernel][Test] Make kernel tests for mamba dual-HW (CUDA + XPU) (#42736)

2026-06-08 08:22:47 +08: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

[PERF] [Qwen3.5] Split mixed prefill+decode batches: route decodes to the recurrent kernel (#44700)

2026-06-06 22:13:50 +08:00

test_gdn_prefill_cutedsl.py

[GDN] GDN Prefill kernel for SM100 (#43273)

2026-05-26 14:02:11 +08: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

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

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

test_mamba_ssm_ssd.py

[Kernel][Test] Make kernel tests for mamba dual-HW (CUDA + XPU) (#42736)

2026-06-08 08:22:47 +08:00

test_mamba_ssm.py

[Kernel][XPU] Adjust kernel unit tests for XPU (#45140)

2026-06-30 09:57:27 +00:00

test_precopy_mamba_align.py

[Model Runner V2] support mamba hybrid models align prefix cache (#42406)

2026-06-29 14:09:16 -07:00

test_ssu_dispatch.py

[Misc] Replace mamba_type string literals with MambaAttentionBackendEnum (#41188)

2026-05-11 03:59:36 +00:00

utils.py

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

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