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 9495be7 commit 587c9bdCopy full SHA for 587c9bd
tests/aot_test.py
@@ -65,6 +65,7 @@ def verify_serialization(lowered):
65
jax.pmap(lambda x: x * x).lower(
66
np.zeros((len(jax.devices()), 4), dtype=np.float32)))
67
68
+ @jtu.skip_on_devices("tpu") # TODO(phawkins): This test is segfaulting on TPU
69
def test_topology_pjit_serialize(self):
70
try:
71
aot_topo = topologies.get_topology_desc(
0 commit comments