Disable tests that time out in CI.

PiperOrigin-RevId: 532792740
google · May 17, 2023 · e6628e2 · e6628e2
1 parent eaf7eb2
commit e6628e2
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/tests/BUILD b/tests/BUILD
@@ -416,6 +416,9 @@ jax_test(
 jax_test(
     name = "lax_test",
     srcs = ["lax_test.py"],
+    backend_tags = {
+        "tpu": ["noasan"],  # Times out.
+    },
     shard_count = {
         "cpu": 40,
         "gpu": 40,
@@ -629,7 +632,10 @@ jax_test(
     name = "random_test",
     srcs = ["random_test.py"],
     backend_tags = {
-        "cpu": ["notsan"],  # Times out
+        "cpu": [
+            "notsan",  # Times out
+            "nomsan",  # Times out
+        ],
         "tpu": [
             "optonly",
             "nomsan",  # Times out