Skip to content

Commit 272f649

Browse files
danielsuoGoogle-ML-Automation
authored andcommitted
Disable failing TPU presubmit test.
Fails with SIGILL in ShardingInTypesTest.test_conv_general_dilated. https://github.com/jax-ml/jax/actions/runs/20105632869/job/57688572609 PiperOrigin-RevId: 842710731
1 parent 6a1397e commit 272f649

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/pjit_test.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8006,6 +8006,8 @@ def f2(x, i, j):
80068006
return x.at[i].set(x_j, out_sharding=jax.typeof(x).sharding)
80078007
f2(x,i,j) # doesn't crash
80088008

8009+
# TODO(b/467679041): SIGILL in libtpu nightly with freethreaded Python.
8010+
@jtu.skip_on_devices("tpu")
80098011
@jtu.with_explicit_mesh((4, 2), ('x', 'y'))
80108012
def test_conv_general_dilated(self, mesh):
80118013
arr = jax.device_put(np.zeros((16, 128, 8)), P('x', 'y'))

0 commit comments

Comments
 (0)