Skip to content

Commit 1296f82

Browse files
authored
Remove extra contigous for B and C in bwd
1 parent 265d4e2 commit 1296f82

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

mamba_ssm/ops/selective_scan_interface_ compilable.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -156,8 +156,6 @@ def custom_selective_scan_bwd_cuda(
156156
):
157157
if dout.stride(-1) != 1:
158158
dout = dout.contiguous()
159-
B = B.contiguous()
160-
C = C.contiguous()
161159

162160
results = selective_scan_cuda.bwd(
163161
u, delta, A, B, C, D, z, delta_bias, dout, x, out, None, delta_softplus, False

0 commit comments

Comments
 (0)