From f5ac7eccd9a5410fc68c5f871e1921809aab5069 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 45d6cf2b5ee2..5d9a1985da5b 100644 --- a/tests/pallas/tpu_pallas_test.py +++ b/tests/pallas/tpu_pallas_test.py @@ -2125,10 +2125,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),