Add notap tag to Mosaic tests

PiperOrigin-RevId: 635379982
jax-ml · May 20, 2024 · 53ec2cd · 53ec2cd
1 parent ffdb9bb
commit 53ec2cd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/tests/mosaic/BUILD b/tests/mosaic/BUILD
@@ -86,6 +86,7 @@ jax_test(
     disable_backends = DISABLED_BACKENDS,
     disable_configs = DISABLED_CONFIGS,
     main = "//third_party/py/jax/experimental/mosaic/gpu/examples:flash_attention.py",
+    tags = ["notap"],
     deps = [
         "//jax:mosaic_gpu",
     ] + py_deps("numpy"),

diff --git a/tests/pallas/BUILD b/tests/pallas/BUILD
@@ -289,6 +289,7 @@ jax_test(
     env = {
         "JAX_PALLAS_USE_MOSAIC_GPU": "1",
     },
+    tags = ["notap"],
     deps = [
         "//jax:pallas",
         "//jax:pallas_gpu",  # build_cleaner: keep