diff options
| author | justheuristic <justheuristic@gmail.com> | 2022-09-18 00:42:23 +0300 |
|---|---|---|
| committer | justheuristic <justheuristic@gmail.com> | 2022-09-18 00:42:23 +0300 |
| commit | f6670329fb9b26dc5547cfef6da73bea75c548ca (patch) | |
| tree | bf088ae679cce8dc230fd30b1e092d9a129ee5fd /tests | |
| parent | fa8e07c7c5186e18d9e2d45042814fe6e8d76d5a (diff) | |
bump threshold to 0.21
Diffstat (limited to 'tests')
| -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 0150fbb..cb82898 100644 --- a/tests/test_autograd.py +++ b/tests/test_autograd.py @@ -374,7 +374,7 @@ def test_matmullt( # print(f'abs error {err:.4f}') idx = torch.isclose(out_bnb, out_torch, atol=0.01, rtol=0.1) - assert (idx == 0).sum().item() <= n * (0.0175 if dtype == torch.float16 else 0.02) + assert (idx == 0).sum().item() <= n * (0.0175 if dtype == torch.float16 else 0.021) idx = torch.isclose(out_bnb, out_torch, atol=0.035, rtol=0.2) assert (idx == 0).sum().item() <= n * 0.001 |