From 305efe050157b1d89065dc85f9d811dbd803067f Mon Sep 17 00:00:00 2001
From: jax authors <no-reply@google.com>
Date: Wed, 4 Oct 2023 13:04:05 -0700
Subject: [PATCH] random_test: reduce num_generated_cases to avoid timeouts

PiperOrigin-RevId: 570781641
---
 tests/BUILD | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/tests/BUILD b/tests/BUILD
index 27f2c115d..f8de3810d 100644
--- a/tests/BUILD
+++ b/tests/BUILD
@@ -704,6 +704,11 @@ jax_test(
             "notsan",  # Times out
         ],
     },
+    # Use fewer cases to prevent timeouts.
+    backend_variant_args = {
+        "cpu": ["--jax_num_generated_cases=40"],
+        "cpu_x32": ["--jax_num_generated_cases=40"],
+    },
     shard_count = {
         "cpu": 30,
         "gpu": 30,