pybamm-team#3646 set parallel variable for build_ext (IDAKLU)

mzzhuo · Dec 21, 2023 · 3dc8c8c · 3dc8c8c
1 parent a04fce6
commit 3dc8c8c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -2,6 +2,7 @@
 import sys
 import logging
 import subprocess
+from multiprocessing import cpu_count
 from pathlib import Path
 from platform import system
 import wheel.bdist_wheel as orig
@@ -79,6 +80,9 @@ def run(self):
         if not self.extensions:
             return
 
+        # Build in parallel wherever possible
+        os.environ["CMAKE_BUILD_PARALLEL_LEVEL"] = str(cpu_count())
+
         if system() == "Windows":
             use_python_casadi = False
         else: