From f699acd9312d63b1f3810d4735bf462e64325379 Mon Sep 17 00:00:00 2001
From: Jake VanderPlas <vanderplas@google.com>
Date: Thu, 27 Oct 2022 13:11:58 -0700
Subject: [PATCH] Internal change

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

diff --git a/tests/BUILD b/tests/BUILD
index dd80984b9..e422722b3 100644
--- a/tests/BUILD
+++ b/tests/BUILD
@@ -700,6 +700,7 @@ jax_test(
     srcs = ["sparsify_test.py"],
     args = ["--jax_bcoo_cusparse_lowering=true"],
     shard_count = {
+        "cpu": 5,
         "gpu": 20,
         "tpu": 10,
     },