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

[ROCm]: Re-enable SvdOnTinyElement test

Browse Source
This commit is contained in:
Rahul Batra 2023-06-23 17:50:00 +00:00
parent 63415a9184
commit c5bf05d66b
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)