diff options
| author | xoviat <xoviat@users.noreply.github.com> | 2017-12-17 10:55:23 -0600 |
|---|---|---|
| committer | xoviat <xoviat@users.noreply.github.com> | 2017-12-17 10:55:23 -0600 |
| commit | 51e1114da42370d6bc64ffe8cb2aa394f47cbea7 (patch) | |
| tree | 838182f8141c982a775c467adc601891e50d93eb /numpy/distutils/misc_util.py | |
| parent | 73df4c3933c6a63f075087f90cac5e538f4e6514 (diff) | |
| download | numpy-51e1114da42370d6bc64ffe8cb2aa394f47cbea7.tar.gz | |
ENH: distutils: set NPY_NUM_BUILD_JOBS to cpu_count
Diffstat (limited to 'numpy/distutils/misc_util.py')
| -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: |