rarias/bscpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

srun: allow commands to run before srun

Browse Source
This commit is contained in:
Rodrigo Arias 2021-02-26 17:00:09 +01:00
parent 8a77900201
commit 051a74b85d
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
garlic/stages/srun.nix
View File

@ -7,6 +7,7 @@
nextStage nextStage
, cpuBind , cpuBind
, nixPrefix , nixPrefix
, preSrun ? ""
, srunOptions ? "" , srunOptions ? ""
, output ? "stdout.log" , output ? "stdout.log"
, error ? "stderr.log" , error ? "stderr.log"
@ -22,6 +23,9 @@ stdenv.mkDerivation rec {
installPhase = '' installPhase = ''
cat > $out <<'EOF' cat > $out <<'EOF'
#!/bin/sh -e #!/bin/sh -e
${preSrun}
exec ${slurm}/bin/srun \ exec ${slurm}/bin/srun \
--mpi=pmi2 \ --mpi=pmi2 \
--cpu-bind=${cpuBind} \ --cpu-bind=${cpuBind} \