srun: add postSrun hook

This commit is contained in:
2021-04-12 17:41:59 +02:00
parent 4afda7dbfb
commit b0af9b8608
2 changed files with 7 additions and 3 deletions

View File

@@ -76,12 +76,12 @@ rec {
inherit nextStage;
};
srun = {nextStage, conf, preSrun ? "", ...}: (
srun = {nextStage, conf, preSrun ? "", postSrun ? "", ...}: (
assert (assertMsg (!(conf ? cpuBind))
"cpuBind is no longer available in the standard srun stage");
stages.srun {
inherit (conf) nixPrefix;
inherit nextStage preSrun;
inherit nextStage preSrun postSrun;
# Binding is set to cores always
cpuBind = "cores,verbose";