fix typo in getenv call (#1972)

c8e7eb1e · Daya Khudia · GitHub · 24f60a54 · c8e7eb1e
Unverified Commit c8e7eb1e authored Dec 07, 2023 by Daya Khudia Committed by GitHub Dec 07, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

setup.py setup.py +1 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -138,7 +138,7 @@ for capability in compute_capabilities:

 # 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)]