b68d66bb9b
This commit adds support for project_framework_component_* parameter matching. This is the first step in allowing the same framework name in multiple projects. This change also bumps the MCA component version to 2.1.0. All master frameworks have been updated to use the new component versioning macro. An mca.h has been added to each project to add a project specific versioning macro of the form PROJECT_MCA_VERSION_2_1_0. Signed-off-by: Nathan Hjelm <hjelmn@me.com>
102 строки
3.2 KiB
C
102 строки
3.2 KiB
C
/*
|
|
* Copyright (c) 2014 The University of Tennessee and The University
|
|
* of Tennessee Research Foundation. All rights
|
|
* reserved.
|
|
* Copyright (c) 2014 NVIDIA Corporation. All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#ifndef MCA_COLL_CUDA_EXPORT_H
|
|
#define MCA_COLL_CUDA_EXPORT_H
|
|
|
|
#include "ompi_config.h"
|
|
|
|
#include "mpi.h"
|
|
|
|
#include "opal/class/opal_object.h"
|
|
#include "ompi/mca/mca.h"
|
|
|
|
#include "ompi/constants.h"
|
|
#include "ompi/mca/coll/coll.h"
|
|
#include "ompi/mca/coll/base/base.h"
|
|
#include "ompi/communicator/communicator.h"
|
|
|
|
BEGIN_C_DECLS
|
|
|
|
/* API functions */
|
|
|
|
int mca_coll_cuda_init_query(bool enable_progress_threads,
|
|
bool enable_mpi_threads);
|
|
mca_coll_base_module_t
|
|
*mca_coll_cuda_comm_query(struct ompi_communicator_t *comm,
|
|
int *priority);
|
|
|
|
int mca_coll_cuda_module_enable(mca_coll_base_module_t *module,
|
|
struct ompi_communicator_t *comm);
|
|
|
|
int
|
|
mca_coll_cuda_allreduce(void *sbuf, void *rbuf, int count,
|
|
struct ompi_datatype_t *dtype,
|
|
struct ompi_op_t *op,
|
|
struct ompi_communicator_t *comm,
|
|
mca_coll_base_module_t *module);
|
|
|
|
int mca_coll_cuda_reduce(void *sbuf, void *rbuf, int count,
|
|
struct ompi_datatype_t *dtype,
|
|
struct ompi_op_t *op,
|
|
int root,
|
|
struct ompi_communicator_t *comm,
|
|
mca_coll_base_module_t *module);
|
|
|
|
int mca_coll_cuda_exscan(void *sbuf, void *rbuf, int count,
|
|
struct ompi_datatype_t *dtype,
|
|
struct ompi_op_t *op,
|
|
struct ompi_communicator_t *comm,
|
|
mca_coll_base_module_t *module);
|
|
|
|
int mca_coll_cuda_scan(void *sbuf, void *rbuf, int count,
|
|
struct ompi_datatype_t *dtype,
|
|
struct ompi_op_t *op,
|
|
struct ompi_communicator_t *comm,
|
|
mca_coll_base_module_t *module);
|
|
|
|
int
|
|
mca_coll_cuda_reduce_scatter_block(void *sbuf, void *rbuf, int rcount,
|
|
struct ompi_datatype_t *dtype,
|
|
struct ompi_op_t *op,
|
|
struct ompi_communicator_t *comm,
|
|
mca_coll_base_module_t *module);
|
|
|
|
/* Types */
|
|
/* Module */
|
|
|
|
typedef struct mca_coll_cuda_module_t {
|
|
mca_coll_base_module_t super;
|
|
|
|
/* Pointers to all the "real" collective functions */
|
|
mca_coll_base_comm_coll_t c_coll;
|
|
} mca_coll_cuda_module_t;
|
|
|
|
OBJ_CLASS_DECLARATION(mca_coll_cuda_module_t);
|
|
|
|
/* Component */
|
|
|
|
typedef struct mca_coll_cuda_component_t {
|
|
mca_coll_base_component_2_0_0_t super;
|
|
|
|
int priority; /* Priority of this component */
|
|
int disable_cuda_coll; /* Force disable of the CUDA collective component */
|
|
} mca_coll_cuda_component_t;
|
|
|
|
/* Globally exported variables */
|
|
|
|
OMPI_MODULE_DECLSPEC extern mca_coll_cuda_component_t mca_coll_cuda_component;
|
|
|
|
END_C_DECLS
|
|
|
|
#endif /* MCA_COLL_CUDA_EXPORT_H */
|