145 строки
4.2 KiB
C
145 строки
4.2 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) 2011-2012 Los Alamos National Security, LLC.
|
|
* All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
|
|
#include "orte_config.h"
|
|
#include "orte/constants.h"
|
|
|
|
#include "opal/mca/mca.h"
|
|
#include "opal/util/output.h"
|
|
#include "opal/mca/base/base.h"
|
|
#include "opal/mca/base/mca_base_param.h"
|
|
|
|
#include "orte/mca/grpcomm/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/grpcomm/base/static-components.h"
|
|
|
|
/*
|
|
* Global variables
|
|
*/
|
|
orte_grpcomm_base_t orte_grpcomm_base;
|
|
|
|
orte_grpcomm_base_module_t orte_grpcomm = {0};
|
|
|
|
/**
|
|
* Function for finding and opening either all MCA components, or the one
|
|
* that was specifically requested via a MCA parameter.
|
|
*/
|
|
int orte_grpcomm_base_open(void)
|
|
{
|
|
/* Debugging / verbose output. Always have stream open, with
|
|
verbose set by the mca open system... */
|
|
orte_grpcomm_base.output = opal_output_open(NULL);
|
|
|
|
/* init globals */
|
|
OBJ_CONSTRUCT(&orte_grpcomm_base.active_colls, opal_list_t);
|
|
orte_grpcomm_base.coll_id = 0;
|
|
|
|
#if OPAL_HAVE_HWLOC
|
|
orte_grpcomm_base.working_cpuset = NULL;
|
|
#endif
|
|
|
|
/* Open up all available components */
|
|
|
|
if (ORTE_SUCCESS !=
|
|
mca_base_components_open("grpcomm", orte_grpcomm_base.output,
|
|
mca_grpcomm_base_static_components,
|
|
&orte_grpcomm_base.components_available, true)) {
|
|
return ORTE_ERROR;
|
|
}
|
|
|
|
/* All done */
|
|
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
orte_grpcomm_collective_t* orte_grpcomm_base_setup_collective(orte_grpcomm_coll_id_t id)
|
|
{
|
|
opal_list_item_t *item;
|
|
orte_grpcomm_collective_t *cptr, *coll;
|
|
|
|
coll = NULL;
|
|
for (item = opal_list_get_first(&orte_grpcomm_base.active_colls);
|
|
item != opal_list_get_end(&orte_grpcomm_base.active_colls);
|
|
item = opal_list_get_next(item)) {
|
|
cptr = (orte_grpcomm_collective_t*)item;
|
|
if (id == cptr->id) {
|
|
coll = cptr;
|
|
break;
|
|
}
|
|
}
|
|
if (NULL == coll) {
|
|
coll = OBJ_NEW(orte_grpcomm_collective_t);
|
|
coll->id = id;
|
|
opal_list_append(&orte_grpcomm_base.active_colls, &coll->super);
|
|
}
|
|
|
|
return coll;
|
|
}
|
|
|
|
/* local objects */
|
|
static void collective_constructor(orte_grpcomm_collective_t *ptr)
|
|
{
|
|
ptr->id = -1;
|
|
ptr->active = false;
|
|
ptr->num_local_recvd = 0;
|
|
OBJ_CONSTRUCT(&ptr->local_bucket, opal_buffer_t);
|
|
ptr->num_peer_buckets = 0;
|
|
ptr->num_global_recvd = 0;
|
|
ptr->locally_complete = false;
|
|
OBJ_CONSTRUCT(&ptr->participants, opal_list_t);
|
|
ptr->cbfunc = NULL;
|
|
ptr->cbdata = NULL;
|
|
OBJ_CONSTRUCT(&ptr->buffer, opal_buffer_t);
|
|
OBJ_CONSTRUCT(&ptr->targets, opal_list_t);
|
|
ptr->next_cb = NULL;
|
|
ptr->next_cbdata = NULL;
|
|
}
|
|
static void collective_destructor(orte_grpcomm_collective_t *ptr)
|
|
{
|
|
opal_list_item_t *item;
|
|
|
|
OBJ_DESTRUCT(&ptr->local_bucket);
|
|
while (NULL != (item = opal_list_remove_first(&ptr->participants))) {
|
|
OBJ_RELEASE(item);
|
|
}
|
|
OBJ_DESTRUCT(&ptr->participants);
|
|
OBJ_DESTRUCT(&ptr->buffer);
|
|
while (NULL != (item = opal_list_remove_first(&ptr->targets))) {
|
|
OBJ_RELEASE(item);
|
|
}
|
|
OBJ_DESTRUCT(&ptr->targets);
|
|
}
|
|
OBJ_CLASS_INSTANCE(orte_grpcomm_collective_t,
|
|
opal_list_item_t,
|
|
collective_constructor,
|
|
collective_destructor);
|
|
|
|
OBJ_CLASS_INSTANCE(orte_grpcomm_caddy_t,
|
|
opal_object_t,
|
|
NULL, NULL);
|