From 32026ad18bd767d2c1b72bae5fa1a65591defea3 Mon Sep 17 00:00:00 2001
From: Peter Hawkins <phawkins@google.com>
Date: Thu, 25 May 2023 10:05:37 -0700
Subject: [PATCH] Disable random_test_with_custom_prng on CPU under msan.

This test flakily times out in CI.

PiperOrigin-RevId: 535293997
---
 tests/BUILD | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/BUILD b/tests/BUILD
index f11b8fec7..5c7891713 100644
--- a/tests/BUILD
+++ b/tests/BUILD
@@ -657,7 +657,8 @@ jax_test(
     args = ["--jax_enable_custom_prng=true"],
     backend_tags = {
         "cpu": [
-            "noasan",  # Times out under asan/tsan.
+            "noasan",  # Times out under asan/msan/tsan.
+            "nomsan",
             "notsan",
         ],
         "tpu": [