tests/kernels/mamba/test_causal_conv1d.py · 47043eb6787bc8f62e4e746d7a9ad2451aaf7983 · 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

test_causal_conv1d.py 14.7 KB

Replace test_causal_conv1d.py