srun: allow commands to run before srun

2021-02-26 17:00:09 +01:00 · 2021-02-26 17:00:09 +01:00 · 051a74b85d
commit 051a74b85d
parent 8a77900201
1 changed files with 4 additions and 0 deletions
--- a/garlic/stages/srun.nix
+++ b/garlic/stages/srun.nix
@ -7,6 +7,7 @@
  nextStage
 , cpuBind
 , nixPrefix
+, preSrun ? ""
 , srunOptions ? ""
 , output ? "stdout.log"
 , error ? "stderr.log"
@ -22,6 +23,9 @@ stdenv.mkDerivation rec {
  installPhase = ''
    cat > $out <<'EOF'
    #!/bin/sh -e
+
+    ${preSrun}
+
    exec ${slurm}/bin/srun \
      --mpi=pmi2 \
      --cpu-bind=${cpuBind} \