JakeHillion/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

python3Packages.numpy: fix tests for >64 core machines

Browse Source
This commit is contained in:
Jonathan Ringer 2020-11-17 08:31:06 -08:00 committed by Jonathan Ringer
parent d72a58a684
commit 8f7cec2297
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pkgs/development/python-modules/numpy/default.nix
View File

@ -58,9 +58,13 @@ in buildPythonPackage rec {
./numpy-distutils-C++.patch
];
# we default openblas to build with 64 threads
# if a machine has more than 64 threads, it will segfault
# see https://github.com/xianyi/OpenBLAS/issues/2993
preConfigure = ''
sed -i 's/-faltivec//' numpy/distutils/system_info.py
export NPY_NUM_BUILD_JOBS=$NIX_BUILD_CORES
export OMP_NUM_THREADS=$((NIX_BUILD_CORES > 64 ? 64 : NIX_BUILD_CORES))
'';
preBuild = ''