ENH: distutils: set NPY_NUM_BUILD_JOBS to cpu_count

brhorgan · Dec 17, 2017 · 51e1114 · 51e1114
1 parent 73df4c3
commit 51e1114
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/numpy/distutils/misc_util.py b/numpy/distutils/misc_util.py
@@ -9,6 +9,7 @@
 import tempfile
 import subprocess
 import shutil
+import multiprocessing
 
 import distutils
 from distutils.errors import DistutilsError
@@ -92,7 +93,11 @@ def get_num_build_jobs():
 
     """
     from numpy.distutils.core import get_distribution
-    envjobs = int(os.environ.get("NPY_NUM_BUILD_JOBS", 1))
+    try:
+        cpu_count = len(os.sched_getaffinity(0))
+    except AttributeError:
+        cpu_count = multiprocessing.cpu_count()
+    envjobs = int(os.environ.get("NPY_NUM_BUILD_JOBS", cpu_count))
     dist = get_distribution()
     # may be None during configuration
     if dist is None: