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 #19485 from ROCmSoftwarePlatform:rocm-enable_tridiagonal_solve

Browse Source 
PiperOrigin-RevId: 601613417
This commit is contained in:
jax authors 2024-01-25 17:19:00 -08:00
commit 70ea84d67f
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/linalg_test.py
View File

@ -2008,7 +2008,6 @@ class LaxLinalgTest(jtu.JaxTestCase):
eigvals_all[first:(last + 1)], eigvals_index, atol=atol)
@jtu.sample_product(dtype=[np.float32, np.float64])
@jtu.skip_on_devices("rocm") # will be fixed in ROCm-5.1
def test_tridiagonal_solve(self, dtype):
dl = np.array([0.0, 2.0, 3.0], dtype=dtype)
d = np.ones(3, dtype=dtype)