From 5b97960d3144542beaf5600e1e9ef3fb76700ce3 Mon Sep 17 00:00:00 2001
From: Peter Hawkins <phawkins@google.com>
Date: Mon, 4 Dec 2023 10:28:28 -0800
Subject: [PATCH] Disable sanitizer builds of shape_poly_test.

These take a very long time and sometimes timeout so it's probably not worth running them in CI.

PiperOrigin-RevId: 587768399
---
 tests/BUILD | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/tests/BUILD b/tests/BUILD
index 0df288d1f..29776452d 100644
--- a/tests/BUILD
+++ b/tests/BUILD
@@ -1307,13 +1307,6 @@ jax_test(
 jax_test(
     name = "shape_poly_test",
     srcs = ["shape_poly_test.py"],
-    backend_tags = {
-        "tpu": [
-            "noasan",
-            "nomsan",
-            "notsan",
-        ],  # Times out
-    },
     disable_configs = [
         "gpu_a100",  # TODO(b/269593297): matmul precision issues
     ],
@@ -1326,6 +1319,11 @@ jax_test(
         "gpu": 4,
         "tpu": 4,
     },
+    tags = [
+        "noasan",  # Times out
+        "nomsan",  # Times out
+        "notsan",  # Times out
+    ],
     deps = [
         "//jax:internal_test_harnesses",
         "//jax/experimental/export",