Add mpich overlay

This commit is contained in:
Rodrigo Arias 2023-06-16 14:16:51 +02:00
parent 684d5e41c5
commit ab04855382
2 changed files with 30 additions and 0 deletions

View File

@ -9,6 +9,8 @@
./slurm.nix ./slurm.nix
./ssh.nix ./ssh.nix
./users.nix ./users.nix
../../pkgs/mpi.nix
]; ];
nixpkgs.overlays = [ bscpkgs.bscOverlay ]; nixpkgs.overlays = [ bscpkgs.bscOverlay ];

28
pkgs/mpi.nix Normal file
View File

@ -0,0 +1,28 @@
let
overlay = final: prev:
{
mpich = with final; prev.mpich.overrideAttrs (old: {
buildInput = old.buildInputs ++ [
libfabric
pmix
];
configureFlags = [
"--enable-shared"
"--enable-sharedlib"
"--with-pm=no"
"--with-device=ch4:ofi"
"--with-pmi=pmix"
"--with-pmix=${final.pmix}"
"--with-libfabric=${final.libfabric}"
"--enable-g=log"
] ++ lib.optionals (lib.versionAtLeast gfortran.version "10") [
"FFLAGS=-fallow-argument-mismatch" # https://github.com/pmodels/mpich/issues/4300
"FCFLAGS=-fallow-argument-mismatch"
];
});
};
in
{
nixpkgs.overlays = [ overlay ];
}