Skip to content

GitLab

Commit 4da2227f authored by justheuristic's avatar justheuristic
Browse files

debug

parent 4b4a9eff
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
bitsandbytes/autograd/_functions.py
View file @ 4da2227f
...@@ -370,6 +370,8 @@ class MatMul8bitLt(torch.autograd.Function): ...@@ -370,6 +370,8 @@ class MatMul8bitLt(torch.autograd.Function):
if state.threshold > 0.0 and subA is not None: if state.threshold > 0.0 and subA is not None:
grad_B[:, idx] += torch.matmul(grad_output.t(), subA) grad_B[:, idx] += torch.matmul(grad_output.t(), subA)
raise NotImplementedError("!!")
if req_gradA: if req_gradA:
if state.CBt is not None: if state.CBt is not None:
C32grad, Sgrad = F.transform(Cgrad, "col32") C32grad, Sgrad = F.transform(Cgrad, "col32")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment