From 20236f1083d68142d888a3d5e18d932086fc3ede Mon Sep 17 00:00:00 2001
From: Gleb Pobudzey <pobudzey@google.com>
Date: Wed, 11 Dec 2024 10:08:08 -0800
Subject: [PATCH] Increase shard count after adding more tests

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

diff --git a/tests/pallas/BUILD b/tests/pallas/BUILD
index 48e2bb5a1a6d..d501f926736c 100644
--- a/tests/pallas/BUILD
+++ b/tests/pallas/BUILD
@@ -481,7 +481,7 @@ jax_multiplatform_test(
         "gpu_a100_x32",
         "gpu_h100_x32",
     ],
-    shard_count = 2,
+    shard_count = 6,
     deps = [
         "//jax:pallas",
         "//jax:pallas_gpu",