pybamm-team#3646 set parallel jobs for pybamm_install_odes

agriyakhetarpal · Dec 22, 2023 · 139e34d · 139e34d
1 parent 3dc8c8c
commit 139e34d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pybamm/install_odes.py b/pybamm/install_odes.py
@@ -5,6 +5,7 @@
 import sys
 import logging
 import subprocess
+from multiprocessing import cpu_count
 
 from pybamm.util import root_dir
 
@@ -16,6 +17,8 @@
 except ModuleNotFoundError:
     NO_WGET = True
 
+# Build in parallel wherever possible
+os.environ["CMAKE_BUILD_PARALLEL_LEVEL"] = str(cpu_count())
 
 def download_extract_library(url, directory):
     # Download and extract archive at url