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

Merge pull request #12197 from ROCmSoftwarePlatform:fixedRocmUnitTestsSkip

Browse Source 
PiperOrigin-RevId: 479566021
This commit is contained in:
jax authors 2022-10-07 06:36:11 -07:00
commit 363cc124e3
2 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/custom_linear_solve_test.py
View File

@ -227,7 +227,6 @@ class CustomLinearSolveTest(jtu.JaxTestCase):
order=2,
rtol={jnp.float32: 6e-2, jnp.float64: 2e-3})
@jtu.skip_on_devices("rocm") # rtol and atol needs to be adjusted for ROCm
def test_custom_linear_solve_cholesky(self):
def positive_definite_solve(a, b):

1
tests/lax_scipy_test.py
View File

@ -409,7 +409,6 @@ class LaxBackedScipyTests(jtu.JaxTestCase):
self.assertAllClose(actual, expected, rtol=1.1e-7, atol=3e-8)
@jtu.skip_on_devices("rocm") # rtol and atol needs to be adjusted for ROCm
@jax.numpy_dtype_promotion('standard') # This test explicitly exercises dtype promotion
def testSphHarmOrderZeroDegreeOne(self):
"""Tests the spherical harmonics of order one and degree zero."""