#grain Fix issue with running over 50 subprocesses.

PiperOrigin-RevId: 714064248
google · Jan 10, 2025 · 7e5e547 · 7e5e547
1 parent 32d62b5
commit 7e5e547
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/grain/_src/python/shared_memory_array.py b/grain/_src/python/shared_memory_array.py
@@ -135,10 +135,7 @@ def enable_async_del(
     """
     with SharedMemoryArray._lock:
       if not SharedMemoryArray._del_thread_pool:
-        if max_outstanding_requests < num_threads:
-          raise ValueError(
-              "max_outstanding_requests must be at least num_threads."
-          )
+        max_outstanding_requests = max(max_outstanding_requests, num_threads)
         SharedMemoryArray._del_thread_pool = pool.ThreadPool(num_threads)
         SharedMemoryArray._outstanding_del_requests = threading.Semaphore(
             max_outstanding_requests