Add threaded build

54b63712 · Casper Hansen · b7290868 · 54b63712
Commit 54b63712 authored Sep 06, 2023 by Casper Hansen
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

setup.py setup.py +3 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -85,9 +85,11 @@ if os.name == "nt":
        "nvcc": arch_flags
    }
 else:
+    threads = ["--threads", str(min(os.cpu_count(), 8))]
+
    extra_compile_args={
        "cxx": ["-g", "-O3", "-fopenmp", "-lgomp", "-std=c++17"],
-        "nvcc": ["-O3", "-std=c++17"] + arch_flags
+        "nvcc": ["-O3", "-std=c++17"] + arch_flags + threads
    }

 extensions = [