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
69
70
71
72
73
74
75
76
77
78
79
80
|
{ stdenv, lib, fetchurl, perl, gfortran
, openssh, hwloc, python3
, darwin
# either libfabric or ucx work for ch4backend on linux. On darwin, neither of
# these libraries currently build so this argument is ignored on Darwin.
, ch4backend
# Process managers to build (`--with-pm`),
# cf. https://github.com/pmodels/mpich/blob/b80a6d7c24defe7cdf6c57c52430f8075a0a41d6/README.vin#L562-L586
, withPm ? [ "hydra" "gforker" ]
, pmix
# PMIX support is likely incompatible with process managers (`--with-pm`)
# https://github.com/NixOS/nixpkgs/pull/274804#discussion_r1432601476
, pmixSupport ? false
} :
let
withPmStr = if withPm != [ ] then builtins.concatStringsSep ":" withPm else "no";
in
assert (ch4backend.pname == "ucx" || ch4backend.pname == "libfabric");
stdenv.mkDerivation rec {
pname = "mpich";
version = "4.2.2";
src = fetchurl {
url = "https://www.mpich.org/static/downloads/${version}/mpich-${version}.tar.gz";
sha256 = "sha256-iD9bs66r9ifLhJLKAqA7GR0Jg2u+D1mdhQg1EXl4HUE=";
};
outputs = [ "out" "doc" "man" ];
configureFlags = [
"--enable-shared"
"--with-pm=${withPmStr}"
] ++ lib.optionals (lib.versionAtLeast gfortran.version "10") [
"FFLAGS=-fallow-argument-mismatch" # https://github.com/pmodels/mpich/issues/4300
"FCFLAGS=-fallow-argument-mismatch"
] ++ lib.optionals pmixSupport [
"--with-pmix"
];
enableParallelBuilding = true;
nativeBuildInputs = [ gfortran python3 ];
buildInputs = [ perl openssh hwloc ]
++ lib.optional (!stdenv.hostPlatform.isDarwin) ch4backend
++ lib.optional pmixSupport pmix
++ lib.optional stdenv.hostPlatform.isDarwin darwin.apple_sdk.frameworks.Foundation;
# test_double_serializer.test fails on darwin
doCheck = !stdenv.hostPlatform.isDarwin;
preFixup = ''
# Ensure the default compilers are the ones mpich was built with
sed -i 's:CC="gcc":CC=${stdenv.cc}/bin/gcc:' $out/bin/mpicc
sed -i 's:CXX="g++":CXX=${stdenv.cc}/bin/g++:' $out/bin/mpicxx
sed -i 's:FC="gfortran":FC=${gfortran}/bin/gfortran:' $out/bin/mpifort
'';
meta = with lib; {
# As far as we know, --with-pmix silently disables all of `--with-pm`
broken = pmixSupport && withPm != [ ];
description = "Implementation of the Message Passing Interface (MPI) standard";
longDescription = ''
MPICH2 is a free high-performance and portable implementation of
the Message Passing Interface (MPI) standard, both version 1 and
version 2.
'';
homepage = "http://www.mcs.anl.gov/mpi/mpich2/";
license = {
url = "http://git.mpich.org/mpich.git/blob/a385d6d0d55e83c3709ae851967ce613e892cd21:/COPYRIGHT";
fullName = "MPICH license (permissive)";
};
maintainers = [ maintainers.markuskowa ];
platforms = platforms.linux ++ platforms.darwin;
};
}
|
