From fe56a1990417591c9543cabfd315e2bac22167c7 Mon Sep 17 00:00:00 2001
From: Adam Paszke <apaszke@google.com>
Date: Wed, 23 Nov 2022 06:32:41 -0800
Subject: [PATCH] Shard fft tests to avoid timeouts

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

diff --git a/tests/BUILD b/tests/BUILD
index 49394523b..64b41fc58 100644
--- a/tests/BUILD
+++ b/tests/BUILD
@@ -124,6 +124,7 @@ jax_test(
     },
     shard_count = {
         "tpu": 20,
+        "cpu": 20,
     },
 )