Increase shard count for triton_support_test

PiperOrigin-RevId: 694183667
openxla · Nov 7, 2024 · b0aae98 · b0aae98
1 parent 2bbc3ea
commit b0aae98
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xla/service/gpu/fusions/triton/BUILD b/xla/service/gpu/fusions/triton/BUILD
@@ -726,7 +726,7 @@ cc_library(
 xla_cc_test(
     name = "triton_support_test",
     srcs = ["triton_support_test.cc"],
-    shard_count = 20,
+    shard_count = 25,
     # TODO(b/353912594): this test does not need to run on GPU, but it is broken on CPU in OSS.
     # Force it to run on GPU temporarily in order to get important OSS coverage.
     tags = ["gpu"],