Disable linalg_test::EighTinyNorm

ROCm · Feb 14, 2024 · 8a11b40 · 8a11b40
1 parent 9e94e6e
commit 8a11b40
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/linalg_test.py b/tests/linalg_test.py
@@ -415,6 +415,7 @@ def testEighZeroDiagonal(self):
           np.linalg.norm(np.matmul(a, v) - w * v), 2 * eps * np.linalg.norm(a)
       )
 
+  @jtu.skip_on_devices("rocm")
   def testEighTinyNorm(self):
     rng = jtu.rand_default(self.rng())
     a = rng((300, 300), dtype=np.float32)
@@ -1747,6 +1748,7 @@ def args_maker():
                               check_dtypes=False, tol=tol)
       self._CompileAndCheck(jsp_fun, args_maker, check_dtypes=False)
 
+  @jtu.skip_on_devices("rocm")
   @jtu.sample_product(
     n=[1, 4, 5, 20, 50],
     dtype=float_types + complex_types,