mirror of
https://github.com/NixOS/nixpkgs.git
synced 2024-12-23 22:23:15 +00:00
277b73ab6f
This adds support building with MKL.
50 lines
1.2 KiB
Nix
50 lines
1.2 KiB
Nix
{ lib
|
|
, buildPythonPackage
|
|
, fetchPypi
|
|
, python
|
|
, numpy
|
|
, llvmPackages ? null
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "numexpr";
|
|
version = "2.6.8";
|
|
|
|
src = fetchPypi {
|
|
inherit pname version;
|
|
sha256 = "ee8bc7201aa2f1962c67d27c326a11eef9df887d7b87b1278a1d4e722bf44375";
|
|
};
|
|
|
|
# Remove existing site.cfg, use the one we built for numpy.
|
|
# Somehow openmp needs to be added to LD_LIBRARY_PATH
|
|
# https://software.intel.com/en-us/forums/intel-system-studio/topic/611682
|
|
preBuild = ''
|
|
rm site.cfg
|
|
ln -s ${numpy.cfg} site.cfg
|
|
export LD_LIBRARY_PATH=${llvmPackages.openmp}/lib
|
|
'';
|
|
|
|
buildInputs = [] ++ lib.optional (numpy.blasImplementation == "mkl") llvmPackages.openmp;
|
|
|
|
propagatedBuildInputs = [ numpy ];
|
|
|
|
# Run the test suite.
|
|
# It requires the build path to be in the python search path.
|
|
checkPhase = ''
|
|
pushd $out
|
|
${python}/bin/${python.executable} <<EOF
|
|
import sys
|
|
import numexpr
|
|
r = numexpr.test()
|
|
if not r.wasSuccessful():
|
|
sys.exit(1)
|
|
EOF
|
|
popd
|
|
'';
|
|
|
|
meta = {
|
|
description = "Fast numerical array expression evaluator for NumPy";
|
|
homepage = "https://github.com/pydata/numexpr";
|
|
license = lib.licenses.mit;
|
|
};
|
|
} |