Skip to content

Commit

Permalink
Remove GPU test with unreasonably large memory footprint.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 695717589
  • Loading branch information
dfm authored and Google-ML-Automation committed Nov 12, 2024
1 parent 21e98b5 commit a99ccd9
Showing 1 changed file with 0 additions and 8 deletions.
8 changes: 0 additions & 8 deletions tests/linalg_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -1450,14 +1450,6 @@ def testLuBatching(self, shape, dtype):
self.assertAllClose(ls, actual_ls, rtol=5e-6)
self.assertAllClose(us, actual_us)

@jtu.skip_on_devices("cpu", "tpu")
@jtu.skip_on_flag("jax_skip_slow_tests", True)
def testBatchedLuOverflow(self):
# see https://github.com/jax-ml/jax/issues/24843
x = self.rng().standard_normal((1500000, 20, 20)).astype(np.float32)
lu, _, _ = lax.linalg.lu(x)
self.assertTrue(jnp.all(lu.std(axis=[1, 2]) > 0.9))

@jtu.skip_on_devices("cpu", "tpu")
@jtu.ignore_warning(category=DeprecationWarning,
message="backend and device argument")
Expand Down

0 comments on commit a99ccd9

Please sign in to comment.