summaryrefslogtreecommitdiff
path: root/bitsandbytes/autograd
diff options
context:
space:
mode:
authordbaranchuk <dmitrybaranchuk@gmail.com>2022-09-11 21:41:46 +0300
committerdbaranchuk <dmitrybaranchuk@gmail.com>2022-09-11 21:41:46 +0300
commite2a75769f22bdc5465240c3f6701a1b002e8ab59 (patch)
tree7fbecaa5cc6ecf7b6dc185b4f1b7eda80fa5ca70 /bitsandbytes/autograd
parent4dd475ced4adcbb31f6e1c42225f6d9b1e3be9f2 (diff)
bug fix
Diffstat (limited to 'bitsandbytes/autograd')
-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