summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--bitsandbytes/autograd/_functions.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/bitsandbytes/autograd/_functions.py b/bitsandbytes/autograd/_functions.py
index 48f867f..bdcbec5 100644
--- a/bitsandbytes/autograd/_functions.py
+++ b/bitsandbytes/autograd/_functions.py
@@ -388,7 +388,7 @@ class MatMul8bitLt(torch.autograd.Function):
grad_bias = grad_output.sum(0)
# Cast grad_A back to grad_output_dtype
- grad_output.to(grad_output_dtype)
+ grad_output = grad_output.to(grad_output_dtype)
return grad_A, grad_B, None, grad_bias, None