117 lines
3.2 KiB
Nix
117 lines
3.2 KiB
Nix
{
|
|
stdenv,
|
|
lib,
|
|
fetchurl,
|
|
perl,
|
|
gfortran,
|
|
automake,
|
|
autoconf,
|
|
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.3";
|
|
|
|
src = fetchurl {
|
|
url = "https://www.mpich.org/static/downloads/${version}/mpich-${version}.tar.gz";
|
|
hash = "sha256-egGRgMUdFzitnF2NRSMU3mXoKO4kC8stH4DemmW+iKg=";
|
|
};
|
|
|
|
patches = [
|
|
# Disables ROMIO test which was enabled in
|
|
# https://github.com/pmodels/mpich/commit/09686f45d77b7739f7aef4c2c6ef4c3060946595
|
|
# The test searches for mpicc in $out/bin, which is not yet present in the checkPhase
|
|
# Moreover it fails one test.
|
|
./disable-romio-tests.patch
|
|
];
|
|
|
|
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
|
|
autoconf
|
|
automake
|
|
];
|
|
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 = {
|
|
# 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 = [ lib.maintainers.markuskowa ];
|
|
platforms = lib.platforms.linux ++ lib.platforms.darwin;
|
|
};
|
|
}
|