check dtypes first

bitsandbytes-foundation · Sep 17, 2022 · 9379df8 · 9379df8
1 parent 140cdbe
commit 9379df8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()