diff --git a/setup.py b/setup.py index 146da3bb..b33fbbb6 100644 --- a/setup.py +++ b/setup.py @@ -85,11 +85,9 @@ def get_compute_capabilities(): "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 + threads + "nvcc": ["-O3", "-std=c++17"] + arch_flags } extensions = [