3 people 9af53a3c13 [Perf] Add tuned selective_state_update configs for H200 and RTX PRO … (#44251) ...

Signed-off-by: Majid Taheri Andani <tahemaji@amazon.com>
Co-authored-by: Majid Taheri Andani <tahemaji@amazon.com>
Co-authored-by: tomeras91 <57313761+tomeras91@users.noreply.github.com>

2026-06-02 23:59:01 -07:00

..

configs/selective_state_update

[Perf] Add tuned selective_state_update configs for H200 and RTX PRO … (#44251)

2026-06-02 23:59:01 -07:00

cpu

[CPU][Perf] Enable fused kernels for GDN's gated delta rules (#43534)

2026-06-02 08:00:48 +00:00

gdn_chunk_cutedsl

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

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

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

2026-05-28 15:40:02 +00: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

[XPU][Mamba] Triton-based selective scan forward op for XPU (#43421)

2026-06-02 03:50:26 -07:00

ssd_bmm.py

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

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

ssd_chunk_scan.py

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

2026-05-28 15:40:02 +00: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

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

2026-03-04 19:47:17 +01: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

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

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

triton_helpers.py

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

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