Alldaychubby

Results 2 comments of Alldaychubby

Thanks for the solution! I tried to modify `causal_conv1d_cuda.causal_conv1d_fwd`, but encountered another problem, may I ask how `causal_conv1d_cuda.causal_conv1d_bwd` should be modified? ![image](https://github.com/hustvl/Vim/assets/61933775/f9909dec-8587-4914-bae9-e9dffdf07135)

在mamba-1p1p1/mamba_ssm/ops/selective_scan_interface.py文件里,`conv1d_out = causal_conv1d_cuda.causal_conv1d_fwd(x, conv1d_weight, conv1d_bias,None, True)` 改为 `conv1d_out = causal_conv1d_cuda.causal_conv1d_fwd(x, conv1d_weight, conv1d_bias, True)`,还有这里也修改 ``` dx, dconv1d_weight, dconv1d_bias = causal_conv1d_cuda.causal_conv1d_bwd( # x, conv1d_weight, conv1d_bias, dconv1d_out, None, dx, True x, conv1d_weight, conv1d_bias,...