Run Pallas Mosaic GPU tests on internal CI

PiperOrigin-RevId: 679508320
jax-ml · Sep 27, 2024 · afaf8b8 · afaf8b8
1 parent 3ae4862
commit afaf8b8
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