Something went wrong. Try again.
Nix Packages collection & NixOS github.com/NixOS/nixpkgs
Something went wrong. Try again.
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394{ lib, buildPythonPackage, fetchFromGitHub, cython, setuptools, stdenv, mpi, toPythonModule, pytest, mpiCheckPhaseHook, mpi4py, mpich,}:
buildPythonPackage (finalAttrs: { pname = "mpi4py"; version = "4.1.2"; pyproject = true;
src = fetchFromGitHub { repo = "mpi4py"; owner = "mpi4py"; tag = finalAttrs.version; hash = "sha256-h9RZr+xLmp+cVvrPkew3AOJLE8okd4A/2oqhsSmVBXU="; };
build-system = [ cython setuptools ];
nativeBuildInputs = [ mpi ];
dependencies = [ # Use toPythonModule so that also the mpi executables will be propagated to # generated Python environment. (toPythonModule mpi) ];
pythonImportsCheck = [ "mpi4py" ];
nativeCheckInputs = [ pytest mpiCheckPhaseHook ];
__darwinAllowLocalNetworking = true;
# skip spawn related tests for openmpi implementation # see https://github.com/mpi4py/mpi4py/issues/545#issuecomment-2343011460 env.MPI4PY_TEST_SPAWN = if mpi.pname == "openmpi" then 0 else 1;
disabledTests = [ ] ++ lib.optionals stdenv.hostPlatform.isDarwin [ # testJoin calls socket.getaddrinfo(socket.gethostname(), ...) "test_dynproc.TestDPM.testJoin" ];
# follow upstream's checkPhase # see https://github.com/mpi4py/mpi4py/blob/4.1.0/.github/workflows/ci-test.yml#L92-L95 checkPhase = '' runHook preCheck
echo 'Testing mpi4py (np=1)' mpiexec -n 1 python test/main.py -v echo 'Testing mpi4py (np=2)' mpiexec -n 2 python test/main.py -v -f -e spawn${ lib.concatMapStrings (test: " -x ${test}") finalAttrs.disabledTests }
runHook postCheck '';
passthru = { inherit mpi;
tests = { mpich = mpi4py.override { mpi = mpich; }; }; };
meta = { description = "Python bindings for the Message Passing Interface standard"; homepage = "https://github.com/mpi4py/mpi4py"; changelog = "https://github.com/mpi4py/mpi4py/blob/${finalAttrs.src.tag}/CHANGES.rst"; license = lib.licenses.bsd2; maintainers = with lib.maintainers; [ doronbehar ]; };})