cast to half before double_quant

a9c7953e · justheuristic · 469d5a63 · a9c7953e
Commit a9c7953e authored Sep 17, 2022 by justheuristic
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/test_autograd.py tests/test_autograd.py +1 -1

No files found.
--- a/tests/test_autograd.py
+++ b/tests/test_autograd.py
@@ -354,7 +354,7 @@ def test_matmullt(
                    state.SCB,
                    SCBt,
                    coo_tensorB,
-                ) = bnb.functional.double_quant(B2)
+                ) = bnb.functional.double_quant(B2.half())
                B2 = state.CB

            if not transpose[0] and transpose[1]: