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

Fix some warnings causing CI failures on ARM.

Browse Source 
PiperOrigin-RevId: 678454816
This commit is contained in:
Peter Hawkins 2024-09-24 17:24:32 -07:00 committed by jax authors
parent 7603901608
commit 291e52a713
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/lax_numpy_test.py
View File

@ -5706,6 +5706,7 @@ class LaxBackedNumpyTests(jtu.JaxTestCase):
self.assertAllClose(x.trace(), jnp.array(x).trace())
self.assertAllClose(x.trace(), jax.jit(lambda y: y.trace())(x))
@jtu.ignore_warning(category=RuntimeWarning, message="divide by zero")
def testIntegerPowersArePrecise(self):
# See https://github.com/jax-ml/jax/pull/3036
# Checks if the squares of float32 integers have no numerical errors.

1
tests/scipy_optimize_test.py
View File

@ -117,6 +117,7 @@ class TestBFGS(jtu.JaxTestCase):
jax_res = jax.scipy.optimize.minimize(fun=eval_func, x0=x0, method='BFGS')
self.assertLess(jax_res.fun, 1e-6)
@jtu.ignore_warning(category=RuntimeWarning, message='divide by zero')
def test_minimize_bad_initial_values(self):
# This test runs deliberately "bad" initial values to test that handling
# of failed line search, etc. is the same across implementations