From d8eafc8ee3d44420331e0e15b9b93abe260c718b Mon Sep 17 00:00:00 2001
From: Sergei Lebedev <slebedev@google.com>
Date: Thu, 8 Aug 2024 13:01:55 -0700
Subject: [PATCH] Disabled nn_test under asan on TPU as well, since it also
 times out

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

diff --git a/tests/BUILD b/tests/BUILD
index b5a99b254..07069b08e 100644
--- a/tests/BUILD
+++ b/tests/BUILD
@@ -692,6 +692,9 @@ jax_test(
         "gpu": [
             "noasan",  # Times out under asan.
         ],
+        "tpu": [
+            "noasan",  # Times out under asan.
+        ],
     },
     shard_count = {
         "cpu": 10,