diff options
-rw-r--r-- | .gitignore | 1 | ||||
-rw-r--r-- | numpy/distutils/misc_util.py | 7 |
2 files changed, 7 insertions, 1 deletions
diff --git a/.gitignore b/.gitignore index 80ada0d00..c11b8c4cd 100644 --- a/.gitignore +++ b/.gitignore @@ -19,6 +19,7 @@ pmip .sw[nop] *.tmp *.vim +.vscode tags cscope.out # gnu global 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: |