From df475287fe496ebea20f68980dc18db2b0a29222 Mon Sep 17 00:00:00 2001 From: Ayaka Date: Wed, 15 Jan 2025 10:31:59 -0800 Subject: [PATCH] [Pallas TPU] Remove obsolete skip condition PiperOrigin-RevId: 715853352 --- tests/pallas/tpu_pallas_test.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/tests/pallas/tpu_pallas_test.py b/tests/pallas/tpu_pallas_test.py index 746485d689fd..86e96ab5fc38 100644 --- a/tests/pallas/tpu_pallas_test.py +++ b/tests/pallas/tpu_pallas_test.py @@ -2127,10 +2127,6 @@ def kernel(x_ref, o_ref): for dtype in (jnp.int32, jnp.uint32, jnp.float32) ) def test_debug_print_vector(self, shape, dtype): - # TODO(ayx): Remove after this date. - if not jtu.if_cloud_tpu_at_least(2025, 1, 16): - self.skipTest("Requires libtpu built after 2025-01-16") - @functools.partial( self.pallas_call, out_shape=jax.ShapeDtypeStruct(shape, dtype),