Thomas Parnell fea3e476aa [Kernel] Chunk-aligned mamba2 (#24683)

2025-09-29 23:18:25 +02: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] Mamba2 varlen refactor (#21467)

2025-09-26 11:31:14 +00:00

layernorm_gated.py

[Model] Replace Mamba2 RMSNorm Gated with Fused Triton Kernel (#20839)

2025-07-25 06:49:36 -07:00

mamba_ssm.py

[Model] Mamba2 preallocate SSM output tensor to avoid d2d copy overhead (#21075)

2025-08-02 01:59:34 -07:00

ssd_bmm.py

[Kernel] Chunk-aligned mamba2 (#24683)

2025-09-29 23:18:25 +02:00

ssd_chunk_scan.py

[Kernel] Chunk-aligned mamba2 (#24683)

2025-09-29 23:18:25 +02:00

ssd_chunk_state.py

[Kernel] Chunk-aligned mamba2 (#24683)

2025-09-29 23:18:25 +02:00

ssd_combined.py

[Kernel] Chunk-aligned mamba2 (#24683)

2025-09-29 23:18:25 +02:00

ssd_state_passing.py

[Kernel] Chunk-aligned mamba2 (#24683)

2025-09-29 23:18:25 +02:00