openmpi/ompi/mca/coll/demo/coll_demo_component.c

/*
 * Copyright (c) 2004-2007 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) 2008      Cisco Systems, Inc.  All rights reserved.
 * $COPYRIGHT$
 * 
 * Additional copyrights may follow
 * 
 * $HEADER$
 *
 * These symbols are in a file by themselves to provide nice linker
 * semantics.  Since linkers generally pull in symbols by object
 * files, keeping these symbols as the only symbols in this file
 * prevents utility programs such as "ompi_info" from having to import
 * entire components just to query their version and parameters.
 */

#include "ompi_config.h"

#include "mpi.h"
#include "ompi/mca/coll/coll.h"
#include "coll_demo.h"

/*
 * Public string showing the coll ompi_demo component version number
 */
const char *mca_coll_demo_component_version_string =
  "OMPI/MPI demo collective MCA component version " OMPI_VERSION;

/*
 * Global variable
 */
int mca_coll_demo_priority_param = -1;
int mca_coll_demo_verbose_param = -1;
int mca_coll_demo_verbose = 0;

/*
 * Local function
 */
static int demo_open(void);


/*
 * Instantiate the public struct with all of our public information
 * and pointers to our public functions in it
 */

const mca_coll_base_component_2_0_0_t mca_coll_demo_component = {

    /* First, the mca_component_t struct containing meta information
       about the component itself */

    {
        MCA_COLL_BASE_VERSION_2_0_0,

        /* Component name and version */
        "demo",
        OMPI_MAJOR_VERSION,
        OMPI_MINOR_VERSION,
        OMPI_RELEASE_VERSION,

        /* Component open and close functions */
        demo_open,
        NULL
    },
    {
        /* The component is checkpoint ready */
        MCA_BASE_METADATA_PARAM_CHECKPOINT
    },
    
    /* Initialization / querying functions */
    
    mca_coll_demo_init_query,
    mca_coll_demo_comm_query
};


static int demo_open(void)
{
    mca_coll_demo_priority_param = 
      mca_base_param_reg_int(&mca_coll_demo_component.collm_version,
                             "priority", NULL, false, false, 20, NULL);
    mca_coll_demo_verbose_param = 
      mca_base_param_reg_int(&mca_coll_demo_component.collm_version,
                             "verbose", NULL, false, false,
                             mca_coll_demo_verbose, NULL);

    return OMPI_SUCCESS;
}


static void
mca_coll_demo_module_construct(mca_coll_demo_module_t *module)
{
    memset(&module->underlying, 0, sizeof(mca_coll_base_comm_coll_t));
}

#define RELEASE(module, func)                                           \
    do {                                                                \
        if (NULL != module->underlying.coll_ ## func ## _module) { \
            OBJ_RELEASE(module->underlying.coll_ ## func ## _module); \
        }                                                               \
    } while (0)

static void
mca_coll_demo_module_destruct(mca_coll_demo_module_t *module)
{
    RELEASE(module, allgather); 
    RELEASE(module, allgatherv); 
    RELEASE(module, allreduce); 
    RELEASE(module, alltoall); 
    RELEASE(module, alltoallv); 
    RELEASE(module, alltoallw); 
    RELEASE(module, barrier); 
    RELEASE(module, bcast); 
    RELEASE(module, exscan); 
    RELEASE(module, gather); 
    RELEASE(module, gatherv); 
    RELEASE(module, reduce); 
    RELEASE(module, reduce_scatter); 
    RELEASE(module, scan); 
    RELEASE(module, scatter); 
    RELEASE(module, scatterv); 
}


OBJ_CLASS_INSTANCE(mca_coll_demo_module_t,
                   mca_coll_base_module_t,
                   mca_coll_demo_module_construct,
                   mca_coll_demo_module_destruct);