Skip to content

Commit cfaf6e8

Browse files
belitskiyGoogle-ML-Automation
authored andcommitted
Skip test_scalar_debug_check in tpu_sparsecore_pallas_debug_check_test.
Causes pytest testsuite to hang in its entirety, and fails on Bazel, while running on v6e-8. PiperOrigin-RevId: 845416829
1 parent ba024e3 commit cfaf6e8

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/pallas/tpu_sparsecore_pallas_debug_check_test.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -68,9 +68,9 @@ def setUp(self):
6868
super().setUp()
6969

7070
def test_scalar_debug_check(self):
71-
if not jtu.is_device_tpu_at_least(6):
72-
# TODO: b/436509694 - Figure out why the test gets stuck on v5p.
73-
self.skipTest("")
71+
if not jtu.is_device_tpu_at_least(7):
72+
# TODO: b/469486032 - Figure out why the test gets stuck on v5p, v6e.
73+
self.skipTest("Fails on v5p and v6e.")
7474

7575
x = jnp.arange(8)
7676

0 commit comments

Comments
 (0)