Merge pmix outputs for MPICH

MPICH expects headers and libraries to be present in the same directory.

Reviewed-by: Aleix Roca Nonell <aleix.rocanonell@bsc.es>
This commit is contained in:
Rodrigo Arias 2024-03-14 16:59:11 +01:00
parent 49be0f208c
commit 834d3187e5

View File

@ -4,10 +4,16 @@ final: prev:
mpi = final.mpich; mpi = final.mpich;
# Configure the network for MPICH # Configure the network for MPICH
mpich = with final; prev.mpich.overrideAttrs (old: { mpich = with final; let
# pmix comes with the libraries in .out and headers in .dev
pmixAll = symlinkJoin {
name = "pmix-all";
paths = [ pmix.dev pmix.out ];
};
in prev.mpich.overrideAttrs (old: {
buildInput = old.buildInputs ++ [ buildInput = old.buildInputs ++ [
libfabric libfabric
pmix pmixAll
]; ];
configureFlags = [ configureFlags = [
"--enable-shared" "--enable-shared"
@ -15,8 +21,8 @@ final: prev:
"--with-pm=no" "--with-pm=no"
"--with-device=ch4:ofi" "--with-device=ch4:ofi"
"--with-pmi=pmix" "--with-pmi=pmix"
"--with-pmix=${final.pmix}" "--with-pmix=${pmixAll}"
"--with-libfabric=${final.libfabric}" "--with-libfabric=${libfabric}"
"--enable-g=log" "--enable-g=log"
] ++ lib.optionals (lib.versionAtLeast gfortran.version "10") [ ] ++ lib.optionals (lib.versionAtLeast gfortran.version "10") [
"FFLAGS=-fallow-argument-mismatch" # https://github.com/pmodels/mpich/issues/4300 "FFLAGS=-fallow-argument-mismatch" # https://github.com/pmodels/mpich/issues/4300