From d52de206cb349173d961a57d0288eafbf423959d Mon Sep 17 00:00:00 2001
From: Tyler Augustine <taugustine@google.com>
Date: Thu, 22 Sep 2022 11:44:22 -0700
Subject: [PATCH] Disable tests that timeout in debug mode in CI

PiperOrigin-RevId: 476157051
---
 tests/BUILD | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/tests/BUILD b/tests/BUILD
index 8283b53ad..7456446d6 100644
--- 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 = {