[Pallas TPU] Enable test for jnp.logical_not because it's now supported

PiperOrigin-RevId: 702439876

Author: ayaka14732

tests/pallas/ops_test.py

@@ -808,14 +808,6 @@ def test_elementwise(self, fn, dtype):
     ):
       self.skipTest(f"{fn.__name__} not implemented on TPU")
 
-    # TODO: https://github.com/jax-ml/jax/issues/24243
-    if (
-        jtu.test_device_matches(["tpu"])
-        and fn == jnp.logical_not
-        and not self.INTERPRET
-    ):
-      self.skipTest("logical_not on TPU is only supported in interpret mode")
-
     @functools.partial(
         self.pallas_call,
         out_shape=jax.ShapeDtypeStruct((8, 128), dtype),