Skip to content

Commit 7906dc4

Browse files
committed
debugpritn
1 parent 2cd047e commit 7906dc4

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

bitsandbytes/autograd/_functions.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -368,6 +368,7 @@ def backward(ctx, grad_output):
368368
gradB32, SgradB32 = F.igemmlt(C32grad, CxAt, Sgrad, SAt)
369369
grad_B = F.mm_dequant(gradB32, SgradB32, SCgradt, SCAt).to(ctx.dtype_B)
370370
if state.threshold > 0.0 and subA is not None:
371+
assert False, idx
371372
grad_B[:, idx] += torch.matmul(grad_output.t(), subA)
372373

373374
if req_gradA:

0 commit comments

Comments
 (0)