chenchenghao/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

Update GDN causal conv1d cuda kernel - prepare for new changes (#13188)

Browse Source
This commit is contained in:
Binyao Jiang
2025-11-13 14:09:47 -08:00
committed by GitHub
parent c9e5799bd1
commit 9db3add319
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
sgl-kernel/csrc/mamba/causal_conv1d.cu
View File

@@ -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;
}