9db4542c2b
maffinity_base_bind_failure_action MCA params to the hwloc base (hwloc_base_alloc_polocy and hwloc_base_bind_failure_action). Since these MCA parameters were never on a release branch, I'm just moving/renaming them outright and not leaving aliases to the old names. Note that some upper layer needs to call opal_hwloc_base_set_process_membind_policy() to set the set-by-MCA-param process-wide memory affinity policy. We can't do this automatically during hwloc_base_open() because, for reasons described elsewhere, opal_hwloc_topology is not automatically filled during hwloc_base_open() (in short: potential scalability issues when launching many MPI processes simultaneously on a single machine, for example). This commit was SVN r25156.
65 строки
1.6 KiB
C
65 строки
1.6 KiB
C
/*
|
|
* Copyright (c) 2011 Cisco Systems, Inc. All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
|
|
#include "opal_config.h"
|
|
|
|
#include "opal/constants.h"
|
|
|
|
#include "opal/mca/hwloc/hwloc.h"
|
|
#include "opal/mca/hwloc/base/base.h"
|
|
|
|
|
|
/*
|
|
* Don't use show_help() here (or print any error message at all).
|
|
* Let the upper layer output a relevant message, because doing so may
|
|
* be complicated (e.g., this might be called from the ORTE ODLS,
|
|
* which has to do some extra steps to get error messages to be
|
|
* displayed).
|
|
*/
|
|
int opal_hwloc_base_set_process_membind_policy(void)
|
|
{
|
|
int rc = 0, flags;
|
|
hwloc_membind_policy_t policy;
|
|
hwloc_cpuset_t cpuset;
|
|
|
|
/* Make sure opal_hwloc_topology has been set by the time we've
|
|
been called */
|
|
if (NULL == opal_hwloc_topology) {
|
|
return OPAL_ERR_BAD_PARAM;
|
|
}
|
|
|
|
/* Set the default memory allocation policy according to MCA
|
|
param */
|
|
switch (opal_hwloc_base_map) {
|
|
case OPAL_HWLOC_BASE_MAP_LOCAL_ONLY:
|
|
policy = HWLOC_MEMBIND_BIND;
|
|
flags = HWLOC_MEMBIND_STRICT;
|
|
break;
|
|
|
|
case OPAL_HWLOC_BASE_MAP_NONE:
|
|
default:
|
|
policy = HWLOC_MEMBIND_DEFAULT;
|
|
flags = 0;
|
|
break;
|
|
}
|
|
|
|
cpuset = hwloc_bitmap_alloc();
|
|
if (NULL == cpuset) {
|
|
rc = OPAL_ERR_OUT_OF_RESOURCE;
|
|
} else {
|
|
hwloc_get_cpubind(opal_hwloc_topology, cpuset, 0);
|
|
rc = hwloc_set_membind(opal_hwloc_topology,
|
|
cpuset, HWLOC_MEMBIND_BIND, flags);
|
|
hwloc_bitmap_free(cpuset);
|
|
}
|
|
|
|
return (0 == rc) ? OPAL_SUCCESS : OPAL_ERROR;
|
|
}
|