[jax] Enable api_test with XLA:CPU thunks

PiperOrigin-RevId: 644268375
google · Jun 18, 2024 · 3fd9326 · 3fd9326
1 parent ed4958c
commit 3fd9326
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/BUILD b/tests/BUILD
@@ -35,6 +35,7 @@ jax_test(
     name = "api_test",
     srcs = ["api_test.py"],
     shard_count = 10,
+    tags = ["test_cpu_thunks"],
 )
 
 jax_test(