mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bump test tolerance for sqrtm test.

Browse Source 
This test fails on ARM with a LAPACK built with gfortran 11.

PiperOrigin-RevId: 569540626
This commit is contained in:
Peter Hawkins 2023-09-29 11:15:05 -07:00 committed by jax authors
parent 59d86b233e
commit ef6fd2ebb6
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/linalg_test.py
View File

@ -1586,7 +1586,7 @@ class ScipyLinalgTest(jtu.JaxTestCase):
rng = jtu.rand_default(self.rng())
arg = rng(shape, dtype)
if dtype == np.float32 or dtype == np.complex64:
tol = 1e-3
tol = 2e-3
else:
tol = 1e-8
R = jsp.linalg.sqrtm(arg)