wassname/vllm
1
0
Fork 0
mirror of https://github.com/wassname/vllm.git synced 2026-06-27 19:01:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix typo in getenv call (#1972)

Browse Source
This commit is contained in:
Daya Khudia
2023-12-07 16:04:41 -08:00
committed by GitHub
parent 24f60a54f4
commit c8e7eb1eb3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
setup.py
+1 -1
View File
@@ -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)]