summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--tests/test_autograd.py3
1 files changed, 1 insertions, 2 deletions
diff --git a/tests/test_autograd.py b/tests/test_autograd.py
index 4e4282a..0150fbb 100644
--- a/tests/test_autograd.py
+++ b/tests/test_autograd.py
@@ -372,10 +372,9 @@ def test_matmullt(
n = out_bnb.numel()
err = torch.abs(out_bnb - out_torch).mean().item()
# print(f'abs error {err:.4f}')
- out_error_rate = 0.0175 if dtype == torch.float16 else 0.02
idx = torch.isclose(out_bnb, out_torch, atol=0.01, rtol=0.1)
- assert (idx == 0).sum().item() <= n * out_error_rate
+ assert (idx == 0).sum().item() <= n * (0.0175 if dtype == torch.float16 else 0.02)
idx = torch.isclose(out_bnb, out_torch, atol=0.035, rtol=0.2)
assert (idx == 0).sum().item() <= n * 0.001