d4f56cff61
It is okay to not have a paffinity module IF you aren't using paffinity anyway. So don't error out of MPI_Init because a paffinity module wasn't selected. Cleanup error reporting in the odls default module to (once and for all!) eliminate messages originating in the fork'd process. Create some new error codes to allow us to pass enough info back to the parent process to provide useful error messages. This commit was SVN r23106.
607 строки
27 KiB
C
607 строки
27 KiB
C
/*
|
|
* Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
|
|
* University Research and Technology
|
|
* Corporation. All rights reserved.
|
|
* Copyright (c) 2004-2006 The University of Tennessee and The University
|
|
* of Tennessee Research Foundation. All rights
|
|
* reserved.
|
|
* Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
|
|
* University of Stuttgart. All rights reserved.
|
|
* Copyright (c) 2004-2005 The Regents of the University of California.
|
|
* All rights reserved.
|
|
* Copyright (c) 2007-2010 Cisco Systems, Inc. All rights reserved.
|
|
* Copyright (c) 2008 Voltaire. All rights reserved
|
|
*
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#include "opal_config.h"
|
|
|
|
#ifdef HAVE_STRING_H
|
|
#include <string.h>
|
|
#endif
|
|
|
|
#include "opal/util/argv.h"
|
|
#include "opal/constants.h"
|
|
#include "opal/mca/paffinity/paffinity.h"
|
|
#include "opal/mca/paffinity/base/base.h"
|
|
#include "opal/runtime/opal.h"
|
|
#include "opal/util/output.h"
|
|
|
|
static bool diag_requested;
|
|
|
|
static int opal_paffinity_base_socket_to_cpu_set(char **socket_list, int socket_cnt,
|
|
long rank, bool logical_map,
|
|
opal_paffinity_base_cpu_set_t *cpumask)
|
|
{
|
|
int i;
|
|
char **range;
|
|
int range_cnt;
|
|
int lower_range, upper_range;
|
|
int processor_id, num_processors;
|
|
int rc;
|
|
int phys_processor;
|
|
|
|
/* get the number of LOGICAL processors on this node */
|
|
if (OPAL_SUCCESS != (rc = opal_paffinity_base_get_processor_info(&num_processors))) {
|
|
return rc;
|
|
}
|
|
OPAL_PAFFINITY_CPU_ZERO(*cpumask);
|
|
for (i=0; i<socket_cnt; i++) {
|
|
if (0 == strcmp("*", socket_list[i])) {
|
|
/* bind to all available logical processors - first set the bits in the cpu mask */
|
|
for ( processor_id=0; processor_id<=num_processors; processor_id++) {
|
|
if (0 > (phys_processor = opal_paffinity_base_get_physical_processor_id(processor_id))) {
|
|
return phys_processor;
|
|
}
|
|
OPAL_PAFFINITY_CPU_SET(phys_processor, *cpumask);
|
|
/* output diagnostic if requested */
|
|
if (diag_requested) {
|
|
opal_output(0, "paffinity slot assignment: rank %ld runs on physical cpu #%d (#%d)",
|
|
rank, phys_processor, processor_id);
|
|
}
|
|
}
|
|
/* tell paffinity to bind us */
|
|
if (OPAL_SUCCESS != ( rc = opal_paffinity_base_set(*cpumask))) {
|
|
return rc;
|
|
}
|
|
continue;
|
|
} /* end if * */
|
|
|
|
range = opal_argv_split(socket_list[i],'-');
|
|
range_cnt = opal_argv_count(range);
|
|
switch (range_cnt) {
|
|
case 1: /* no - was present, so just one processor given */
|
|
processor_id = atoi(range[0]);
|
|
if (logical_map) {
|
|
/* need to convert this to physical processor id */
|
|
if (0 > (phys_processor = opal_paffinity_base_get_physical_processor_id(processor_id))) {
|
|
return phys_processor;
|
|
}
|
|
} else {
|
|
phys_processor = processor_id;
|
|
}
|
|
/* set the bit for this physical processor */
|
|
OPAL_PAFFINITY_CPU_SET(phys_processor, *cpumask);
|
|
/* tell paffinity to bind us */
|
|
if (OPAL_SUCCESS != ( rc = opal_paffinity_base_set(*cpumask))) {
|
|
return rc;
|
|
}
|
|
/* output diagnostic if requested */
|
|
if (diag_requested) {
|
|
opal_output(0, "paffinity slot assignment: rank %ld runs on cpu #%d (#%d)",
|
|
rank, phys_processor, processor_id);
|
|
}
|
|
break;
|
|
|
|
case 2: /* range of processor id's was given */
|
|
lower_range = atoi(range[0]);
|
|
upper_range = atoi(range[1]);
|
|
if (num_processors < (upper_range - lower_range) ||
|
|
upper_range <= lower_range) {
|
|
return OPAL_ERR_SLOT_LIST_RANGE;
|
|
}
|
|
|
|
for (processor_id=lower_range; processor_id<=upper_range; processor_id++) {
|
|
if (logical_map) {
|
|
/* need to convert this to physical processor id */
|
|
if (0 > (phys_processor = opal_paffinity_base_get_physical_processor_id(processor_id))) {
|
|
return phys_processor;
|
|
}
|
|
} else {
|
|
phys_processor = processor_id;
|
|
}
|
|
/* set the bit for this physical processor */
|
|
OPAL_PAFFINITY_CPU_SET(phys_processor, *cpumask);
|
|
/* output diagnostic if requested */
|
|
if (diag_requested) {
|
|
opal_output(0, "paffinity slot assignment: rank %ld runs on cpu #%d (#%d)",
|
|
rank, phys_processor, processor_id);
|
|
}
|
|
}
|
|
/* tell paffinity to bind us */
|
|
if (OPAL_SUCCESS != (rc = opal_paffinity_base_set(*cpumask))) {
|
|
return rc;
|
|
}
|
|
break;
|
|
default:
|
|
opal_argv_free(range);
|
|
return OPAL_ERROR;
|
|
}
|
|
opal_argv_free(range);
|
|
}
|
|
return OPAL_SUCCESS;
|
|
}
|
|
|
|
static int opal_paffinity_base_socket_core_to_cpu_set(char **socket_core_list, int socket_core_list_cnt,
|
|
long rank, bool logical_map,
|
|
opal_paffinity_base_cpu_set_t *cpumask)
|
|
{
|
|
int rc, i;
|
|
char **socket_core;
|
|
int socket_core_cnt;
|
|
char **range;
|
|
int range_cnt;
|
|
int lower_range, upper_range;
|
|
int socket, core;
|
|
int num_sockets, num_cores;
|
|
int phys_socket, phys_core, phys_processor;
|
|
|
|
socket_core = opal_argv_split (socket_core_list[0], ':');
|
|
socket_core_cnt = opal_argv_count(socket_core);
|
|
OPAL_PAFFINITY_CPU_ZERO(*cpumask);
|
|
socket = atoi(socket_core[0]);
|
|
core = atoi(socket_core[1]);
|
|
|
|
/* get the number of LOGICAL sockets on this node */
|
|
if ( OPAL_SUCCESS != ( rc = opal_paffinity_base_get_socket_info(&num_sockets))) {
|
|
return rc;
|
|
}
|
|
|
|
if (logical_map) {
|
|
/* need to convert provided socket to a PHYSICAL socket id */
|
|
phys_socket = opal_paffinity_base_get_physical_socket_id(socket);
|
|
if (0 > phys_socket) {
|
|
return phys_socket;
|
|
}
|
|
} else {
|
|
phys_socket = socket;
|
|
}
|
|
phys_core = opal_paffinity_base_get_physical_core_id(phys_socket, core);
|
|
|
|
/* get the LOGICAL core info for this socket */
|
|
if ( OPAL_SUCCESS != ( rc = opal_paffinity_base_get_core_info(phys_socket, &num_cores))) {
|
|
return rc;
|
|
}
|
|
|
|
if (0 == strcmp("*",socket_core[1])) {
|
|
/* bind to all available LOGICAL cores */
|
|
for (core = 0; core < num_cores; core++) {
|
|
/* convert to PHYSICAL core id */
|
|
if (0 > (phys_core = opal_paffinity_base_get_physical_core_id(phys_socket, core))) {
|
|
return phys_core;
|
|
}
|
|
/* get the PHYSICAL processor id for the PHYSICAL socket/core */
|
|
if ( OPAL_SUCCESS != (rc = opal_paffinity_base_get_map_to_processor_id (phys_socket, phys_core, &phys_processor))) {
|
|
return rc;
|
|
}
|
|
/* set the bit for this processor */
|
|
OPAL_PAFFINITY_CPU_SET(phys_processor, *cpumask);
|
|
}
|
|
/* tell paffinity to bind us */
|
|
if (OPAL_SUCCESS != (rc = opal_paffinity_base_set(*cpumask))) {
|
|
return rc;
|
|
}
|
|
/* output diagnostic if requested */
|
|
if (diag_requested) {
|
|
opal_output(0, "paffinity slot assignment: rank %ld runs on physical processor #%d ( %d : %d)",
|
|
rank, phys_processor, socket, core);
|
|
}
|
|
} else {
|
|
range = opal_argv_split(socket_core[1], '-');
|
|
range_cnt = opal_argv_count(range);
|
|
switch (range_cnt) {
|
|
case 1: /* only one core specified */
|
|
core = atoi(range[0]);
|
|
if (logical_map) {
|
|
/* convert to physical core */
|
|
if (0 > (phys_core = opal_paffinity_base_get_physical_core_id(phys_socket, core))) {
|
|
return phys_core;
|
|
}
|
|
} else {
|
|
phys_core = core;
|
|
}
|
|
/* get the PHYSICAL processor id for this PHYSICAL socket/core */
|
|
if ( OPAL_SUCCESS != (rc = opal_paffinity_base_get_map_to_processor_id (phys_socket, phys_core, &phys_processor))) {
|
|
return rc;
|
|
}
|
|
/* set the bit for this processor */
|
|
OPAL_PAFFINITY_CPU_SET(phys_processor, *cpumask);
|
|
/* tell paffinity to bind us */
|
|
if (OPAL_SUCCESS != (rc = opal_paffinity_base_set(*cpumask))) {
|
|
return rc;
|
|
}
|
|
/* output diagnostic if requested */
|
|
if (diag_requested) {
|
|
opal_output(0, "paffinity slot assignment: rank %ld runs on physical cpu #%d ( %d[%d] : %d[%d])",
|
|
rank, phys_processor, phys_socket, socket, phys_core, core);
|
|
}
|
|
break;
|
|
|
|
case 2: /* range of core id's was given */
|
|
lower_range = atoi(range[0]);
|
|
upper_range = atoi(range[1]);
|
|
if ( 0 > lower_range || num_cores < (upper_range - lower_range) || lower_range >= upper_range ) {
|
|
return OPAL_ERR_SLOT_LIST_RANGE;
|
|
}
|
|
for (core=lower_range; core<=upper_range; core++) {
|
|
if (logical_map) {
|
|
/* convert to physical core */
|
|
if (0 > (phys_core = opal_paffinity_base_get_physical_core_id(phys_socket, core))) {
|
|
opal_output(0, "Rank %ld: PAFFINITY cannot get physical core id for logical core %ld in physical socket %ld (%ld)",
|
|
rank, (long)core, (long)phys_socket, (long)socket);
|
|
return phys_core;
|
|
}
|
|
} else {
|
|
phys_core = core;
|
|
}
|
|
/* get the PHYSICAL processor id for this PHYSICAL socket/core */
|
|
if ( OPAL_SUCCESS != (rc = opal_paffinity_base_get_map_to_processor_id (phys_socket, phys_core, &phys_processor))) {
|
|
return rc;
|
|
}
|
|
/* set the bit for this processor */
|
|
OPAL_PAFFINITY_CPU_SET(phys_processor, *cpumask);
|
|
/* output diagnostic if requested */
|
|
if (diag_requested) {
|
|
opal_output(0,"paffinity slot assignment: rank %ld runs on cpu #%d ( %d[%d] : %d[%d])",
|
|
rank, phys_processor, phys_socket, socket, phys_core, core);
|
|
}
|
|
}
|
|
/* tell paffinity to bind us */
|
|
if ( OPAL_SUCCESS != (rc = opal_paffinity_base_set(*cpumask))) {
|
|
return rc;
|
|
}
|
|
break;
|
|
|
|
default:
|
|
opal_argv_free(range);
|
|
opal_argv_free(socket_core);
|
|
return OPAL_ERROR;
|
|
}
|
|
opal_argv_free(range);
|
|
opal_argv_free(socket_core);
|
|
}
|
|
|
|
for (i=1; i<socket_core_list_cnt; i++) {
|
|
socket_core = opal_argv_split (socket_core_list[i], ':');
|
|
socket_core_cnt = opal_argv_count(socket_core);
|
|
switch (socket_core_cnt) {
|
|
case 1:
|
|
/* no colon => these cores are on the same socket as the last one specified,
|
|
* so we map them on that same physical socket
|
|
*/
|
|
range = opal_argv_split(socket_core[0], '-');
|
|
range_cnt = opal_argv_count(range);
|
|
switch (range_cnt) {
|
|
case 1: /* only one core provided */
|
|
core = atoi(range[0]);
|
|
if (logical_map) {
|
|
/* convert to physical core */
|
|
if (0 > (phys_core = opal_paffinity_base_get_physical_core_id(phys_socket, core))) {
|
|
return phys_core;
|
|
}
|
|
} else {
|
|
phys_core = core;
|
|
}
|
|
/* get the PHYSICAL processor id for this PHYSICAL socket/core */
|
|
if ( OPAL_SUCCESS != (rc = opal_paffinity_base_get_map_to_processor_id (phys_socket, phys_core, &phys_processor))) {
|
|
return rc;
|
|
}
|
|
/* set the bit for this processor */
|
|
OPAL_PAFFINITY_CPU_SET(phys_processor, *cpumask);
|
|
/* tell paffinity to bind us */
|
|
if (OPAL_SUCCESS != (rc = opal_paffinity_base_set(*cpumask))) {
|
|
return rc;
|
|
}
|
|
/* output diagnostic if requested */
|
|
if (diag_requested) {
|
|
opal_output(0, "paffinity slot assignment: rank %ld runs on physical cpu #%d ( %d[%d] : %d[%d])",
|
|
rank, phys_processor, phys_socket, socket, phys_core, core);
|
|
}
|
|
break;
|
|
|
|
case 2: /* range of core id's was given */
|
|
lower_range = atoi(range[0]);
|
|
upper_range = atoi(range[1]);
|
|
if ( 0 > lower_range || num_cores < (upper_range - lower_range) || lower_range >= upper_range ) {
|
|
return OPAL_ERR_SLOT_LIST_RANGE;
|
|
}
|
|
for (core=lower_range; core<=upper_range; core++) {
|
|
if (logical_map) {
|
|
/* convert to physical core */
|
|
if (0 > (phys_core = opal_paffinity_base_get_physical_core_id(phys_socket, core))) {
|
|
return phys_core;
|
|
}
|
|
} else {
|
|
phys_core = core;
|
|
}
|
|
/* get the PHYSICAL processor id for this PHYSICAL socket/core */
|
|
if ( OPAL_SUCCESS != (rc = opal_paffinity_base_get_map_to_processor_id (phys_socket, phys_core, &phys_processor))) {
|
|
return rc;
|
|
}
|
|
/* set the bit for this processor */
|
|
OPAL_PAFFINITY_CPU_SET(phys_processor, *cpumask);
|
|
/* output diagnostic if requested */
|
|
if (diag_requested) {
|
|
opal_output(0, "paffinity slot assignment: rank %ld runs on physical cpu #%d ( %d[%d] : %d[%d])",
|
|
rank, phys_processor, phys_socket, socket, phys_core, core);
|
|
}
|
|
}
|
|
/* tell paffinity to bind us */
|
|
if ( OPAL_SUCCESS != (rc = opal_paffinity_base_set(*cpumask))) {
|
|
return rc;
|
|
}
|
|
break;
|
|
|
|
default:
|
|
opal_argv_free(range);
|
|
opal_argv_free(socket_core);
|
|
return OPAL_ERROR;
|
|
}
|
|
opal_argv_free(range);
|
|
break;
|
|
|
|
case 2: /* colon was given => refers to a new socket! */
|
|
socket = atoi(socket_core[0]);
|
|
if (logical_map) {
|
|
/* need to convert provided socket to a PHYSICAL socket id */
|
|
phys_socket = opal_paffinity_base_get_physical_socket_id(socket);
|
|
if (0 > phys_socket) {
|
|
return phys_socket;
|
|
}
|
|
} else {
|
|
phys_socket = socket;
|
|
}
|
|
|
|
/* get the LOGICAL core info for this socket */
|
|
if ( OPAL_SUCCESS != ( rc = opal_paffinity_base_get_core_info(phys_socket, &num_cores))) {
|
|
return rc;
|
|
}
|
|
|
|
if (0 == strcmp("*",socket_core[1])) {
|
|
/* bind to all available LOGICAL cores */
|
|
for (core = 0; core < num_cores; core++) {
|
|
/* convert to PHYSICAL core id */
|
|
if (0 > (phys_core = opal_paffinity_base_get_physical_core_id(phys_socket, core))) {
|
|
return phys_core;
|
|
}
|
|
/* get the PHYSICAL processor id for the PHYSICAL socket/core */
|
|
if ( OPAL_SUCCESS != (rc = opal_paffinity_base_get_map_to_processor_id (phys_socket, phys_core, &phys_processor))) {
|
|
return rc;
|
|
}
|
|
/* set the bit for this processor */
|
|
OPAL_PAFFINITY_CPU_SET(phys_processor, *cpumask);
|
|
}
|
|
/* tell paffinity to bind us */
|
|
if (OPAL_SUCCESS != (rc = opal_paffinity_base_set(*cpumask))) {
|
|
return rc;
|
|
}
|
|
/* output diagnostic if requested */
|
|
if (diag_requested) {
|
|
opal_output(0, "paffinity slot assignment: rank %ld runs on physical cpu #%d ( %d[%d] : %d[%d])",
|
|
rank, phys_processor, phys_socket, socket, phys_core, core);
|
|
}
|
|
} else {
|
|
range = opal_argv_split(socket_core[1], '-');
|
|
range_cnt = opal_argv_count(range);
|
|
socket = atoi(socket_core[0]);
|
|
switch (range_cnt) {
|
|
case 1: /* only one core specified */
|
|
core = atoi(range[0]);
|
|
if (logical_map) {
|
|
/* convert to physical core */
|
|
if (0 > (phys_core = opal_paffinity_base_get_physical_core_id(phys_socket, core))) {
|
|
return phys_core;
|
|
}
|
|
} else {
|
|
phys_core = core;
|
|
}
|
|
/* get the PHYSICAL processor id for this PHYSICAL socket/core */
|
|
if ( OPAL_SUCCESS != (rc = opal_paffinity_base_get_map_to_processor_id (phys_socket, phys_core, &phys_processor))) {
|
|
return rc;
|
|
}
|
|
/* set the bit for this processor */
|
|
OPAL_PAFFINITY_CPU_SET(phys_processor, *cpumask);
|
|
/* tell paffinity to bind us */
|
|
if (OPAL_SUCCESS != (rc = opal_paffinity_base_set(*cpumask))) {
|
|
return rc;
|
|
}
|
|
/* output diagnostic if requested */
|
|
if (diag_requested) {
|
|
opal_output(0, "paffinity slot assignment: rank %ld runs on physical cpu #%d ( %d[%d] : %d[%d])",
|
|
rank, phys_processor, phys_socket, socket, phys_core, core);
|
|
}
|
|
break;
|
|
|
|
case 2: /* range of core id's was given */
|
|
lower_range = atoi(range[0]);
|
|
upper_range = atoi(range[1]);
|
|
if ( 0 > lower_range || num_cores < (upper_range - lower_range) || lower_range >= upper_range ) {
|
|
return OPAL_ERR_SLOT_LIST_RANGE;
|
|
}
|
|
for (core=lower_range; core<=upper_range; core++) {
|
|
if (logical_map) {
|
|
/* convert to physical core */
|
|
if (0 > (phys_core = opal_paffinity_base_get_physical_core_id(phys_socket, core))) {
|
|
return phys_core;
|
|
}
|
|
} else {
|
|
phys_core = core;
|
|
}
|
|
/* get the PHYSICAL processor id for this PHYSICAL socket/core */
|
|
if ( OPAL_SUCCESS != (rc = opal_paffinity_base_get_map_to_processor_id (phys_socket, phys_core, &phys_processor))) {
|
|
return rc;
|
|
}
|
|
/* set the bit for this processor */
|
|
OPAL_PAFFINITY_CPU_SET(phys_processor, *cpumask);
|
|
/* output diagnostic if requested */
|
|
if (diag_requested) {
|
|
opal_output(0, "paffinity slot assignment: rank %ld runs on physical cpu #%d ( %d[%d] : %d[%d])",
|
|
rank, phys_processor, phys_socket, socket, phys_core, core);
|
|
}
|
|
}
|
|
/* tell paffinity to bind us */
|
|
if ( OPAL_SUCCESS != (rc = opal_paffinity_base_set(*cpumask))) {
|
|
return rc;
|
|
}
|
|
|
|
default:
|
|
opal_argv_free(range);
|
|
opal_argv_free(socket_core);
|
|
return OPAL_ERROR;
|
|
}
|
|
opal_argv_free(range);
|
|
}
|
|
break;
|
|
|
|
default:
|
|
opal_argv_free(socket_core);
|
|
return OPAL_ERROR;
|
|
}
|
|
opal_argv_free(socket_core);
|
|
}
|
|
return OPAL_SUCCESS;
|
|
}
|
|
|
|
int opal_paffinity_base_slot_list_set(long rank, char *slot_str, opal_paffinity_base_cpu_set_t *cpumask)
|
|
{
|
|
char **item;
|
|
char **socket_core;
|
|
int item_cnt, socket_core_cnt, rc;
|
|
bool logical_map;
|
|
|
|
if (NULL == slot_str){
|
|
return OPAL_ERR_BAD_PARAM;
|
|
}
|
|
|
|
/* if the slot string is empty, that is an error */
|
|
if (0 == strlen(slot_str)) {
|
|
return OPAL_ERR_BAD_PARAM;
|
|
}
|
|
|
|
/* check for diag request to avoid repeatedly doing so */
|
|
if (4 < opal_output_get_verbosity(opal_paffinity_base_output)) {
|
|
diag_requested = true;
|
|
} else {
|
|
diag_requested = false;
|
|
}
|
|
|
|
opal_output_verbose(5, opal_paffinity_base_output, "paffinity slot assignment: slot_list == %s", slot_str);
|
|
|
|
if ('P' == slot_str[0] || 'p' == slot_str[0]) {
|
|
/* user has specified physical mapping */
|
|
logical_map = false;
|
|
item = opal_argv_split (&slot_str[1], ',');
|
|
} else {
|
|
logical_map = true; /* default to logical mapping */
|
|
item = opal_argv_split (slot_str, ',');
|
|
}
|
|
|
|
item_cnt = opal_argv_count (item);
|
|
socket_core = opal_argv_split (item[0], ':');
|
|
socket_core_cnt = opal_argv_count(socket_core);
|
|
opal_argv_free(socket_core);
|
|
switch (socket_core_cnt) {
|
|
case 1: /* binding to cpu's */
|
|
if (OPAL_SUCCESS != (rc = opal_paffinity_base_socket_to_cpu_set(item, item_cnt,
|
|
rank, logical_map, cpumask))) {
|
|
opal_argv_free(item);
|
|
return rc;
|
|
}
|
|
break;
|
|
case 2: /* binding to socket/core specification */
|
|
if (OPAL_SUCCESS != (rc = opal_paffinity_base_socket_core_to_cpu_set(item, item_cnt,
|
|
rank, logical_map, cpumask))) {
|
|
opal_argv_free(item);
|
|
return rc;
|
|
}
|
|
break;
|
|
default:
|
|
opal_argv_free(item);
|
|
return OPAL_ERROR;
|
|
}
|
|
opal_argv_free(item);
|
|
return OPAL_SUCCESS;
|
|
}
|
|
|
|
|
|
/**
|
|
* Make a prettyprint string for a cset.
|
|
*/
|
|
int opal_paffinity_base_cset2str(char *str, int len,
|
|
opal_paffinity_base_cpu_set_t *cset)
|
|
{
|
|
int ret, i, j, k, num_sockets, num_cores, flag, count,
|
|
range_first, range_last;
|
|
char tmp[BUFSIZ];
|
|
const int stmp = sizeof(tmp) - 1;
|
|
|
|
str[0] = tmp[stmp] = '\0';
|
|
|
|
/* Loop over the number of sockets in this machine */
|
|
ret = opal_paffinity_base_get_socket_info(&num_sockets);
|
|
if (OPAL_SUCCESS != ret) {
|
|
return ret;
|
|
}
|
|
for (i = 0; i < num_sockets; ++i) {
|
|
/* Loop over the number of cores in this socket */
|
|
ret = opal_paffinity_base_get_core_info(i, &num_cores);
|
|
if (OPAL_SUCCESS != ret) {
|
|
return ret;
|
|
}
|
|
/* Must initially set range_last to a low number -- smaller
|
|
than -1, so that the comparisons below work out
|
|
properly. */
|
|
for (range_last = -5, count = j = 0; j < num_cores; ++j) {
|
|
ret = opal_paffinity_base_get_map_to_processor_id(i, j, &k);
|
|
if (OPAL_SUCCESS != ret) {
|
|
return ret;
|
|
}
|
|
|
|
/* Prettyprint the cores that we're actually bound to */
|
|
flag = OPAL_PAFFINITY_CPU_ISSET(k, *cset);
|
|
if (flag) {
|
|
if (0 == count) {
|
|
snprintf(tmp, stmp, "socket %d[core %d", i, j);
|
|
strncat(str, tmp, len - strlen(str));
|
|
range_first = range_last = j;
|
|
} else {
|
|
if (j - 1 == range_last) {
|
|
range_last = j;
|
|
} else {
|
|
snprintf(tmp, stmp, "-%d,%d", range_last, j);
|
|
strncat(str, tmp, len - strlen(str));
|
|
range_first = range_last = j;
|
|
}
|
|
}
|
|
++count;
|
|
}
|
|
}
|
|
if (count > 0) {
|
|
if (range_first != range_last) {
|
|
snprintf(tmp, stmp, "-%d", range_last);
|
|
strncat(str, tmp, len - strlen(str));
|
|
}
|
|
strncat(str, "] ", len - strlen(str));
|
|
}
|
|
}
|
|
|
|
return OPAL_SUCCESS;
|
|
}
|
|
|
|
|