diff options
| author | dbaranchuk <dmitrybaranchuk@gmail.com> | 2022-09-11 21:41:46 +0300 |
|---|---|---|
| committer | dbaranchuk <dmitrybaranchuk@gmail.com> | 2022-09-11 21:41:46 +0300 |
| commit | e2a75769f22bdc5465240c3f6701a1b002e8ab59 (patch) | |
| tree | 7fbecaa5cc6ecf7b6dc185b4f1b7eda80fa5ca70 /bitsandbytes/autograd | |
| parent | 4dd475ced4adcbb31f6e1c42225f6d9b1e3be9f2 (diff) | |
bug fix
Diffstat (limited to 'bitsandbytes/autograd')
| -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 |