[Pallas TPU] Remove obsolete skip condition

PiperOrigin-RevId: 715853352
jax-ml · Jan 16, 2025 · 57682a9 · 57682a9
1 parent bc7204f
commit 57682a9
Showing 1 changed file with 0 additions and 4 deletions.
diff --git 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),