From 6fc72840d0d5107e030e4ac25593c4e9aa9c07d5 Mon Sep 17 00:00:00 2001 From: Phuong Nguyen Date: Mon, 22 Jun 2026 17:48:40 +0200 Subject: [PATCH] remove manually nproc Signed-off-by: Phuong Nguyen --- setup.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 64ed120268..0aae4de622 100644 --- a/setup.py +++ b/setup.py @@ -254,7 +254,6 @@ 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 env = os.environ.copy() env["NVCC_GENCODE"] = gencode # NCCL EP needs the core NCCL headers + libnccl.so; write NCCL EP build @@ -277,7 +276,7 @@ def build_nccl_ep_submodule() -> str: ) print(f"[NCCL EP] Building libnccl_ep.a (gencode='{gencode}')") subprocess.check_call( - ["make", "-j", str(nproc), "-C", "contrib/nccl_ep", "lib"], + ["make", "-j", "-C", "contrib/nccl_ep", "lib"], cwd=str(nccl_root), env=env, )