Skip to content

Commit a99ccd9

Browse files
dfmGoogle-ML-Automation
authored andcommitted
Remove GPU test with unreasonably large memory footprint.
PiperOrigin-RevId: 695717589
1 parent 21e98b5 commit a99ccd9

File tree

1 file changed

+0
-8
lines changed

1 file changed

+0
-8
lines changed

tests/linalg_test.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1450,14 +1450,6 @@ def testLuBatching(self, shape, dtype):
14501450
self.assertAllClose(ls, actual_ls, rtol=5e-6)
14511451
self.assertAllClose(us, actual_us)
14521452

1453-
@jtu.skip_on_devices("cpu", "tpu")
1454-
@jtu.skip_on_flag("jax_skip_slow_tests", True)
1455-
def testBatchedLuOverflow(self):
1456-
# see https://github.com/jax-ml/jax/issues/24843
1457-
x = self.rng().standard_normal((1500000, 20, 20)).astype(np.float32)
1458-
lu, _, _ = lax.linalg.lu(x)
1459-
self.assertTrue(jnp.all(lu.std(axis=[1, 2]) > 0.9))
1460-
14611453
@jtu.skip_on_devices("cpu", "tpu")
14621454
@jtu.ignore_warning(category=DeprecationWarning,
14631455
message="backend and device argument")

0 commit comments

Comments
 (0)