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 #16542 from ROCmSoftwarePlatform:rocm-enable-svdontiny-test

Browse Source 
PiperOrigin-RevId: 547178719
This commit is contained in:
jax authors 2023-07-11 07:04:45 -07:00
commit 3ec5f73db0
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/svd_test.py
View File

@ -217,7 +217,6 @@ class SvdTest(jtu.JaxTestCase):
for m, n, r, c in zip([2, 4, 8], [4, 4, 6], [1, 0, 1], [1, 0, 1])],
dtype=jtu.dtypes.floating,
)
@jtu.skip_on_devices("rocm")
def testSvdOnTinyElement(self, m, n, r, c, dtype):
"""Tests SVD on matrix of zeros and close-to-zero entries."""
a = jnp.zeros((m, n), dtype=dtype)