bug fix

e2a75769 · dbaranchuk · 4dd475ce · e2a75769
Commit e2a75769 authored Sep 11, 2022 by dbaranchuk
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

bitsandbytes/autograd/_functions.py bitsandbytes/autograd/_functions.py +1 -1

No files found.
--- 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