check dtypes first

9379df85 · justheuristic · 140cdbe8 · 9379df85
Commit 9379df85 authored Sep 17, 2022 by justheuristic
Hide 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
@@ -367,7 +367,7 @@ def test_matmullt(
            if has_bias:
                out_torch += bias

-            assert out_bnb.dtype == torch.dtype
+            assert out_bnb.dtype == out_torch.dtype

            n = out_bnb.numel()
            err = torch.abs(out_bnb - out_torch).mean().item()