fix typo in getenv call (vllm-project#1972)

EmbeddedLLM · Dec 8, 2023 · c8e7eb1 · c8e7eb1
1 parent 24f60a5
commit c8e7eb1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -138,7 +138,7 @@ def get_torch_arch_list() -> Set[str]:
 
 # Use NVCC threads to parallelize the build.
 if nvcc_cuda_version >= Version("11.2"):
-    nvcc_threads = int(os.getenv("NVCC_THREADS"), 8)
+    nvcc_threads = int(os.getenv("NVCC_THREADS", 8))
     num_threads = min(os.cpu_count(), nvcc_threads)
     NVCC_FLAGS += ["--threads", str(num_threads)]