vLLM/vllm/model_executor/layers/mamba at 7728dd77bb802e1876012eb264df4d2fa2fc6f3c - vLLM

Files

T

Chih-Chieh YangandYu Chin Fabian Lim eab2f3980c [Model] Replace Mamba2 RMSNorm Gated with Fused Triton Kernel (#20839 )

Signed-off-by: Chih-Chieh-Yang <7364402+cyang49@users.noreply.github.com>
Signed-off-by: Yu Chin Fabian Lim <fabian.lim@gmail.com>
Signed-off-by: Chih-Chieh Yang <7364402+cyang49@users.noreply.github.com>
Co-authored-by: Yu Chin Fabian Lim <fabian.lim@gmail.com>

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

ops

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

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

__init__.py

[Kernel/Model] Migrate mamba_ssm and causal_conv1d kernels to vLLM (#7651 )

2024-08-28 15:06:52 -07:00

abstract.py

[V1] Enable Mamba2 layers other than MambaMixer2 in the v1 engine (#20660 )

2025-07-11 05:53:31 +00:00

mamba2_metadata.py

[Kernel] Triton implementation of causal-conv1d for Mamba-based models (#18218 )