From 01b00c482162e1de2606ca8f81b102c4ea72eabd Mon Sep 17 00:00:00 2001
From: Peter Hawkins <phawkins@google.com>
Date: Thu, 9 Mar 2023 10:12:44 -0800
Subject: [PATCH] Increase sharding of shard_map test on CPU.

This test is timing out in CI with sanitizers enabled (asan/tsan).

PiperOrigin-RevId: 515369731
---
 tests/BUILD | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/BUILD b/tests/BUILD
index 61aa58b84..2eb2d1a56 100644
--- 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,
     },