From e6628e2e721587fd48c751b5f69d8acdbb92b4bf Mon Sep 17 00:00:00 2001
From: Peter Hawkins <phawkins@google.com>
Date: Wed, 17 May 2023 08:15:32 -0700
Subject: [PATCH] Disable tests that time out in CI.

PiperOrigin-RevId: 532792740
---
 tests/BUILD | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/tests/BUILD b/tests/BUILD
index 6fa02dce4..88e437693 100644
--- a/tests/BUILD
+++ b/tests/BUILD
@@ -416,6 +416,9 @@ jax_test(
 jax_test(
     name = "lax_test",
     srcs = ["lax_test.py"],
+    backend_tags = {
+        "tpu": ["noasan"],  # Times out.
+    },
     shard_count = {
         "cpu": 40,
         "gpu": 40,
@@ -629,7 +632,10 @@ jax_test(
     name = "random_test",
     srcs = ["random_test.py"],
     backend_tags = {
-        "cpu": ["notsan"],  # Times out
+        "cpu": [
+            "notsan",  # Times out
+            "nomsan",  # Times out
+        ],
         "tpu": [
             "optonly",
             "nomsan",  # Times out