diff options
author | Charles Harris <charlesr.harris@gmail.com> | 2017-12-22 19:08:14 -0700 |
---|---|---|
committer | GitHub <noreply@github.com> | 2017-12-22 19:08:14 -0700 |
commit | 3fd3a71952eb3fd35261b002e6f7731b8bb708cc (patch) | |
tree | 515fefc37a14405b67999f1d390541581419a9af /numpy | |
parent | 0e57af86d429e2658b1fbde22960c4c51fc62e84 (diff) | |
parent | 51e1114da42370d6bc64ffe8cb2aa394f47cbea7 (diff) | |
download | numpy-3fd3a71952eb3fd35261b002e6f7731b8bb708cc.tar.gz |
Merge pull request #10225 from xoviat/parallel-builds
ENH: distutils: parallelize builds by default
Diffstat (limited to 'numpy')
-rw-r--r-- | numpy/distutils/misc_util.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/numpy/distutils/misc_util.py b/numpy/distutils/misc_util.py index 102af874f..8bf69ffbd 100644 --- a/numpy/distutils/misc_util.py +++ b/numpy/distutils/misc_util.py @@ -9,6 +9,7 @@ import atexit 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: |