Skip to content

Commit 34a10de

Browse files
apaszkeGoogle-ML-Automation
authored andcommitted
[Pallas] Bump some libtpu date conditions
Looks like some changes are taking longer to make it into libtpu than expected. PiperOrigin-RevId: 835236077
1 parent 3a9b471 commit 34a10de

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

tests/pallas/ops_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1511,9 +1511,9 @@ def test_dot_general_non_front_batch_dims(self, shapes_and_dims_numbers):
15111511
self.skipTest("TPU only test")
15121512

15131513
if jtu.test_device_matches(["tpu"]) and not jtu.if_cloud_tpu_at_least(
1514-
2025, 11, 21
1514+
2025, 11, 30
15151515
):
1516-
self.skipTest("Requires libtpu built after 2025-11-21")
1516+
self.skipTest("Requires libtpu built after 2025-11-30")
15171517

15181518
x_shape, y_shape, dims_numbers = shapes_and_dims_numbers
15191519

tests/pallas/tpu_sparsecore_pallas_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1566,7 +1566,7 @@ def kernel(in_ref, o_ref, scratch_ref):
15661566
np.testing.assert_array_equal(f(x), x)
15671567

15681568
def test_exp(self):
1569-
if not jtu.if_cloud_tpu_at_least(2025, 11, 21):
1569+
if not jtu.if_cloud_tpu_at_least(2025, 11, 30):
15701570
self.skipTest("Test requires a newer libtpu")
15711571

15721572
x = jnp.arange(8, dtype=jnp.float32)

0 commit comments

Comments
 (0)