77 lines
2.1 KiB
Nix
77 lines
2.1 KiB
Nix
{ stdenv, lib, fetchurl, perl, gfortran
|
|
, openssh, hwloc, autoconf, automake, libtool
|
|
# device options are ch3 or ch4
|
|
, device ? "ch4"
|
|
# backend option are libfabric or ucx
|
|
, ch4backend ? "libfabric"
|
|
, ucx, libfabric
|
|
# Process manager to build
|
|
, withPm ? "hydra:gforker"
|
|
} :
|
|
|
|
assert (ch4backend == "ucx" || ch4backend == "libfabric");
|
|
|
|
stdenv.mkDerivation rec {
|
|
pname = "mpich";
|
|
version = "3.4.1";
|
|
|
|
src = fetchurl {
|
|
url = "https://www.mpich.org/static/downloads/${version}/mpich-${version}.tar.gz";
|
|
sha256 = "09wpfw9lsrc84vcmfw94razd4xv4znx4mmg7rqmljvgg0jc96dl8";
|
|
};
|
|
|
|
patches = [
|
|
# Reverts an upstream change that causes json-c test to fail
|
|
./jsonc-test.patch
|
|
];
|
|
|
|
# Required for the json-c patch
|
|
nativeBuildInputs = [ autoconf automake libtool ];
|
|
|
|
# Update configure after patch
|
|
postPatch = ''
|
|
cd modules/json-c
|
|
./autogen.sh
|
|
cd ../..
|
|
'';
|
|
|
|
|
|
configureFlags = [
|
|
"--enable-shared"
|
|
"--enable-sharedlib"
|
|
"--with-pm=${withPm}"
|
|
];
|
|
|
|
enableParallelBuilding = true;
|
|
|
|
buildInputs = [ perl gfortran openssh hwloc ]
|
|
++ lib.optional (ch4backend == "ucx") ucx
|
|
++ lib.optional (ch4backend == "libfabric") libfabric;
|
|
|
|
doCheck = true;
|
|
|
|
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; {
|
|
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;
|
|
};
|
|
}
|