Skip to content

Commit bdadc53

Browse files
hawkinspGoogle-ML-Automation
authored andcommitted
Disable JaxAotTest.test_topology_pjit_serialize on GPU, which fails in CI.
PiperOrigin-RevId: 702759889
1 parent 46eb77b commit bdadc53

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
@@ -62,6 +62,7 @@ def verify_serialization(lowered):
6262
jax.pmap(lambda x: x * x).lower(
6363
np.zeros((len(jax.devices()), 4), dtype=np.float32)))
6464

65+
@jtu.skip_on_devices('gpu') # Test fails in CI
6566
def test_topology_pjit_serialize(self):
6667
try:
6768
aot_topo = topologies.get_topology_desc(

0 commit comments

Comments
 (0)