static HYD_status list_to_nodes(char *str)
{
hostlist_t hostlist;
char *host;
int k = 0;
HYD_status status = HYD_SUCCESS;
if ((hostlist = slurm_hostlist_create(str)) == NULL) {
status = HYD_FAILURE;
goto fn_fail;
}
for (host = slurm_hostlist_shift(hostlist); host; host = slurm_hostlist_shift(hostlist)) {
status = HYDU_add_to_node_list(host, tasks_per_node[k++], &global_node_list);
HYDU_ERR_POP(status, "unable to add to node list\n");
}
slurm_hostlist_destroy(hostlist);
fn_exit:
return status;
fn_fail:
goto fn_exit;
}
int _x11_connect_nodes (char* nodes,uint32_t jobid,uint32_t stepid)
{
char* host;
hostlist_t hlist;
int n=0;
int i;
/* count allocated nodes... */
hlist = slurm_hostlist_create(nodes);
n=-1;
do {
n++;
host = slurm_hostlist_shift(hlist);
}
while ( host != NULL ) ;
slurm_hostlist_destroy(hlist);
/* do the export stuff */
hlist = slurm_hostlist_create(nodes);
for (i=0; i < n; i++ ) {
host = slurm_hostlist_shift(hlist);
switch ( x11_mode ) {
case X11_MODE_FIRST :
if ( i == 0 ) {
_connect_node(host,jobid,stepid);
}
break;
case X11_MODE_LAST :
if ( i == (n - 1) ) {
_connect_node(host,jobid,stepid);
}
break;
case X11_MODE_ALL :
_connect_node(host,jobid,stepid);
break;
default :
break;
}
}
slurm_hostlist_destroy(hlist);
return 0;
}
static int _job_resrcs_to_hv(job_info_t *job_info, HV *hv)
{
AV *av;
HV *nr_hv;
bitstr_t *cpu_bitmap;
int sock_inx, sock_reps, last, cnt = 0, i, j, k;
char tmp1[128], tmp2[128];
char *host;
job_resources_t *job_resrcs = job_info->job_resrcs;
int bit_inx, bit_reps;
int abs_node_inx, rel_node_inx;
uint64_t *last_mem_alloc_ptr = NULL;
uint64_t last_mem_alloc = NO_VAL64;
char *last_hosts;
hostlist_t hl, hl_last;
uint32_t threads;
if (!job_resrcs || !job_resrcs->core_bitmap
|| ((last = slurm_bit_fls(job_resrcs->core_bitmap)) == -1))
return 0;
if (!(hl = slurm_hostlist_create(job_resrcs->nodes)))
return 1;
if (!(hl_last = slurm_hostlist_create(NULL)))
return 1;
av = newAV();
bit_inx = 0;
i = sock_inx = sock_reps = 0;
abs_node_inx = job_info->node_inx[i];
/* tmp1[] stores the current cpu(s) allocated */
tmp2[0] = '\0'; /* stores last cpu(s) allocated */
for (rel_node_inx=0; rel_node_inx < job_resrcs->nhosts;
rel_node_inx++) {
if (sock_reps >= job_resrcs->sock_core_rep_count[sock_inx]) {
sock_inx++;
sock_reps = 0;
}
sock_reps++;
bit_reps = job_resrcs->sockets_per_node[sock_inx] *
job_resrcs->cores_per_socket[sock_inx];
host = slurm_hostlist_shift(hl);
threads = _threads_per_core(host);
cpu_bitmap = slurm_bit_alloc(bit_reps * threads);
for (j = 0; j < bit_reps; j++) {
if (slurm_bit_test(job_resrcs->core_bitmap, bit_inx)){
for (k = 0; k < threads; k++)
slurm_bit_set(cpu_bitmap,
(j * threads) + k);
}
bit_inx++;
}
slurm_bit_fmt(tmp1, sizeof(tmp1), cpu_bitmap);
FREE_NULL_BITMAP(cpu_bitmap);
/*
* If the allocation values for this host are not the same as the
* last host, print the report of the last group of hosts that had
* identical allocation values.
*/
if (strcmp(tmp1, tmp2) ||
(last_mem_alloc_ptr != job_resrcs->memory_allocated) ||
(job_resrcs->memory_allocated &&
(last_mem_alloc !=
job_resrcs->memory_allocated[rel_node_inx]))) {
if (slurm_hostlist_count(hl_last)) {
last_hosts =
slurm_hostlist_ranged_string_xmalloc(
hl_last);
nr_hv = newHV();
hv_store_charp(nr_hv, "nodes", last_hosts);
hv_store_charp(nr_hv, "cpu_ids", tmp2);
hv_store_uint64_t(nr_hv, "mem",
last_mem_alloc_ptr ?
last_mem_alloc : 0);
av_store(av, cnt++, newRV_noinc((SV*)nr_hv));
xfree(last_hosts);
slurm_hostlist_destroy(hl_last);
hl_last = slurm_hostlist_create(NULL);
}
strcpy(tmp2, tmp1);
last_mem_alloc_ptr = job_resrcs->memory_allocated;
if (last_mem_alloc_ptr)
last_mem_alloc = job_resrcs->
memory_allocated[rel_node_inx];
else
last_mem_alloc = NO_VAL64;
}
slurm_hostlist_push_host(hl_last, host);
free(host);
if (bit_inx > last)
break;
if (abs_node_inx > job_info->node_inx[i+1]) {
i += 2;
abs_node_inx = job_info->node_inx[i];
} else {
abs_node_inx++;
}
}
if (slurm_hostlist_count(hl_last)) {
last_hosts = slurm_hostlist_ranged_string_xmalloc(hl_last);
nr_hv = newHV();
hv_store_charp(nr_hv, "nodes", last_hosts);
hv_store_charp(nr_hv, "cpu_ids", tmp2);
hv_store_uint64_t(nr_hv, "mem",
last_mem_alloc_ptr ?
last_mem_alloc : 0);
av_store(av, cnt++, newRV_noinc((SV*)nr_hv));
xfree(last_hosts);
}
slurm_hostlist_destroy(hl);
slurm_hostlist_destroy(hl_last);
hv_store_sv(hv, "node_resrcs", newRV_noinc((SV*)av));
return 0;
}
