Run Pallas Mosaic GPU tests on internal CI

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