Reenable some test cases that were disabled due to bugs that now seem…

… fixed.

PiperOrigin-RevId: 684451155

tests/BUILD

@@ -736,13 +736,6 @@ jax_multiplatform_test(
 jax_multiplatform_test(
     name = "pytorch_interoperability_test",
     srcs = ["pytorch_interoperability_test.py"],
-    # The following cases are disabled because they time out in Google's CI, mostly because the
-    # CUDA kernels in Torch take a very long time to compile.
-    disable_configs = [
-        "gpu_p100",  # Pytorch P100 build times out in Google's CI.
-        "gpu_a100",  # Pytorch A100 build times out in Google's CI.
-        "gpu_h100",  # Pytorch H100 build times out in Google's CI.
-    ],
     enable_backends = [
         "cpu",
         "gpu",
@@ -1395,10 +1388,8 @@ jax_multiplatform_test(
         "tpu": 20,
     },
     tags = [
-        "noasan",  # Times out, TODO(b/314760446): test failures on Sapphire Rapids.
+        "noasan",  # Times out
         "nodebug",  # Times out.
-        "nomsan",  # TODO(b/314760446): test failures on Sapphire Rapids.
-        "notsan",  # TODO(b/314760446): test failures on Sapphire Rapids.
     ],
     deps = [
         "//jax:internal_test_harnesses",