From e2a75769f22bdc5465240c3f6701a1b002e8ab59 Mon Sep 17 00:00:00 2001 From: dbaranchuk Date: Sun, 11 Sep 2022 21:41:46 +0300 Subject: bug fix --- bitsandbytes/autograd/_functions.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'bitsandbytes/autograd/_functions.py') 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 -- cgit v1.2.3