From c13ae3242936889328620af258fd7bd7f0949411 Mon Sep 17 00:00:00 2001
From: Daniel Suo <dsuo@google.com>
Date: Wed, 10 Dec 2025 07:03:28 -0800
Subject: [PATCH] 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
---
 tests/pjit_test.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/pjit_test.py b/tests/pjit_test.py
index 70e2e1d3503b..bf1e052ad681 100644
--- a/tests/pjit_test.py
+++ b/tests/pjit_test.py
@@ -8006,6 +8006,8 @@ def f2(x, i, j):
       return x.at[i].set(x_j, out_sharding=jax.typeof(x).sharding)
     f2(x,i,j)  # doesn't crash
 
+  # TODO(b/467679041): SIGILL in libtpu nightly with freethreaded Python.
+  @jtu.skip_on_devices("tpu")
   @jtu.with_explicit_mesh((4, 2), ('x', 'y'))
   def test_conv_general_dilated(self, mesh):
     arr = jax.device_put(np.zeros((16, 128, 8)), P('x', 'y'))