Update Slurm launch support

Assign all cpu's on node to the daemon

Signed-off-by: Ralph Castain <[email protected]>

Author: rhc54

orte/mca/plm/slurm/plm_slurm_module.c

@@ -184,6 +184,7 @@ static void launch_daemons(int fd, short args, void *cbdata)
     char **custom_strings;
     int num_args, i;
     char *cur_prefix;
+    char *cpus_on_node;
     int proc_vpid_index;
     bool failed_launch=true;
     orte_job_t *daemons;
@@ -267,6 +268,14 @@ static void launch_daemons(int fd, short args, void *cbdata)
     /* start one orted on each node */
     opal_argv_append(&argc, &argv, "--ntasks-per-node=1");
 
+    /* add all CPUs to this task */
+     cpus_on_node = getenv("SLURM_CPUS_ON_NODE");
+     if (NULL != cpus_on_node) {
+         asprintf(&tmp, "--cpus-per-task=%s", cpus_on_node);
+         opal_argv_append(&argc, &argv, tmp);
+         free(tmp);
+     }
+
     if (!orte_enable_recovery) {
         /* kill the job if any orteds die */
         opal_argv_append(&argc, &argv, "--kill-on-bad-exit");