We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 67ab07e commit 68e321cCopy full SHA for 68e321c
tests/linalg_test.py
@@ -340,6 +340,7 @@ def testEighZeroDiagonal(self):
340
np.linalg.norm(np.matmul(a, v) - w * v), 2 * eps * np.linalg.norm(a)
341
)
342
343
+ @jtu.skip_on_devices("rocm")
344
def testEighTinyNorm(self):
345
rng = jtu.rand_default(self.rng())
346
a = rng((300, 300), dtype=np.float32)
@@ -1482,6 +1483,7 @@ def args_maker():
1482
1483
check_dtypes=False, tol=tol)
1484
self._CompileAndCheck(jsp_fun, args_maker, check_dtypes=False)
1485
1486
1487
@jtu.sample_product(
1488
n=[1, 4, 5, 20, 50],
1489
dtype=float_types + complex_types,
0 commit comments