Skip to content

Commit 62ccf13

Browse files
relax tolerance of float32 test
1 parent 687877c commit 62ccf13

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/tensor/test_slinalg.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1086,4 +1086,7 @@ def test_banded_dot(A_shape, kl, ku):
10861086

10871087
out_val, out_2_val = fn(A_val, x_val)
10881088

1089-
np.testing.assert_allclose(out_val, out_2_val)
1089+
atol = 1e-4 if config.floatX == "float32" else 1e-8
1090+
rtol = 1e-4 if config.floatX == "float32" else 1e-8
1091+
1092+
np.testing.assert_allclose(out_val, out_2_val, atol=atol, rtol=rtol)

0 commit comments

Comments
 (0)