clearer assertions

fc4a135e · justheuristic · e29c5f5c · fc4a135e
Commit fc4a135e authored Sep 17, 2022 by justheuristic
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

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

No files found.
--- a/bitsandbytes/autograd/_functions.py
+++ b/bitsandbytes/autograd/_functions.py
@@ -232,8 +232,8 @@ class MatMul8bitLt(torch.autograd.Function):
        # Cast A to fp16
        A_dtype = A.dtype
        if A_dtype != torch.float16:
-            warnings.warn(f"MatMul8bitLt: temporarily casting input matrix from {A_dtype} to float16")
-        A = A.to(torch.float16)
+            warnings.warn(f"MatMul8bitLt: input matrix will be converted from {A_dtype} to float16")
+            A = A.to(torch.float16)

        # 1. Quantize A
        if len(A.shape) == 3: