From 9ac92bce878f4ee7d917f032713a9f675d73e3d8 Mon Sep 17 00:00:00 2001
From: Morris Jette <jette@schedmd.com>
Date: Thu, 25 Jul 2013 16:31:09 -0700
Subject: [PATCH] Set srun default task count properly when --ntasks-per-node
specified
Similar problem to that described in bug 343 for sbatch
There are many differences in the salloc/srun/sbatch argument
processing that should probably be made more uniform, but no time
to do so now
---
src/srun/libsrun/opt.c | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/src/srun/libsrun/opt.c b/src/srun/libsrun/opt.c
index 96a8a21dab5..81a7549a901 100644
--- a/src/srun/libsrun/opt.c
+++ b/src/srun/libsrun/opt.c
@@ -1960,7 +1960,8 @@ static bool _opt_verify(void)
opt.ntasks *= opt.cores_per_socket;
opt.ntasks *= opt.threads_per_core;
opt.ntasks_set = true;
- }
+ } else if (opt.ntasks_per_node > 0)
+ opt.ntasks *= opt.ntasks_per_node;
/* massage the numbers */
if (opt.nodelist) {
--
GitLab