From 051a74b85dbc2fe8d486a516500a64b8b1218c81 Mon Sep 17 00:00:00 2001
From: Rodrigo Arias Mallo <rodrigo.arias@bsc.es>
Date: Fri, 26 Feb 2021 17:00:09 +0100
Subject: [PATCH] srun: allow commands to run before srun

---
 garlic/stages/srun.nix | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/garlic/stages/srun.nix b/garlic/stages/srun.nix
index e6231b5..d7e8539 100644
--- 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} \