e7ecd56bd2
such, the commit message back to the master SVN repository is fairly long. = ORTE Job-Level Output Messages = Add two new interfaces that should be used for all new code throughout the ORTE and OMPI layers (we already make the search-and-replace on the existing ORTE / OMPI layers): * orte_output(): (and corresponding friends ORTE_OUTPUT, orte_output_verbose, etc.) This function sends the output directly to the HNP for processing as part of a job-specific output channel. It supports all the same outputs as opal_output() (syslog, file, stdout, stderr), but for stdout/stderr, the output is sent to the HNP for processing and output. More on this below. * orte_show_help(): This function is a drop-in-replacement for opal_show_help(), with two differences in functionality: 1. the rendered text help message output is sent to the HNP for display (rather than outputting directly into the process' stderr stream) 1. the HNP detects duplicate help messages and does not display them (so that you don't see the same error message N times, once from each of your N MPI processes); instead, it counts "new" instances of the help message and displays a message every ~5 seconds when there are new ones ("I got X new copies of the help message...") opal_show_help and opal_output still exist, but they only output in the current process. The intent for the new orte_* functions is that they can apply job-level intelligence to the output. As such, we recommend that all new ORTE and OMPI code use the new orte_* functions, not thei opal_* functions. === New code === For ORTE and OMPI programmers, here's what you need to do differently in new code: * Do not include opal/util/show_help.h or opal/util/output.h. Instead, include orte/util/output.h (this one header file has declarations for both the orte_output() series of functions and orte_show_help()). * Effectively s/opal_output/orte_output/gi throughout your code. Note that orte_output_open() takes a slightly different argument list (as a way to pass data to the filtering stream -- see below), so you if explicitly call opal_output_open(), you'll need to slightly adapt to the new signature of orte_output_open(). * Literally s/opal_show_help/orte_show_help/. The function signature is identical. === Notes === * orte_output'ing to stream 0 will do similar to what opal_output'ing did, so leaving a hard-coded "0" as the first argument is safe. * For systems that do not use ORTE's RML or the HNP, the effect of orte_output_* and orte_show_help will be identical to their opal counterparts (the additional information passed to orte_output_open() will be lost!). Indeed, the orte_* functions simply become trivial wrappers to their opal_* counterparts. Note that we have not tested this; the code is simple but it is quite possible that we mucked something up. = Filter Framework = Messages sent view the new orte_* functions described above and messages output via the IOF on the HNP will now optionally be passed through a new "filter" framework before being output to stdout/stderr. The "filter" OPAL MCA framework is intended to allow preprocessing to messages before they are sent to their final destinations. The first component that was written in the filter framework was to create an XML stream, segregating all the messages into different XML tags, etc. This will allow 3rd party tools to read the stdout/stderr from the HNP and be able to know exactly what each text message is (e.g., a help message, another OMPI infrastructure message, stdout from the user process, stderr from the user process, etc.). Filtering is not active by default. Filter components must be specifically requested, such as: {{{ $ mpirun --mca filter xml ... }}} There can only be one filter component active. = New MCA Parameters = The new functionality described above introduces two new MCA parameters: * '''orte_base_help_aggregate''': Defaults to 1 (true), meaning that help messages will be aggregated, as described above. If set to 0, all help messages will be displayed, even if they are duplicates (i.e., the original behavior). * '''orte_base_show_output_recursions''': An MCA parameter to help debug one of the known issues, described below. It is likely that this MCA parameter will disappear before v1.3 final. = Known Issues = * The XML filter component is not complete. The current output from this component is preliminary and not real XML. A bit more work needs to be done to configure.m4 search for an appropriate XML library/link it in/use it at run time. * There are possible recursion loops in the orte_output() and orte_show_help() functions -- e.g., if RML send calls orte_output() or orte_show_help(). We have some ideas how to fix these, but figured that it was ok to commit before feature freeze with known issues. The code currently contains sub-optimal workarounds so that this will not be a problem, but it would be good to actually solve the problem rather than have hackish workarounds before v1.3 final. This commit was SVN r18434.
754 строки
26 KiB
C
754 строки
26 KiB
C
/*
|
|
* Copyright (c) 2007 Los Alamos National Security, LLC.
|
|
* All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#include "orte_config.h"
|
|
#include "orte/constants.h"
|
|
|
|
#include "orte/util/output.h"
|
|
#include "opal/threads/condition.h"
|
|
#include "opal/runtime/opal_progress.h"
|
|
#include "opal/dss/dss.h"
|
|
#include "opal/class/opal_hash_table.h"
|
|
#include "opal/util/bit_ops.h"
|
|
|
|
#include "orte/mca/errmgr/errmgr.h"
|
|
#include "orte/mca/grpcomm/grpcomm.h"
|
|
#include "orte/mca/rml/rml.h"
|
|
#include "orte/mca/odls/odls_types.h"
|
|
#include "orte/util/name_fns.h"
|
|
#include "orte/runtime/orte_globals.h"
|
|
#include "orte/runtime/orte_wait.h"
|
|
|
|
#include "orte/mca/rml/base/rml_contact.h"
|
|
|
|
#include "orte/mca/routed/base/base.h"
|
|
#include "routed_binomial.h"
|
|
|
|
static int init(void);
|
|
static int finalize(void);
|
|
static int update_route(orte_process_name_t *target,
|
|
orte_process_name_t *route);
|
|
static orte_process_name_t get_route(orte_process_name_t *target);
|
|
static int init_routes(orte_jobid_t job, opal_buffer_t *ndat);
|
|
static int route_lost(const orte_process_name_t *route);
|
|
static bool route_is_defined(const orte_process_name_t *target);
|
|
static int update_routing_tree(void);
|
|
static orte_vpid_t get_routing_tree(orte_jobid_t job, opal_list_t *children);
|
|
static int get_wireup_info(orte_jobid_t job, opal_buffer_t *buf);
|
|
|
|
#if OPAL_ENABLE_FT == 1
|
|
static int binomial_ft_event(int state);
|
|
#endif
|
|
|
|
static orte_process_name_t *lifeline=NULL;
|
|
static orte_process_name_t my_parent;
|
|
static int num_children;
|
|
static opal_list_t my_children;
|
|
|
|
orte_routed_module_t orte_routed_binomial_module = {
|
|
init,
|
|
finalize,
|
|
update_route,
|
|
get_route,
|
|
init_routes,
|
|
route_lost,
|
|
route_is_defined,
|
|
update_routing_tree,
|
|
get_routing_tree,
|
|
get_wireup_info,
|
|
#if OPAL_ENABLE_FT == 1
|
|
binomial_ft_event
|
|
#else
|
|
NULL
|
|
#endif
|
|
};
|
|
|
|
/* local globals */
|
|
static opal_hash_table_t peer_list;
|
|
static opal_hash_table_t vpid_wildcard_list;
|
|
static orte_process_name_t wildcard_route;
|
|
static opal_condition_t cond;
|
|
static opal_mutex_t lock;
|
|
|
|
|
|
static int init(void)
|
|
{
|
|
OBJ_CONSTRUCT(&peer_list, opal_hash_table_t);
|
|
opal_hash_table_init(&peer_list, 128);
|
|
|
|
OBJ_CONSTRUCT(&vpid_wildcard_list, opal_hash_table_t);
|
|
opal_hash_table_init(&vpid_wildcard_list, 128);
|
|
|
|
wildcard_route.jobid = ORTE_NAME_INVALID->jobid;
|
|
wildcard_route.vpid = ORTE_NAME_INVALID->vpid;
|
|
|
|
/* setup the global condition and lock */
|
|
OBJ_CONSTRUCT(&cond, opal_condition_t);
|
|
OBJ_CONSTRUCT(&lock, opal_mutex_t);
|
|
|
|
lifeline = NULL;
|
|
|
|
/* setup the list of children */
|
|
OBJ_CONSTRUCT(&my_children, opal_list_t);
|
|
num_children = 0;
|
|
my_parent.jobid = ORTE_PROC_MY_NAME->jobid;
|
|
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
static int finalize(void)
|
|
{
|
|
int rc;
|
|
uint64_t key;
|
|
void * value, *node, *next_node;
|
|
opal_list_item_t *item;
|
|
|
|
/* if I am an application process, indicate that I am
|
|
* truly finalizing prior to departure
|
|
*/
|
|
if (!orte_process_info.hnp &&
|
|
!orte_process_info.daemon &&
|
|
!orte_process_info.tool) {
|
|
if (ORTE_SUCCESS != (rc = orte_routed_base_register_sync(false))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return rc;
|
|
}
|
|
}
|
|
|
|
/* if I am the HNP, I need to stop the comm recv */
|
|
if (orte_process_info.hnp) {
|
|
orte_routed_base_comm_stop();
|
|
}
|
|
|
|
/* don't destruct the routes until *after* we send the
|
|
* sync as the oob will be asking us how to route
|
|
* the message!
|
|
*/
|
|
rc = opal_hash_table_get_first_key_uint64(&peer_list,
|
|
&key, &value, &node);
|
|
while(OPAL_SUCCESS == rc) {
|
|
if(NULL != value) {
|
|
free(value);
|
|
}
|
|
rc = opal_hash_table_get_next_key_uint64(&peer_list,
|
|
&key, &value, node, &next_node);
|
|
node = next_node;
|
|
}
|
|
OBJ_DESTRUCT(&peer_list);
|
|
OBJ_DESTRUCT(&vpid_wildcard_list);
|
|
/* destruct the global condition and lock */
|
|
OBJ_DESTRUCT(&cond);
|
|
OBJ_DESTRUCT(&lock);
|
|
|
|
lifeline = NULL;
|
|
|
|
/* deconstruct the list of children */
|
|
while (NULL != (item = opal_list_remove_first(&my_children))) {
|
|
OBJ_RELEASE(item);
|
|
}
|
|
OBJ_DESTRUCT(&my_children);
|
|
num_children = 0;
|
|
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
static int update_route(orte_process_name_t *target,
|
|
orte_process_name_t *route)
|
|
{
|
|
int rc;
|
|
orte_process_name_t * route_copy;
|
|
|
|
if (target->jobid == ORTE_JOBID_INVALID ||
|
|
target->vpid == ORTE_VPID_INVALID) {
|
|
return ORTE_ERR_BAD_PARAM;
|
|
}
|
|
|
|
ORTE_OUTPUT_VERBOSE((1, orte_routed_base_output,
|
|
"%s routed_binomial_update: %s --> %s",
|
|
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME),
|
|
ORTE_NAME_PRINT(target),
|
|
ORTE_NAME_PRINT(route)));
|
|
|
|
|
|
/* if I am an application process, we don't update the route unless
|
|
* the conditions dictate it. This is done to avoid creating large
|
|
* hash tables when they aren't needed
|
|
*/
|
|
if (!orte_process_info.hnp && !orte_process_info.daemon &&
|
|
!orte_process_info.tool) {
|
|
/* if the route is the daemon, then do nothing - we already route
|
|
* everything through the daemon anyway
|
|
*/
|
|
if (OPAL_EQUAL == orte_util_compare_name_fields(ORTE_NS_CMP_ALL, route,
|
|
ORTE_PROC_MY_DAEMON)) {
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
/* if this is for my own job family, then do nothing - we -always- route
|
|
* our own job family through the daemons
|
|
*/
|
|
if (ORTE_JOB_FAMILY(target->jobid) == ORTE_JOB_FAMILY(ORTE_PROC_MY_NAME->jobid)) {
|
|
return ORTE_SUCCESS;
|
|
}
|
|
}
|
|
|
|
route_copy = malloc(sizeof(orte_process_name_t));
|
|
*route_copy = *route;
|
|
/* exact match */
|
|
if (target->jobid != ORTE_JOBID_WILDCARD &&
|
|
target->vpid != ORTE_VPID_WILDCARD) {
|
|
rc = opal_hash_table_set_value_uint64(&peer_list,
|
|
orte_util_hash_name(target), route_copy);
|
|
if (ORTE_SUCCESS != rc) {
|
|
ORTE_ERROR_LOG(rc);
|
|
}
|
|
return rc;
|
|
}
|
|
|
|
/* vpid wildcard */
|
|
if (target->jobid != ORTE_JOBID_WILDCARD &&
|
|
target->vpid == ORTE_VPID_WILDCARD) {
|
|
rc = opal_hash_table_set_value_uint32(&vpid_wildcard_list,
|
|
target->jobid, route_copy);
|
|
if (ORTE_SUCCESS != rc) {
|
|
ORTE_ERROR_LOG(rc);
|
|
}
|
|
return rc;
|
|
}
|
|
free(route_copy);
|
|
|
|
return ORTE_ERR_NOT_SUPPORTED;
|
|
}
|
|
|
|
|
|
static orte_process_name_t get_route(orte_process_name_t *target)
|
|
{
|
|
orte_process_name_t *ret;
|
|
int rc;
|
|
|
|
/* if it is me, then the route is just direct */
|
|
if (OPAL_EQUAL == opal_dss.compare(ORTE_PROC_MY_NAME, target, ORTE_NAME)) {
|
|
ret = target;
|
|
goto found;
|
|
}
|
|
|
|
/* check exact matches */
|
|
rc = opal_hash_table_get_value_uint64(&peer_list,
|
|
orte_util_hash_name(target), (void**)&ret);
|
|
if (ORTE_SUCCESS == rc) {
|
|
/* got a good result - return it */
|
|
goto found;
|
|
}
|
|
|
|
/* didn't find an exact match - check to see if a route for this job was defined */
|
|
rc = opal_hash_table_get_value_uint32(&vpid_wildcard_list,
|
|
target->jobid, (void**)&ret);
|
|
if (ORTE_SUCCESS == rc) {
|
|
/* got a good result - return it */
|
|
goto found;
|
|
}
|
|
|
|
/* default to wildcard route */
|
|
ret = &wildcard_route;
|
|
|
|
found:
|
|
|
|
ORTE_OUTPUT_VERBOSE((2, orte_routed_base_output,
|
|
"%s routed_binomial_get(%s) --> %s",
|
|
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME),
|
|
ORTE_NAME_PRINT(target),
|
|
ORTE_NAME_PRINT(ret)));
|
|
|
|
return *ret;
|
|
}
|
|
|
|
static int process_callback(orte_jobid_t job, opal_buffer_t *buffer)
|
|
{
|
|
orte_proc_t **procs;
|
|
orte_job_t *jdata;
|
|
orte_std_cntr_t cnt;
|
|
char *rml_uri;
|
|
orte_process_name_t name;
|
|
int rc;
|
|
|
|
/* lookup the job object for this process */
|
|
if (NULL == (jdata = orte_get_job_data_object(job))) {
|
|
ORTE_ERROR_LOG(ORTE_ERR_NOT_FOUND);
|
|
return ORTE_ERR_NOT_FOUND;
|
|
}
|
|
procs = (orte_proc_t**)jdata->procs->addr;
|
|
|
|
/* unpack the data for each entry */
|
|
cnt = 1;
|
|
while (ORTE_SUCCESS == (rc = opal_dss.unpack(buffer, &rml_uri, &cnt, OPAL_STRING))) {
|
|
|
|
ORTE_OUTPUT_VERBOSE((2, orte_routed_base_output,
|
|
"%s routed_binomial:callback got uri %s",
|
|
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME),
|
|
(NULL == rml_uri) ? "NULL" : rml_uri));
|
|
|
|
if (rml_uri == NULL) continue;
|
|
|
|
/* we don't need to set the contact info into our rml
|
|
* hash table as we won't talk to the proc directly
|
|
*/
|
|
|
|
/* extract the proc's name */
|
|
if (ORTE_SUCCESS != (rc = orte_rml_base_parse_uris(rml_uri, &name, NULL))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
free(rml_uri);
|
|
continue;
|
|
}
|
|
/* the procs are stored in vpid order, so update the record */
|
|
procs[name.vpid]->rml_uri = strdup(rml_uri);
|
|
free(rml_uri);
|
|
|
|
/* update the proc state */
|
|
if (procs[name.vpid]->state < ORTE_PROC_STATE_RUNNING) {
|
|
procs[name.vpid]->state = ORTE_PROC_STATE_RUNNING;
|
|
}
|
|
|
|
++jdata->num_reported;
|
|
cnt = 1;
|
|
}
|
|
if (ORTE_ERR_UNPACK_READ_PAST_END_OF_BUFFER != rc) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return rc;
|
|
}
|
|
|
|
/* if all procs have reported, update our job state */
|
|
if (jdata->num_reported == jdata->num_procs) {
|
|
/* update the job state */
|
|
if (jdata->state < ORTE_JOB_STATE_RUNNING) {
|
|
jdata->state = ORTE_JOB_STATE_RUNNING;
|
|
}
|
|
}
|
|
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
static int init_routes(orte_jobid_t job, opal_buffer_t *ndat)
|
|
{
|
|
/* the binomial module routes all proc communications through
|
|
* the local daemon. Daemons must identify which of their
|
|
* daemon-peers is "hosting" the specified recipient and
|
|
* route the message to that daemon. Daemon contact info
|
|
* is handled elsewhere, so all we need to do here is
|
|
* ensure that the procs are told to route through their
|
|
* local daemon, and that daemons are told how to route
|
|
* for each proc
|
|
*/
|
|
int rc;
|
|
|
|
/* if I am a tool, then I stand alone - there is nothing to do */
|
|
if (orte_process_info.tool) {
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
/* if I am a daemon or HNP, then I have to extract the routing info for this job
|
|
* from the data sent to me for launch and update the routing tables to
|
|
* point at the daemon for each proc
|
|
*/
|
|
if (orte_process_info.daemon) {
|
|
|
|
ORTE_OUTPUT_VERBOSE((1, orte_routed_base_output,
|
|
"%s routed_binomial: init routes for daemon job %s\n\thnp_uri %s",
|
|
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME),
|
|
ORTE_JOBID_PRINT(job),
|
|
(NULL == orte_process_info.my_hnp_uri) ? "NULL" : orte_process_info.my_hnp_uri));
|
|
|
|
if (NULL == ndat) {
|
|
/* indicates this is being called during orte_init.
|
|
* Get the HNP's name for possible later use
|
|
*/
|
|
if (NULL == orte_process_info.my_hnp_uri) {
|
|
/* fatal error */
|
|
ORTE_ERROR_LOG(ORTE_ERR_FATAL);
|
|
return ORTE_ERR_FATAL;
|
|
}
|
|
/* set the contact info into the hash table */
|
|
if (ORTE_SUCCESS != (rc = orte_rml.set_contact_info(orte_process_info.my_hnp_uri))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return(rc);
|
|
}
|
|
|
|
/* extract the hnp name and store it */
|
|
if (ORTE_SUCCESS != (rc = orte_rml_base_parse_uris(orte_process_info.my_hnp_uri,
|
|
ORTE_PROC_MY_HNP, NULL))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return rc;
|
|
}
|
|
|
|
/* if ndat is NULL, then this is being called during init,
|
|
* so just seed the routing table with a path back to the HNP...
|
|
*/
|
|
if (ORTE_SUCCESS != (rc = update_route(ORTE_PROC_MY_HNP, ORTE_PROC_MY_HNP))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return rc;
|
|
}
|
|
/* set the wildcard route for anybody whose name we don't recognize
|
|
* to be the HNP
|
|
*/
|
|
wildcard_route.jobid = ORTE_PROC_MY_HNP->jobid;
|
|
wildcard_route.vpid = ORTE_PROC_MY_HNP->vpid;
|
|
|
|
/* set our lifeline to the the HNP - we will abort if that connection is lost */
|
|
lifeline = ORTE_PROC_MY_HNP;
|
|
|
|
/* daemons will send their contact info back to the HNP as
|
|
* part of the message confirming they are read to go. HNP's
|
|
* load their contact info during orte_init
|
|
*/
|
|
} else {
|
|
/* ndat != NULL means we are getting an update of RML info
|
|
* for the daemons - so update our contact info and routes
|
|
*/
|
|
if (ORTE_SUCCESS != (rc = orte_rml_base_update_contact_info(ndat))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
}
|
|
return rc;
|
|
}
|
|
|
|
ORTE_OUTPUT_VERBOSE((2, orte_routed_base_output,
|
|
"%s routed_binomial: completed init routes",
|
|
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME)));
|
|
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
|
|
if (orte_process_info.hnp) {
|
|
|
|
ORTE_OUTPUT_VERBOSE((1, orte_routed_base_output,
|
|
"%s routed_binomial: init routes for HNP job %s",
|
|
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME),
|
|
ORTE_JOBID_PRINT(job)));
|
|
|
|
if (NULL == ndat) {
|
|
/* if ndat is NULL, then this is being called during init, so just
|
|
* make myself available to catch any reported contact info
|
|
*/
|
|
if (ORTE_SUCCESS != (rc = orte_routed_base_comm_start())) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return rc;
|
|
}
|
|
/* the HNP has no lifeline */
|
|
lifeline = NULL;
|
|
} else {
|
|
/* if this is for my own jobid, then I am getting an update of RML info
|
|
* for the daemons - so update our contact info and routes
|
|
*/
|
|
if (ORTE_PROC_MY_NAME->jobid == job) {
|
|
if (ORTE_SUCCESS != (rc = orte_rml_base_update_contact_info(ndat))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return rc;
|
|
}
|
|
} else {
|
|
/* if not, then I need to process the callback */
|
|
if (ORTE_SUCCESS != (rc = process_callback(job, ndat))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return rc;
|
|
}
|
|
}
|
|
}
|
|
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
{ /* MUST BE A PROC */
|
|
/* if ndat != NULL, then this is being invoked by the proc to
|
|
* init a route to a specified process that is outside of our
|
|
* job family. We want that route to go through our HNP, routed via
|
|
* out local daemon - however, we cannot know for
|
|
* certain that the HNP already knows how to talk to the specified
|
|
* procs. For example, in OMPI's publish/subscribe procedures, the
|
|
* DPM framework looks for an mca param containing the global ompi-server's
|
|
* uri. This info will come here so the proc can setup a route to
|
|
* the server - we need to pass the routing info to our HNP
|
|
*/
|
|
if (NULL != ndat) {
|
|
int rc;
|
|
|
|
ORTE_OUTPUT_VERBOSE((1, orte_routed_base_output,
|
|
"%s routed_binomial: init routes w/non-NULL data",
|
|
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME)));
|
|
|
|
/* send the buffer to the proper tag on the daemon */
|
|
if (0 > (rc = orte_rml.send_buffer(ORTE_PROC_MY_HNP, ndat,
|
|
ORTE_RML_TAG_RML_INFO_UPDATE, 0))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return rc;
|
|
}
|
|
/* we already have defined our routes to everyone to
|
|
* be through the local daemon, so nothing further to do
|
|
*/
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
/* if ndat=NULL, then we are being called during orte_init. In this
|
|
* case, we need to setup a few critical pieces of info
|
|
*/
|
|
|
|
ORTE_OUTPUT_VERBOSE((1, orte_routed_base_output,
|
|
"%s routed_binomial: init routes for proc job %s\n\thnp_uri %s\n\tdaemon uri %s",
|
|
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME), ORTE_JOBID_PRINT(job),
|
|
(NULL == orte_process_info.my_hnp_uri) ? "NULL" : orte_process_info.my_hnp_uri,
|
|
(NULL == orte_process_info.my_daemon_uri) ? "NULL" : orte_process_info.my_daemon_uri));
|
|
|
|
if (NULL == orte_process_info.my_daemon_uri) {
|
|
/* in this module, we absolutely MUST have this information - if
|
|
* we didn't get it, then error out
|
|
*/
|
|
orte_output(0, "%s ERROR: Failed to identify the local daemon's URI",
|
|
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME));
|
|
orte_output(0, "%s ERROR: This is a fatal condition when the binomial router",
|
|
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME));
|
|
orte_output(0, "%s ERROR: has been selected - either select the unity router",
|
|
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME));
|
|
orte_output(0, "%s ERROR: or ensure that the local daemon info is provided",
|
|
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME));
|
|
return ORTE_ERR_FATAL;
|
|
}
|
|
|
|
/* we have to set the HNP's name, even though we won't route messages directly
|
|
* to it. This is required to ensure that we -do- send messages to the correct
|
|
* HNP name
|
|
*/
|
|
if (ORTE_SUCCESS != (rc = orte_rml_base_parse_uris(orte_process_info.my_hnp_uri,
|
|
ORTE_PROC_MY_HNP, NULL))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return rc;
|
|
}
|
|
|
|
/* Set the contact info in the RML - this won't actually establish
|
|
* the connection, but just tells the RML how to reach the daemon
|
|
* if/when we attempt to send to it
|
|
*/
|
|
if (ORTE_SUCCESS != (rc = orte_rml.set_contact_info(orte_process_info.my_daemon_uri))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return(rc);
|
|
}
|
|
/* extract the daemon's name so we can update the routing table */
|
|
if (ORTE_SUCCESS != (rc = orte_rml_base_parse_uris(orte_process_info.my_daemon_uri,
|
|
ORTE_PROC_MY_DAEMON, NULL))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return rc;
|
|
}
|
|
|
|
/* setup the route to all other procs to flow through the daemon */
|
|
wildcard_route.jobid = ORTE_PROC_MY_DAEMON->jobid;
|
|
wildcard_route.vpid = ORTE_PROC_MY_DAEMON->vpid;
|
|
|
|
/* set our lifeline to the local daemon - we will abort if this connection is lost */
|
|
lifeline = ORTE_PROC_MY_DAEMON;
|
|
|
|
/* register ourselves -this sends a message to the daemon (warming up that connection)
|
|
* and sends our contact info to the HNP when all local procs have reported
|
|
*
|
|
* NOTE: it may seem odd that we send our contact info to the HNP - after all,
|
|
* the HNP doesn't really need to know how to talk to us directly if we are
|
|
* using this routing method. However, this is good for two reasons:
|
|
*
|
|
* (1) some debuggers and/or tools may need RML contact
|
|
* info to set themselves up
|
|
*
|
|
* (2) doing so allows the HNP to "block" in a dynamic launch
|
|
* until all procs are reported running, thus ensuring that no communication
|
|
* is attempted until the overall ORTE system knows how to talk to everyone -
|
|
* otherwise, the system can just hang.
|
|
*/
|
|
if (ORTE_SUCCESS != (rc = orte_routed_base_register_sync(true))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return rc;
|
|
}
|
|
/* no answer is expected or coming */
|
|
|
|
return ORTE_SUCCESS;
|
|
}
|
|
}
|
|
|
|
static int route_lost(const orte_process_name_t *route)
|
|
{
|
|
/* if we lose the connection to the lifeline and we are NOT already,
|
|
* in finalize, tell the OOB to abort.
|
|
* NOTE: we cannot call abort from here as the OOB needs to first
|
|
* release a thread-lock - otherwise, we will hang!!
|
|
*/
|
|
if (!orte_finalizing &&
|
|
NULL != lifeline &&
|
|
OPAL_EQUAL == orte_util_compare_name_fields(ORTE_NS_CMP_ALL, route, lifeline)) {
|
|
orte_output(0, "%s routed:binomial: Connection to lifeline %s lost",
|
|
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME),
|
|
ORTE_NAME_PRINT(lifeline));
|
|
return ORTE_ERR_FATAL;
|
|
}
|
|
|
|
/* we don't care about this one, so return success */
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
|
|
|
|
/******* stub functions - to be implemented ******/
|
|
static bool route_is_defined(const orte_process_name_t *target)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
/*************************************/
|
|
|
|
static int binomial_tree(int rank, int parent, int me, int num_procs)
|
|
{
|
|
int i, bitmap, peer, hibit, mask, found;
|
|
orte_namelist_t *child;
|
|
|
|
/* is this me? */
|
|
if (me == rank) {
|
|
bitmap = opal_cube_dim(num_procs);
|
|
|
|
hibit = opal_hibit(rank, bitmap);
|
|
--bitmap;
|
|
|
|
for (i = hibit + 1, mask = 1 << i; i <= bitmap; ++i, mask <<= 1) {
|
|
peer = rank | mask;
|
|
if (peer < num_procs) {
|
|
child = OBJ_NEW(orte_namelist_t);
|
|
child->name.jobid = ORTE_PROC_MY_NAME->jobid;
|
|
child->name.vpid = peer;
|
|
ORTE_OUTPUT_VERBOSE((3, orte_routed_base_output,
|
|
"%s routed:binomial found child %s",
|
|
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME),
|
|
ORTE_NAME_PRINT(&child->name)));
|
|
|
|
opal_list_append(&my_children, &child->item);
|
|
num_children++;
|
|
}
|
|
}
|
|
ORTE_OUTPUT_VERBOSE((3, orte_routed_base_output,
|
|
"%s routed:binomial found parent %d",
|
|
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME),
|
|
parent));
|
|
return parent;
|
|
}
|
|
|
|
/* find the children of this rank */
|
|
bitmap = opal_cube_dim(num_procs);
|
|
|
|
hibit = opal_hibit(rank, bitmap);
|
|
--bitmap;
|
|
|
|
for (i = hibit + 1, mask = 1 << i; i <= bitmap; ++i, mask <<= 1) {
|
|
peer = rank | mask;
|
|
if (peer < num_procs) {
|
|
/* execute compute on this child */
|
|
if (0 <= (found = binomial_tree(peer, rank, me, num_procs))) {
|
|
return found;
|
|
}
|
|
}
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
static int update_routing_tree(void)
|
|
{
|
|
opal_list_item_t *item;
|
|
|
|
/* clear the list of children if any are already present */
|
|
while (NULL != (item = opal_list_remove_first(&my_children))) {
|
|
OBJ_RELEASE(item);
|
|
}
|
|
num_children = 0;
|
|
|
|
/* recompute the tree */
|
|
my_parent.vpid = binomial_tree(0, 0, ORTE_PROC_MY_NAME->vpid,
|
|
orte_process_info.num_procs);
|
|
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
static orte_vpid_t get_routing_tree(orte_jobid_t job,
|
|
opal_list_t *children)
|
|
{
|
|
opal_list_item_t *item;
|
|
orte_namelist_t *nm, *child;
|
|
|
|
/* the binomial routing tree always goes to our children,
|
|
* for any job
|
|
*/
|
|
if (NULL != children) {
|
|
for (item = opal_list_get_first(&my_children);
|
|
item != opal_list_get_end(&my_children);
|
|
item = opal_list_get_next(item)) {
|
|
child = (orte_namelist_t*)item;
|
|
nm = OBJ_NEW(orte_namelist_t);
|
|
nm->name.jobid = child->name.jobid;
|
|
nm->name.vpid = child->name.vpid;
|
|
opal_list_append(children, &nm->item);
|
|
}
|
|
}
|
|
/* return my parent's vpid */
|
|
return my_parent.vpid;
|
|
}
|
|
|
|
|
|
static int get_wireup_info(orte_jobid_t job, opal_buffer_t *buf)
|
|
{
|
|
int rc;
|
|
|
|
/* if we are not using static ports, then we need to share the
|
|
* comm info - otherwise, just return
|
|
*/
|
|
if (orte_static_ports) {
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
if (ORTE_SUCCESS != (rc = orte_rml_base_get_contact_info(ORTE_PROC_MY_NAME->jobid, buf))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
OBJ_RELEASE(buf);
|
|
return rc;
|
|
}
|
|
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
#if OPAL_ENABLE_FT == 1
|
|
static int binomial_ft_event(int state)
|
|
{
|
|
int ret, exit_status = ORTE_SUCCESS;
|
|
|
|
/******** Checkpoint Prep ********/
|
|
if(OPAL_CRS_CHECKPOINT == state) {
|
|
}
|
|
/******** Continue Recovery ********/
|
|
else if (OPAL_CRS_CONTINUE == state ) {
|
|
}
|
|
/******** Restart Recovery ********/
|
|
else if (OPAL_CRS_RESTART == state ) {
|
|
/*
|
|
* Re-exchange the routes
|
|
*/
|
|
if (ORTE_SUCCESS != (ret = orte_routed.init_routes(ORTE_PROC_MY_NAME->jobid, NULL))) {
|
|
exit_status = ret;
|
|
goto cleanup;
|
|
}
|
|
}
|
|
else if (OPAL_CRS_TERM == state ) {
|
|
/* Nothing */
|
|
}
|
|
else {
|
|
/* Error state = Nothing */
|
|
}
|
|
|
|
cleanup:
|
|
return exit_status;
|
|
}
|
|
#endif
|
|
|