vllm/csrc/mamba/causal_conv1d
Mor Zusman 9fb12f7848
[BugFix][Kernel] Fix Illegal memory access in causal_conv1d in H100 (#9838)
Signed-off-by: mzusman <mor.zusmann@gmail.com>
2024-10-31 20:06:25 +00:00
..
causal_conv1d.cu [BugFix][Kernel] Fix Illegal memory access in causal_conv1d in H100 (#9838) 2024-10-31 20:06:25 +00:00
causal_conv1d.h [Kernel][Model] Improve continuous batching for Jamba and Mamba (#9189) 2024-10-16 12:12:43 -04:00
static_switch.h [Kernel/Model] Migrate mamba_ssm and causal_conv1d kernels to vLLM (#7651) 2024-08-28 15:06:52 -07:00