From 912a5ef7714749cf06a635e78faf98cadff8df61 Mon Sep 17 00:00:00 2001
From: Peter Hawkins <phawkins@google.com>
Date: Mon, 15 Jan 2024 01:40:05 -0800
Subject: [PATCH] Disable some tests that time out or OOM under ASAN.

PiperOrigin-RevId: 598543036
---
 tests/BUILD        | 2 ++
 tests/pallas/BUILD | 1 +
 2 files changed, 3 insertions(+)

diff --git a/tests/BUILD b/tests/BUILD
index c020de74a..81822c5b6 100644
--- a/tests/BUILD
+++ b/tests/BUILD
@@ -337,6 +337,7 @@ jax_test(
         "tpu": 10,
         "iree": 10,
     },
+    tags = ["noasan"],  # Linking TF causes a linker OOM.
     deps = py_deps("pil") + py_deps("tensorflow_core"),
 )
 
@@ -1150,6 +1151,7 @@ jax_test(
 jax_test(
     name = "host_callback_to_tf_test",
     srcs = ["host_callback_to_tf_test.py"],
+    tags = ["noasan"],  # Linking TF causes a linker OOM.
     deps = [
         "//jax:experimental_host_callback",
         "//jax:ode",
diff --git a/tests/pallas/BUILD b/tests/pallas/BUILD
index fd6ef2a1e..f5ab3d56d 100644
--- a/tests/pallas/BUILD
+++ b/tests/pallas/BUILD
@@ -93,6 +93,7 @@ jax_test(
     ],
     shard_count = 18,
     tags = [
+        "noasan",  # Times out.
         "nomsan",  # Times out.
         "notsan",  # Times out.
     ],