nixpkgs/pkgs/development/python-modules/mpi4py/default.nix

52 lines
1.5 KiB
Nix
Raw Normal View History

{ lib, fetchPypi, python, buildPythonPackage, mpi, openssh }:
2014-08-31 19:02:25 +01:00
buildPythonPackage rec {
pname = "mpi4py";
2019-11-11 11:13:37 +00:00
version = "3.0.3";
2014-08-31 19:02:25 +01:00
2017-09-15 17:11:45 +01:00
src = fetchPypi {
inherit pname version;
2019-11-11 11:13:37 +00:00
sha256 = "012d716c8b9ed1e513fcc4b18e5af16a8791f51e6d1716baccf988ad355c5a1f";
2014-08-31 19:02:25 +01:00
};
passthru = {
inherit mpi;
};
postPatch = ''
substituteInPlace test/test_spawn.py --replace \
"unittest.skipMPI('openmpi(<3.0.0)')" \
"unittest.skipMPI('openmpi')"
'';
2014-08-31 19:02:25 +01:00
configurePhase = "";
installPhase = ''
mkdir -p "$out/lib/${python.libPrefix}/site-packages"
export PYTHONPATH="$out/lib/${python.libPrefix}/site-packages:$PYTHONPATH"
${python}/bin/${python.executable} setup.py install \
--install-lib=$out/lib/${python.libPrefix}/site-packages \
--prefix="$out"
# --install-lib:
# sometimes packages specify where files should be installed outside the usual
# python lib prefix, we override that back so all infrastructure (setup hooks)
# work as expected
# Needed to run the tests reliably. See:
# https://bitbucket.org/mpi4py/mpi4py/issues/87/multiple-test-errors-with-openmpi-30
export OMPI_MCA_rmaps_base_oversubscribe=yes
2014-08-31 19:02:25 +01:00
'';
setupPyBuildFlags = ["--mpicc=${mpi}/bin/mpicc"];
nativeBuildInputs = [ mpi openssh ];
2017-09-15 17:11:45 +01:00
meta = with lib; {
description = "Python bindings for the Message Passing Interface standard";
homepage = "https://bitbucket.org/mpi4py/mpi4py/";
license = licenses.bsd3;
2014-08-31 19:02:25 +01:00
};
}