From 401fbb61a97116a4588d5aac2ef9ed01ab85e406 Mon Sep 17 00:00:00 2001
From: Peter Hawkins <phawkins@google.com>
Date: Mon, 5 Dec 2022 06:51:28 -0800
Subject: [PATCH] Disable xmap_test on TPU under asan due to CI timeouts.

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

diff --git a/tests/BUILD b/tests/BUILD
index 1ce103b32..21d8657a3 100644
--- a/tests/BUILD
+++ b/tests/BUILD
@@ -169,6 +169,9 @@ py_test(
 jax_test(
     name = "xmap_test",
     srcs = ["xmap_test.py"],
+    backend_tags = {
+        "tpu": ["noasan"],  # Times out.
+    },
     pjrt_c_api_bypass = True,
     shard_count = {
         "cpu": 10,