140 строки
4.1 KiB
C
140 строки
4.1 KiB
C
/*
|
|
* Copyright (c) 2004-2009 The Trustees of Indiana University and Indiana
|
|
* University Research and Technology
|
|
* Corporation. All rights reserved.
|
|
* Copyright (c) 2010 Cisco Systems, Inc. All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*
|
|
* Simple routine to expose three things to the MPI process:
|
|
*
|
|
* 1. What processor(s) Open MPI bound this process to
|
|
* 2. What processor(s) this process is bound to
|
|
* 3. What processor(s) exist on this host
|
|
*
|
|
* Note that 1 and 2 may be different!
|
|
*/
|
|
|
|
#include "ompi_config.h"
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
|
|
#include "opal/mca/paffinity/base/base.h"
|
|
|
|
#include "ompi/communicator/communicator.h"
|
|
#include "ompi/errhandler/errhandler.h"
|
|
#include "ompi/mpi/c/bindings.h"
|
|
#include "ompi/mpiext/affinity/mpiext_affinity_c.h"
|
|
|
|
static const char FUNC_NAME[] = "OMPI_Affinity";
|
|
|
|
static int fill_ompi_bound(char str[OMPI_AFFINITY_STRING_MAX]);
|
|
static int fill_current_binding(char str[OMPI_AFFINITY_STRING_MAX]);
|
|
static int fill_exists(char str[OMPI_AFFINITY_STRING_MAX]);
|
|
|
|
|
|
int OMPI_Affinity_str(char ompi_bound[OMPI_AFFINITY_STRING_MAX],
|
|
char current_binding[OMPI_AFFINITY_STRING_MAX],
|
|
char exists[OMPI_AFFINITY_STRING_MAX])
|
|
{
|
|
int ret;
|
|
|
|
memset(ompi_bound, 0, sizeof(ompi_bound));
|
|
memset(current_binding, 0, sizeof(current_binding));
|
|
|
|
if (OPAL_SUCCESS != (ret = fill_ompi_bound(ompi_bound)) ||
|
|
OPAL_SUCCESS != (ret = fill_current_binding(current_binding)) ||
|
|
OPAL_SUCCESS != (ret = fill_exists(exists))) {
|
|
return OMPI_ERRHANDLER_INVOKE(MPI_COMM_WORLD, ret, FUNC_NAME);
|
|
}
|
|
|
|
return MPI_SUCCESS;
|
|
}
|
|
|
|
static int fill_ompi_bound(char str[OMPI_AFFINITY_STRING_MAX])
|
|
{
|
|
int ret;
|
|
opal_paffinity_base_cpu_set_t cset;
|
|
|
|
/* If OMPI did not bind, indicate that */
|
|
if (!opal_paffinity_base_bound) {
|
|
const char tmp[] = "Open MPI did not bind this process";
|
|
strncpy(str, tmp, OMPI_AFFINITY_STRING_MAX - 1);
|
|
return OPAL_SUCCESS;
|
|
}
|
|
|
|
/* Find out what OMPI bound us to and prettyprint it */
|
|
ret =
|
|
opal_paffinity_base_parse_binding(opal_paffinity_base_applied_binding,
|
|
&cset);
|
|
if (OPAL_SUCCESS != ret) {
|
|
return ret;
|
|
}
|
|
|
|
return opal_paffinity_base_cset2str(str, OMPI_AFFINITY_STRING_MAX, &cset);
|
|
}
|
|
|
|
static int fill_current_binding(char str[OMPI_AFFINITY_STRING_MAX])
|
|
{
|
|
int ret, flag;
|
|
opal_paffinity_base_cpu_set_t cset;
|
|
|
|
/* Get our binding */
|
|
ret = opal_paffinity_base_get(&cset);
|
|
if (OPAL_SUCCESS != ret) {
|
|
return ret;
|
|
}
|
|
|
|
/* Are we bound anywhere? */
|
|
OPAL_PAFFINITY_PROCESS_IS_BOUND(cset, &flag);
|
|
if (!flag) {
|
|
const char tmp[] = "Not bound (or bound to all available processors)";
|
|
strncat(str, tmp, OMPI_AFFINITY_STRING_MAX - 1);
|
|
return OPAL_SUCCESS;
|
|
}
|
|
|
|
return opal_paffinity_base_cset2str(str, OMPI_AFFINITY_STRING_MAX, &cset);
|
|
}
|
|
|
|
|
|
/* Prettyprint a list of all available processors */
|
|
static int fill_exists(char str[OMPI_AFFINITY_STRING_MAX])
|
|
{
|
|
int ret, i, num_sockets, num_cores;
|
|
char tmp[BUFSIZ];
|
|
const int stmp = sizeof(tmp) - 1;
|
|
|
|
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) {
|
|
if (i > 0) {
|
|
strncat(str, ", ", OMPI_AFFINITY_STRING_MAX - strlen(str));
|
|
}
|
|
snprintf(tmp, stmp, "socket %d has ", i);
|
|
strncat(str, tmp, OMPI_AFFINITY_STRING_MAX - strlen(str));
|
|
|
|
/* 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;
|
|
}
|
|
if (1 == num_cores) {
|
|
strncat(str, "1 core", OMPI_AFFINITY_STRING_MAX - strlen(str));
|
|
} else {
|
|
snprintf(tmp, stmp, "%d cores", num_cores);
|
|
strncat(str, tmp, OMPI_AFFINITY_STRING_MAX - strlen(str));
|
|
}
|
|
}
|
|
|
|
return OPAL_SUCCESS;
|
|
}
|