summaryrefslogtreecommitdiff
path: root/pkgs/development/python-modules/mpi4py/default.nix
blob: bf894a84e5689cdd0b7d786b0ae8750085be3311 (about) (plain)
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
{
  lib,
  buildPythonPackage,
  fetchFromGitHub,
  cython,
  setuptools,
  mpi,
  toPythonModule,
  pytestCheckHook,
  mpiCheckPhaseHook,
}:

buildPythonPackage rec {
  pname = "mpi4py";
  version = "4.0.3";
  pyproject = true;

  src = fetchFromGitHub {
    repo = "mpi4py";
    owner = "mpi4py";
    tag = version;
    hash = "sha256-eN/tjlnNla6RHYOXcprVVqtec1nwCEGn+MBcV/5mHJg=";
  };

  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 = [
    pytestCheckHook
    mpiCheckPhaseHook
  ];
  disabledTestPaths = lib.optionals (mpi.pname == "mpich") [
    # These tests from some reason cause pytest to crash, and therefor it is
    # hard to debug them. Upstream mentions these tests to raise issues in
    # https://github.com/mpi4py/mpi4py/issues/418  but the workaround suggested
    # there (setting MPI4PY_RC_RECV_MPROBE=0) doesn't work.
    "test/test_util_pool.py"
    "demo/futures/test_futures.py"
  ];

  __darwinAllowLocalNetworking = true;

  passthru = {
    inherit mpi;
  };

  meta = {
    description = "Python bindings for the Message Passing Interface standard";
    homepage = "https://github.com/mpi4py/mpi4py";
    changelog = "https://github.com/mpi4py/mpi4py/blob/${src.tag}/CHANGES.rst";
    license = lib.licenses.bsd2;
    maintainers = with lib.maintainers; [ doronbehar ];
  };
}