From 7102c7adbfd618e33fc5e1915eb69517d2948fd9 Mon Sep 17 00:00:00 2001
From: Adam Paszke <apaszke@google.com>
Date: Tue, 8 Oct 2024 02:43:57 -0700
Subject: [PATCH] Bump the shard_count of FFT tests to avoid timeouts

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

diff --git a/tests/BUILD b/tests/BUILD
index 188b5ae81..e444c01d0 100644
--- a/tests/BUILD
+++ b/tests/BUILD
@@ -852,7 +852,7 @@ jax_multiplatform_test(
             "nomsan",
         ],  # Times out on TPU with asan/tsan/msan.
     },
-    shard_count = 4,
+    shard_count = 12,
 )
 
 jax_multiplatform_test(