Disable tests that timeout in debug mode in CI

PiperOrigin-RevId: 476157051
spundun · Sep 22, 2022 · d52de20 · d52de20
1 parent a157982
commit d52de20
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/BUILD b/tests/BUILD
@@ -477,6 +477,7 @@ jax_test(
         "tpu": [
             "cpu:8",
             "noasan",  # Times out.
+            "nodebug",  # Times out.
             "notsan",  # Times out.
         ],
     },
@@ -535,9 +536,10 @@ jax_test(
     srcs = ["pmap_test.py"],
     backend_tags = {
         "tpu": [
-            "noasan",
-            "notsan",
-        ],  # Times out under asan/tsan.
+            "noasan",  # Times out.
+            "nodebug",  # Times out.
+            "notsan",  # Times out.
+        ],
     },
     pjrt_c_api_bypass = True,
     shard_count = {