diff options
| -rw-r--r-- | bitsandbytes/autograd/_functions.py | 2 |
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 |