Skip to content

GitLab

Commit 54b63712 authored by Casper Hansen's avatar Casper Hansen
Browse files

Add threaded build

parent b7290868
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
setup.py
View file @ 54b63712
...@@ -85,9 +85,11 @@ if os.name == "nt": ...@@ -85,9 +85,11 @@ if os.name == "nt":
"nvcc": arch_flags "nvcc": arch_flags
} }
else: else:
threads = ["--threads", str(min(os.cpu_count(), 8))]
extra_compile_args={ extra_compile_args={
"cxx": ["-g", "-O3", "-fopenmp", "-lgomp", "-std=c++17"], "cxx": ["-g", "-O3", "-fopenmp", "-lgomp", "-std=c++17"],
"nvcc": ["-O3", "-std=c++17"] + arch_flags "nvcc": ["-O3", "-std=c++17"] + arch_flags + threads
} }
extensions = [ extensions = [
......
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