roikoren755 b897f00c9c Gate SSU dispatch setup (#40039) ...

Signed-off-by: Roi Koren <roik@nvidia.com>

2026-04-16 13:06:01 -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

[Kernel] Mamba support different layout for Conv state (#37416)

2026-04-03 01:50:09 +02: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

[Mamba] Flashinfer selective_state_update (#36162)

2026-04-14 15:10:58 -04:00

ssd_bmm.py

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

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

ssd_chunk_scan.py

[Bugfix] clamp dA_cumsum differences to prevent Inf in Mamba2 SSD kernels (#37501)

2026-03-31 17:35:51 +02:00

ssd_chunk_state.py

[Bugfix] clamp dA_cumsum differences to prevent Inf in Mamba2 SSD kernels (#37501)

2026-03-31 17:35:51 +02: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

Gate SSU dispatch setup (#40039)

2026-04-16 13:06:01 -07:00

triton_helpers.py

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

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