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>
113 строки
3.2 KiB
C
113 строки
3.2 KiB
C
/**
|
|
Copyright (c) 2011 Mellanox Technologies. All rights reserved.
|
|
$COPYRIGHT$
|
|
|
|
Additional copyrights may follow
|
|
|
|
$HEADER$
|
|
*/
|
|
|
|
#ifndef MCA_SCOLL_MPI_H
|
|
#define MCA_SCOLL_MPI_H
|
|
|
|
#include "oshmem_config.h"
|
|
|
|
#include "shmem.h"
|
|
#include "oshmem/mca/mca.h"
|
|
#include "oshmem/mca/scoll/scoll.h"
|
|
#include "oshmem/request/request.h"
|
|
#include "oshmem/util/oshmem_util.h"
|
|
#include "oshmem/proc/proc.h"
|
|
#include "ompi/communicator/communicator.h"
|
|
|
|
#include "orte/runtime/orte_globals.h"
|
|
|
|
#include "scoll_mpi_debug.h"
|
|
BEGIN_C_DECLS
|
|
|
|
|
|
/**
|
|
* Globally exported structure
|
|
*/
|
|
|
|
struct mca_scoll_mpi_component_t {
|
|
/** Base coll component */
|
|
mca_scoll_base_component_1_0_0_t super;
|
|
|
|
/** MCA parameter: Priority of this component */
|
|
int mpi_priority;
|
|
|
|
/** MCA parameter: Verbose level of this component */
|
|
int mpi_verbose;
|
|
|
|
/** MCA parameter: Enable MPI */
|
|
int mpi_enable;
|
|
|
|
/** MCA parameter: Minimal number of processes in the communicator
|
|
for the corresponding mpi context to be created */
|
|
int mpi_np;
|
|
};
|
|
typedef struct mca_scoll_mpi_component_t mca_scoll_mpi_component_t;
|
|
|
|
OMPI_MODULE_DECLSPEC extern mca_scoll_mpi_component_t mca_scoll_mpi_component;
|
|
|
|
|
|
/**
|
|
* MPI enabled communicator
|
|
*/
|
|
struct mca_scoll_mpi_module_t {
|
|
mca_scoll_base_module_t super;
|
|
|
|
ompi_communicator_t *comm;
|
|
int rank;
|
|
/* Saved handlers - for fallback */
|
|
mca_scoll_base_module_reduce_fn_t previous_reduce;
|
|
mca_scoll_base_module_t *previous_reduce_module;
|
|
mca_scoll_base_module_broadcast_fn_t previous_broadcast;
|
|
mca_scoll_base_module_t *previous_broadcast_module;
|
|
mca_scoll_base_module_barrier_fn_t previous_barrier;
|
|
mca_scoll_base_module_t *previous_barrier_module;
|
|
mca_scoll_base_module_collect_fn_t previous_collect;
|
|
mca_scoll_base_module_t *previous_collect_module;
|
|
};
|
|
typedef struct mca_scoll_mpi_module_t mca_scoll_mpi_module_t;
|
|
|
|
OBJ_CLASS_DECLARATION(mca_scoll_mpi_module_t);
|
|
|
|
|
|
/* API functions */
|
|
int mca_scoll_mpi_init_query(bool enable_progress_threads, bool enable_mpi_threads);
|
|
|
|
mca_scoll_base_module_t* mca_scoll_mpi_comm_query(oshmem_group_t *osh_group, int *priority);
|
|
|
|
int mca_scoll_mpi_barrier(struct oshmem_group_t *group, long *pSync, int alg);
|
|
|
|
int mca_scoll_mpi_broadcast(struct oshmem_group_t *group,
|
|
int PE_root,
|
|
void *target,
|
|
const void *source,
|
|
size_t nlong,
|
|
long *pSync,
|
|
int alg);
|
|
|
|
int mca_scoll_mpi_collect(struct oshmem_group_t *group,
|
|
void *target,
|
|
const void *source,
|
|
size_t nlong,
|
|
long *pSync,
|
|
bool nlong_type,
|
|
int alg);
|
|
|
|
int mca_scoll_mpi_reduce(struct oshmem_group_t *group,
|
|
struct oshmem_op_t *op,
|
|
void *target,
|
|
const void *source,
|
|
size_t nlong,
|
|
long *pSync,
|
|
void *pWrk,
|
|
int alg);
|
|
|
|
END_C_DECLS
|
|
|
|
#endif
|