From 6f7ce9d0486b3f9f9ae3f0f5dcaf62800004fdf5 Mon Sep 17 00:00:00 2001
From: Adam Paszke <apaszke@google.com>
Date: Tue, 11 Mar 2025 10:27:37 -0700
Subject: [PATCH] Skip ASAN tests for the big Mosaic GPU tests

They are timing out.

PiperOrigin-RevId: 735804647
---
 tests/mosaic/BUILD | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/tests/mosaic/BUILD b/tests/mosaic/BUILD
index 4b231939a..4ec2dbf3b 100644
--- a/tests/mosaic/BUILD
+++ b/tests/mosaic/BUILD
@@ -39,7 +39,10 @@ jax_multiplatform_test(
     ],
     env = {"XLA_FLAGS": "--xla_gpu_autotune_level=0"},
     shard_count = 16,
-    tags = ["multiaccelerator"],
+    tags = [
+        "multiaccelerator",
+        "noasan",  # Times out.
+    ],
     deps = [
         "//jax:mosaic_gpu",
     ] + py_deps("absl/testing") + py_deps("numpy"),