From 1477ba75e5996d14f907177060f88c9bc36c6259 Mon Sep 17 00:00:00 2001 From: Youngjoon Lee <5462944+youngjoon-lee@users.noreply.github.com> Date: Fri, 2 Aug 2024 22:28:46 +0900 Subject: [PATCH] use only n-1 cores --- mixnet/queuesim/queuesim.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/mixnet/queuesim/queuesim.py b/mixnet/queuesim/queuesim.py index b753a10..e1d3dce 100644 --- a/mixnet/queuesim/queuesim.py +++ b/mixnet/queuesim/queuesim.py @@ -104,7 +104,10 @@ def run_session( future_map: dict[ concurrent.futures.Future[list[float]], tuple[int, ParameterSet, int] ] = dict() - with concurrent.futures.ProcessPoolExecutor() as executor: + total_cores = os.cpu_count() + assert total_cores is not None + max_workers = max(1, total_cores - 1) + with concurrent.futures.ProcessPoolExecutor(max_workers=max_workers) as executor: # Submit all iterations of all parameter sets to the ProcessPoolExecutor for paramset_idx, paramset in enumerate(paramsets): paramset_id = paramset_idx + 1