File "/xxx/selective_scan_interface.py", line 331, in backward dx, dconv1d_weight, dconv1d_bias = causal_conv1d_cuda.causal_conv1d_bwd( RuntimeError: weight must have shape (dim, width)

[zyzcuczyu]

This is the line where the error code is located：

    dx, dconv1d_weight, dconv1d_bias = causal_conv1d_cuda.causal_conv1d_bwd(
        x, conv1d_weight, conv1d_bias, dconv1d_out, dx, None, None, None, True, True
    )

Before that, I print the shape of conv1d_weight:
shape: torch.Size([1536, 4]) is_contiguous: True

It seems that the shape of conv1d_weight is right, but the code still goes wrong. I really don't know why.
The image shows the details.

btw, these are versions of my packages:

Hope someone can solve my problem :(