From 77b89c0d8e841b0b338729410d3f8ac9a426fa0a Mon Sep 17 00:00:00 2001 From: Sergei Lebedev Date: Fri, 27 Sep 2024 01:54:05 -0700 Subject: [PATCH] Run Pallas Mosaic GPU tests on internal CI PiperOrigin-RevId: 679494302 --- tests/pallas/BUILD | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/pallas/BUILD b/tests/pallas/BUILD index ba82b8c4223c..b1db085a8c3a 100644 --- 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