From 25afcf79ba0d445df0ae5f1301ff658d564c65ba Mon Sep 17 00:00:00 2001
From: Phuong Nguyen <phuonguyen@nvidia.com>
Date: Wed, 8 Jul 2026 13:04:36 -0700
Subject: [PATCH] fix nproc

Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>
---
 setup.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 150d92969c..b61795bc56 100644
--- a/setup.py
+++ b/setup.py
@@ -25,6 +25,7 @@
     remove_dups,
     min_python_version_str,
     nccl_ep_enabled,
+    get_max_jobs_for_parallel_build,
 )
 
 frameworks = get_frameworks()
@@ -250,7 +251,7 @@ def build_nccl_ep_submodule() -> str:
         )
     gencode = " ".join(f"-gencode=arch=compute_{a},code=sm_{a}" for a in arch_list)
 
-    nproc = os.cpu_count() or 8
+    nproc = get_max_jobs_for_parallel_build()
     env = os.environ.copy()
     env["NVCC_GENCODE"] = gencode
     # NCCL EP needs the core NCCL headers + libnccl.so; write NCCL EP build