Increase shard count after adding more tests

PiperOrigin-RevId: 705146601
jax-ml · Dec 11, 2024 · 20236f1 · 20236f1
1 parent 01206f8
commit 20236f1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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",