diff options
Diffstat (limited to 'lib/slurm.nix')
| -rw-r--r-- | lib/slurm.nix | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/lib/slurm.nix b/lib/slurm.nix index daca4af..30a5f5e 100644 --- a/lib/slurm.nix +++ b/lib/slurm.nix @@ -2,7 +2,7 @@ with lib; -{ ppn, mem, walltime, partition ? null, slurmFlags ? null, salloc ? "/usr/bin/salloc" }: +{ ppn, mem, walltime, partition ? null, slurmFlags ? null, salloc ? "/usr/bin/salloc", srun ? "/usr/bin/srun" }: drv: let ppnReified = if drv.multicore then ppn else 1; in lib.overrideDerivation drv ({ args, builder, name, ... }: { @@ -21,7 +21,7 @@ drv: -J "${name}" \ ${optionalString (partition != null) "-p ${partition}"} \ ${optionalString (slurmFlags != null) slurmFlags} \ - ${script} + ${srun} ${script} ''; in [ "-c" slurm ]; |