Skip to content

Commit 587c9bd

Browse files
hawkinspGoogle-ML-Automation
authored andcommitted
Disable a TPU test that is segfaulting in presubmits.
PiperOrigin-RevId: 800934633
1 parent 9495be7 commit 587c9bd

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/aot_test.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,7 @@ def verify_serialization(lowered):
6565
jax.pmap(lambda x: x * x).lower(
6666
np.zeros((len(jax.devices()), 4), dtype=np.float32)))
6767

68+
@jtu.skip_on_devices("tpu") # TODO(phawkins): This test is segfaulting on TPU
6869
def test_topology_pjit_serialize(self):
6970
try:
7071
aot_topo = topologies.get_topology_desc(

0 commit comments

Comments
 (0)