mpich: enable pmix support

Useful e.g. for using `srun` directly, assuming slurm built with pmix support
NixOS · Jan 12, 2024 · 5b4ad73 · 5b4ad73
1 parent 1a1c53c
commit 5b4ad73
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkgs/development/libraries/mpich/default.nix b/pkgs/development/libraries/mpich/default.nix
@@ -5,6 +5,8 @@
 , ch4backend
 # Process manager to build
 , withPm ? "hydra:gforker"
+, pmix
+, pmixSupport ? true
 } :
 
 assert (ch4backend.pname == "ucx" || ch4backend.pname == "libfabric");
@@ -27,6 +29,8 @@ stdenv.mkDerivation  rec {
   ] ++ 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=${lib.getDev pmix}"
   ];
 
   enableParallelBuilding = true;