main (int argc, char **argv) { char *task_str; MY_MASK cpu_mask; MY_MASK mem_mask; int task_id; if (numa_available() < 0) { fprintf(stderr, "ERROR: numa support not available\n"); exit(0); } if ((task_str = getenv("SLURM_PROCID")) == NULL) { fprintf(stderr, "ERROR: getenv(SLURM_PROCID) failed\n"); exit(1); } task_id = atoi(task_str); _load_cpu_mask(&cpu_mask); _load_mem_mask(&mem_mask); printf("TASK_ID:%d,CPU_MASK:%lu,MEM_MASK:%lu\n", task_id, _mask_to_int(&cpu_mask), _mask_to_int(&mem_mask)); exit(0); }
main (int argc, char **argv) { char *task_str; cpu_set_t mask; int task_id; _load_mask(&mask); /* On POE systems, MP_CHILD is equivalent to SLURM_PROCID */ if (((task_str = getenv("SLURM_PROCID")) == NULL) && ((task_str = getenv("MP_CHILD")) == NULL)) { fprintf(stderr, "ERROR: getenv(SLURM_PROCID) failed\n"); exit(1); } task_id = atoi(task_str); printf("TASK_ID:%d,MASK:%u\n", task_id, _mask_to_int(&mask)); exit(0); }