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

Merge pull request #17860 from google:test_fix

Browse Source 
PiperOrigin-RevId: 569549919
This commit is contained in:
jax authors 2023-09-29 11:53:02 -07:00
commit 4d933ebc58
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

@ -365,7 +365,7 @@ class NumpyLinalgTest(jtu.JaxTestCase):
with jax.numpy_rank_promotion("allow"):
self.assertLessEqual(
np.linalg.norm(np.matmul(a, v) - w * v),
80 * eps * np.linalg.norm(a),
81 * eps * np.linalg.norm(a),
)
@jtu.sample_product(