Increase sharding of shard_map test on CPU.

This test is timing out in CI with sanitizers enabled (asan/tsan). PiperOrigin-RevId: 515369731
jax-ml · Mar 9, 2023 · 01b00c4 · 01b00c4
1 parent 9912a8e
commit 01b00c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/BUILD b/tests/BUILD
@@ -1054,7 +1054,7 @@ jax_test(
     name = "shard_map_test",
     srcs = ["shard_map_test.py"],
     shard_count = {
-        "cpu": 10,
+        "cpu": 20,
         "gpu": 10,
         "tpu": 10,
     },