1
1
openmpi/orte/mca/rmaps/base/rmaps_base_open.c
2006-10-03 17:40:00 +00:00

188 строки
7.8 KiB
C

/*
* Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
* University Research and Technology
* Corporation. All rights reserved.
* Copyright (c) 2004-2005 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) 2006 Cisco Systems, Inc. All rights reserved.
* $COPYRIGHT$
*
* Additional copyrights may follow
*
* $HEADER$
*/
#include "orte_config.h"
#include "orte/orte_constants.h"
#include "opal/mca/mca.h"
#include "opal/mca/base/base.h"
#include "opal/mca/base/mca_base_param.h"
#include "opal/util/output.h"
#include "orte/dss/dss.h"
#include "orte/mca/errmgr/errmgr.h"
#include "orte/mca/rmaps/base/rmaps_private.h"
#include "orte/mca/rmaps/base/base.h"
/*
* The following file was created by configure. It contains extern
* statements and the definition of an array of pointers to each
* component's public mca_base_component_t struct.
*/
#include "orte/mca/rmaps/base/static-components.h"
/*
* Global variables
*/
orte_rmaps_base_t orte_rmaps_base;
/*
* Declare the RMAPS module to hold the API function pointers
*/
orte_rmaps_base_module_t orte_rmaps = {
orte_rmaps_base_map,
orte_rmaps_base_finalize
};
orte_rmaps_base_module_t orte_rmaps_no_op = {
orte_rmaps_base_map_no_op,
orte_rmaps_base_finalize
};
/**
* Function for finding and opening either all MCA components, or the one
* that was specifically requested via a MCA parameter.
*/
int orte_rmaps_base_open(void)
{
int param, rc, value;
char *policy, *requested;
orte_data_type_t tmp;
/* Debugging / verbose output */
param = mca_base_param_reg_int_name("rmaps", "base_verbose",
"Verbosity level for the rmaps framework",
false, false, 0, &value);
if (value != 0) {
orte_rmaps_base.rmaps_output = opal_output_open(NULL);
} else {
orte_rmaps_base.rmaps_output = -1;
}
/* Are we scheduling by node or by slot? */
param = mca_base_param_reg_string_name("rmaps", "base_schedule_policy",
"Scheduling Policy for RMAPS. [slot | node]",
false, false, "slot", &policy);
if (0 == strcmp(policy, "node")) {
mca_base_param_set_string(param, "node");
}
/* Should we schedule on the local node or not? */
mca_base_param_reg_int_name("rmaps", "base_schedule_local",
"If nonzero, allow scheduling MPI applications on the same node as mpirun (default). If zero, do not schedule any MPI applications on the same node as mpirun",
false, false, 1, &value);
/* Should we oversubscribe or not? */
mca_base_param_reg_int_name("rmaps", "base_no_oversubscribe",
"If nonzero, then do not allow oversubscription of nodes - mpirun will return an error if there aren't enough nodes to launch all processes without oversubscribing",
false, false, 0, &value);
if ((int)false == value) {
orte_rmaps_base.oversubscribe = true; /** default condition that allows oversubscription */
} else {
orte_rmaps_base.oversubscribe = false;
}
/** register the base system types with the DSS */
tmp = ORTE_JOB_MAP;
if (ORTE_SUCCESS != (rc = orte_dss.register_type(orte_rmaps_base_pack_map,
orte_rmaps_base_unpack_map,
(orte_dss_copy_fn_t)orte_rmaps_base_copy_map,
(orte_dss_compare_fn_t)orte_rmaps_base_compare_map,
(orte_dss_size_fn_t)orte_rmaps_base_size_map,
(orte_dss_print_fn_t)orte_rmaps_base_print_map,
(orte_dss_release_fn_t)orte_rmaps_base_std_obj_release,
ORTE_DSS_STRUCTURED,
"ORTE_JOB_MAP", &tmp))) {
ORTE_ERROR_LOG(rc);
return rc;
}
tmp = ORTE_MAPPED_PROC;
if (ORTE_SUCCESS != (rc = orte_dss.register_type(orte_rmaps_base_pack_mapped_proc,
orte_rmaps_base_unpack_mapped_proc,
(orte_dss_copy_fn_t)orte_rmaps_base_copy_mapped_proc,
(orte_dss_compare_fn_t)orte_rmaps_base_compare_mapped_proc,
(orte_dss_size_fn_t)orte_rmaps_base_size_mapped_proc,
(orte_dss_print_fn_t)orte_rmaps_base_print_mapped_proc,
(orte_dss_release_fn_t)orte_rmaps_base_std_obj_release,
ORTE_DSS_STRUCTURED,
"ORTE_MAPPED_PROC", &tmp))) {
ORTE_ERROR_LOG(rc);
return rc;
}
tmp = ORTE_MAPPED_NODE;
if (ORTE_SUCCESS != (rc = orte_dss.register_type(orte_rmaps_base_pack_mapped_node,
orte_rmaps_base_unpack_mapped_node,
(orte_dss_copy_fn_t)orte_rmaps_base_copy_mapped_node,
(orte_dss_compare_fn_t)orte_rmaps_base_compare_mapped_node,
(orte_dss_size_fn_t)orte_rmaps_base_size_mapped_node,
(orte_dss_print_fn_t)orte_rmaps_base_print_mapped_node,
(orte_dss_release_fn_t)orte_rmaps_base_std_obj_release,
ORTE_DSS_STRUCTURED,
"ORTE_MAPPED_NODE", &tmp))) {
ORTE_ERROR_LOG(rc);
return rc;
}
/* Some systems do not want any RMAPS support. In those cases,
* memory consumption is also an issue. For those systems, we
* avoid opening the RMAPS components by checking for a directive
* to use the "null" component.
*/
param = mca_base_param_reg_string_name("rmaps", NULL, NULL,
false, false, NULL, NULL);
if (ORTE_ERROR == mca_base_param_lookup_string(param, &requested)) {
return ORTE_ERROR;
}
if (NULL != requested && 0 == strcmp(requested, "null")) {
/* the user has specifically requested that we use the "null"
* component. In this case, that means we do NOT open any
* components, and we simply use the default module we have
* already defined above
*/
orte_rmaps_base.no_op_selected = true;
orte_rmaps = orte_rmaps_no_op; /* use the no_op module */
return ORTE_SUCCESS;
}
orte_rmaps_base.no_op_selected = false;
/* Open up all the components that we can find */
if (ORTE_SUCCESS !=
mca_base_components_open("rmaps", orte_rmaps_base.rmaps_output,
mca_rmaps_base_static_components,
&orte_rmaps_base.rmaps_opened, true)) {
return ORTE_ERROR;
}
/* All done */
return ORTE_SUCCESS;
}