Skip to content

GitLab

Unverified Commit c8e7eb1e authored by Daya Khudia's avatar Daya Khudia Committed by GitHub
Browse files

fix typo in getenv call (#1972)

parent 24f60a54
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
setup.py
View file @ c8e7eb1e
...@@ -138,7 +138,7 @@ for capability in compute_capabilities: ...@@ -138,7 +138,7 @@ for capability in compute_capabilities:
# Use NVCC threads to parallelize the build. # Use NVCC threads to parallelize the build.
if nvcc_cuda_version >= Version("11.2"): 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) num_threads = min(os.cpu_count(), nvcc_threads)
NVCC_FLAGS += ["--threads", str(num_threads)] NVCC_FLAGS += ["--threads", str(num_threads)]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment