From c5bf05d66b015f3318f232a9a79ec15dcc5e48ee Mon Sep 17 00:00:00 2001 From: Rahul Batra Date: Fri, 23 Jun 2023 17:50:00 +0000 Subject: [PATCH] [ROCm]: Re-enable SvdOnTinyElement test --- tests/svd_test.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/svd_test.py b/tests/svd_test.py index abb1f4037..58c912cb0 100644 --- a/tests/svd_test.py +++ b/tests/svd_test.py @@ -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)