diff options
| author | justheuristic <justheuristic@gmail.com> | 2022-09-17 23:10:21 +0300 |
|---|---|---|
| committer | justheuristic <justheuristic@gmail.com> | 2022-09-17 23:10:21 +0300 |
| commit | a9c7953e0a68a934a18a9495b20deeed9665b2a6 (patch) | |
| tree | eecb0302f509452827e64b473d0ca37c52b05f40 | |
| parent | 469d5a631d77d135f055d3aa012ac852a0ef0856 (diff) | |
cast to half before double_quant
| -rw-r--r-- | tests/test_autograd.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/test_autograd.py b/tests/test_autograd.py index 05da6ed..636fe86 100644 --- 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]: |