tomeras91 7faba503c4 [Kernel][Mamba] Optimize Mamba2 SSD prefill Triton kernels (#35397) ...

Signed-off-by: Tomer Asida <57313761+tomeras91@users.noreply.github.com>

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

[Model][Spec Decode] Nemotron-H MTP and Mamba Speculative Decoding Support (#33726)

2026-02-24 09:49:56 -08:00

layernorm_gated.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

mamba_ssm.py

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

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

ssd_bmm.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

ssd_chunk_scan.py

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

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

ssd_chunk_state.py

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

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

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

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

triton_helpers.py

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

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