Run Pallas Mosaic GPU tests on internal CI

PiperOrigin-RevId: 679494302
jax-ml · Sep 27, 2024 · 7b06f70 · 7b06f70
1 parent ab4590c
commit 7b06f70
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/tests/pallas/BUILD b/tests/pallas/BUILD
@@ -205,7 +205,6 @@ jax_multiplatform_test(
     env = {
         "JAX_PALLAS_USE_MOSAIC_GPU": "1",
     },
-    tags = ["notap"],
     deps = [
         "//jax:pallas",
         "//jax:pallas_gpu",  # build_cleaner: keep