From e8c735125ca260af6e7d50b0dffaca5d38625983 Mon Sep 17 00:00:00 2001
From: Peter Hawkins <phawkins@google.com>
Date: Fri, 5 May 2023 08:32:56 -0700
Subject: [PATCH] Disable more tests that are flaky in CI.

PiperOrigin-RevId: 529724306
---
 tests/BUILD | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/tests/BUILD b/tests/BUILD
index 829832dde..97be608d7 100644
--- a/tests/BUILD
+++ b/tests/BUILD
@@ -630,7 +630,10 @@ jax_test(
     srcs = ["random_test.py"],
     backend_tags = {
         "cpu": ["notsan"],  # Times out
-        "tpu": ["optonly"],
+        "tpu": [
+            "optonly",
+            "nomsan",  # Times out
+        ],
     },
     shard_count = {
         "cpu": 30,
@@ -775,6 +778,7 @@ jax_test(
     srcs = ["sparsify_test.py"],
     args = ["--jax_bcoo_cusparse_lowering=true"],
     backend_tags = {
+        "cpu": ["noasan"],  # Times out under asan
         "tpu": [
             "noasan",  # Times out under asan.
         ],