vllm/model_executor/models/mamba_cache.py · 711aa9d5b6784d50c2f9ab89642566d5cc3fe4ab · OpenDAS / vllm_cscc

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

Tuan, Hoang-Trong authored Jul 09, 2025


Signed-off-by: Tuan M. Hoang-Trong <tmhoangt@us.ibm.com>
Co-authored-by: Tuan M. Hoang-Trong <tmhoangt@us.ibm.com>
Co-authored-by: Tyler Michael Smith <tysmith@redhat.com>
Co-authored-by: Tyler Michael Smith <tyler@neuralmagic.com>

47043eb6

mamba_cache.py 3.04 KB

Replace mamba_cache.py