blob: acf41fb539e3c396b266e9c5dd8926e8ee60909b (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
{ 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;
};
}
|