Skip to content

Commit f667032

Browse files
committed
bump threshold to 0.21
1 parent fa8e07c commit f667032

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_autograd.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -374,7 +374,7 @@ def test_matmullt(
374374
# print(f'abs error {err:.4f}')
375375

376376
idx = torch.isclose(out_bnb, out_torch, atol=0.01, rtol=0.1)
377-
assert (idx == 0).sum().item() <= n * (0.0175 if dtype == torch.float16 else 0.02)
377+
assert (idx == 0).sum().item() <= n * (0.0175 if dtype == torch.float16 else 0.021)
378378
idx = torch.isclose(out_bnb, out_torch, atol=0.035, rtol=0.2)
379379
assert (idx == 0).sum().item() <= n * 0.001
380380

0 commit comments

Comments
 (0)