From c8e7eb1eb3e3862cde4f7c27952d2eb9104ef416 Mon Sep 17 00:00:00 2001
From: Daya Khudia <37562707+dskhudia@users.noreply.github.com>
Date: Thu, 7 Dec 2023 16:04:41 -0800
Subject: [PATCH] fix typo in getenv call (#1972)

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 95a7d7060fefc..09e36a0c8a124 100644
--- 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)]