srun: add postSrun hook

2021-04-12 17:41:59 +02:00
parent 4afda7dbfb
commit b0af9b8608
2 changed files with 7 additions and 3 deletions
--- a/garlic/stages/srun.nix
+++ b/garlic/stages/srun.nix
@@ -8,6 +8,7 @@
 , cpuBind
 , nixPrefix
 , preSrun ? ""
+, postSrun ? ""
 , srunOptions ? ""
 , output ? "stdout.log"
 , error ? "stderr.log"
@@ -26,14 +27,17 @@ stdenv.mkDerivation rec {

    ${preSrun}

-    exec ${slurm}/bin/srun \
+    ${slurm}/bin/srun \
      --mpi=pmi2 \
      --cpu-bind=${cpuBind} \
      --output=${output} \
      --error=${error} \
      ${srunOptions} \
      ${nixPrefix}${stageProgram nextStage}
+
+    ${postSrun}
    EOF
+
    chmod +x $out
  '';
 }