cast properly

577275bd · justheuristic · 45dc1983 · 577275bd
Commit 577275bd authored Sep 18, 2022 by justheuristic
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
@@ -294,7 +294,7 @@ class MatMul8bitLt(torch.autograd.Function):
                (outliers * state.SCB.view(-1, 1) / 127.0)
                .t()
                .contiguous()
-                .to(B.dtype)
+                .to(A.dtype)
            )
            CA[:, state.idx.long()] = 0
            CAt[:, state.idx.long()] = 0