From 9db3add31923303bc1cc298f7c598d6061daa7b9 Mon Sep 17 00:00:00 2001 From: Binyao Jiang Date: Thu, 13 Nov 2025 14:09:47 -0800 Subject: [PATCH] Update GDN causal conv1d cuda kernel - prepare for new changes (#13188) --- sgl-kernel/csrc/mamba/causal_conv1d.cu | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sgl-kernel/csrc/mamba/causal_conv1d.cu b/sgl-kernel/csrc/mamba/causal_conv1d.cu index 5eb3d9b1b..f31479fdb 100644 --- a/sgl-kernel/csrc/mamba/causal_conv1d.cu +++ b/sgl-kernel/csrc/mamba/causal_conv1d.cu @@ -182,8 +182,8 @@ void causal_conv1d_fwd(const at::Tensor &x, const at::Tensor &weight, TORCH_CHECK(conv_states_.is_cuda()); params.conv_states_ptr = conv_states_.data_ptr(); params.conv_states_batch_stride = conv_states_.stride(0); - params.conv_states_c_stride = conv_states_.stride(1); - params.conv_states_l_stride = conv_states_.stride(2); + params.conv_states_c_stride = conv_states_.stride(-2); + params.conv_states_l_stride = conv_states_.stride(-1); } else { params.conv_states_ptr = nullptr; }