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>
198 строки
7.7 KiB
C
198 строки
7.7 KiB
C
/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil -*- */
|
|
/*
|
|
* Copyright (c) 2013 Mellanox Technologies, Inc.
|
|
* All rights reserved.
|
|
* Copyright (c) 2015 Los Alamos National Security, LLC. All rights
|
|
* reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
/**
|
|
* @file
|
|
*
|
|
* Collective Communication Interface
|
|
*
|
|
*/
|
|
|
|
#ifndef OSHMEM_MCA_SCOLL_H
|
|
#define OSHMEM_MCA_SCOLL_H
|
|
|
|
#include "oshmem_config.h"
|
|
#include "oshmem/types.h"
|
|
#include "oshmem/constants.h"
|
|
|
|
#include "opal/util/output.h"
|
|
#include "mpi.h"
|
|
#include "oshmem/mca/mca.h"
|
|
#include "opal/mca/base/base.h"
|
|
|
|
BEGIN_C_DECLS
|
|
|
|
/* ******************************************************************** */
|
|
|
|
struct oshmem_group_t;
|
|
struct oshmem_op_t;
|
|
|
|
/* ******************************************************************** */
|
|
|
|
typedef int (*mca_scoll_base_component_init_fn_t)(bool enable_progress_threads,
|
|
bool enable_threads);
|
|
|
|
typedef struct mca_scoll_base_module_1_0_0_t* (*mca_scoll_base_component_query_fn_t)(struct oshmem_group_t *group,
|
|
int *priority);
|
|
|
|
/* ******************************************************************** */
|
|
|
|
/**
|
|
* Collective component interface
|
|
*
|
|
* Component interface for the collective framework. A public
|
|
* instance of this structure, called
|
|
* mca_scoll_[component_name]_component, must exist in any collective
|
|
* component.
|
|
*/
|
|
struct mca_scoll_base_component_1_0_0_t {
|
|
/** Base component description */
|
|
mca_base_component_t scoll_version;
|
|
/** Base component data block */
|
|
mca_base_component_data_t scoll_data;
|
|
|
|
/** Component initialization function */
|
|
mca_scoll_base_component_init_fn_t scoll_init;
|
|
mca_scoll_base_component_query_fn_t scoll_query;
|
|
};
|
|
typedef struct mca_scoll_base_component_1_0_0_t mca_scoll_base_component_1_0_0_t;
|
|
|
|
/** Per guidence in mca.h, use the unversioned struct name if you just
|
|
want to always keep up with the most recent version of the
|
|
interace. */
|
|
typedef struct mca_scoll_base_component_1_0_0_t mca_scoll_base_component_t;
|
|
|
|
/**
|
|
* Collective module interface
|
|
*
|
|
* Module interface to the Collective framework. Modules are
|
|
* reference counted based on the number of functions from the module
|
|
* used on the commuicator. There is at most one module per component
|
|
* on a given communicator, and there can be many component modules on
|
|
* a given communicator.
|
|
*
|
|
* @note The collective framework and the
|
|
* communicator functionality only stores a pointer to the module
|
|
* function, so the component is free to create a structure that
|
|
* inherits from this one for use as the module structure.
|
|
*/
|
|
typedef int
|
|
(*mca_scoll_base_module_enable_1_0_0_fn_t)(struct mca_scoll_base_module_1_0_0_t* module,
|
|
struct oshmem_group_t *comm);
|
|
typedef int (*mca_scoll_base_module_ft_event_fn_t)(int state);
|
|
|
|
#define SCOLL_DEFAULT_ALG (-1)
|
|
|
|
#define SCOLL_ALG_BARRIER_CENTRAL_COUNTER 0
|
|
#define SCOLL_ALG_BARRIER_TOURNAMENT 1
|
|
#define SCOLL_ALG_BARRIER_RECURSIVE_DOUBLING 2
|
|
#define SCOLL_ALG_BARRIER_DISSEMINATION 3
|
|
#define SCOLL_ALG_BARRIER_BASIC 4
|
|
#define SCOLL_ALG_BARRIER_ADAPTIVE 5
|
|
|
|
#define SCOLL_ALG_BROADCAST_CENTRAL_COUNTER 0
|
|
#define SCOLL_ALG_BROADCAST_BINOMIAL 1
|
|
|
|
#define SCOLL_ALG_COLLECT_CENTRAL_COUNTER 0
|
|
#define SCOLL_ALG_COLLECT_TOURNAMENT 1
|
|
#define SCOLL_ALG_COLLECT_RECURSIVE_DOUBLING 2
|
|
#define SCOLL_ALG_COLLECT_RING 3
|
|
|
|
#define SCOLL_ALG_REDUCE_CENTRAL_COUNTER 0
|
|
#define SCOLL_ALG_REDUCE_TOURNAMENT 1
|
|
#define SCOLL_ALG_REDUCE_RECURSIVE_DOUBLING 2
|
|
#define SCOLL_ALG_REDUCE_LEGACY_LINEAR 3 /* Based linear algorithm from OMPI coll:basic */
|
|
#define SCOLL_ALG_REDUCE_LEGACY_LOG 4 /* Based log algorithm from OMPI coll:basic */
|
|
|
|
typedef int (*mca_scoll_base_module_barrier_fn_t)(struct oshmem_group_t *group,
|
|
long *pSync,
|
|
int alg);
|
|
typedef int (*mca_scoll_base_module_broadcast_fn_t)(struct oshmem_group_t *group,
|
|
int PE_root,
|
|
void *target,
|
|
const void *source,
|
|
size_t nlong,
|
|
long *pSync,
|
|
int alg);
|
|
typedef int (*mca_scoll_base_module_collect_fn_t)(struct oshmem_group_t *group,
|
|
void *target,
|
|
const void *source,
|
|
size_t nlong,
|
|
long *pSync,
|
|
bool nlong_type,
|
|
int alg);
|
|
typedef int (*mca_scoll_base_module_reduce_fn_t)(struct oshmem_group_t *group,
|
|
struct oshmem_op_t *op,
|
|
void *target,
|
|
const void *source,
|
|
size_t nlong,
|
|
long *pSync,
|
|
void *pWrk,
|
|
int alg);
|
|
|
|
struct mca_scoll_base_module_1_0_0_t {
|
|
/** Collective modules all inherit from opal_object */
|
|
opal_object_t super;
|
|
|
|
/* Collective function pointers */
|
|
mca_scoll_base_module_barrier_fn_t scoll_barrier;
|
|
mca_scoll_base_module_broadcast_fn_t scoll_broadcast;
|
|
mca_scoll_base_module_collect_fn_t scoll_collect;
|
|
mca_scoll_base_module_reduce_fn_t scoll_reduce;
|
|
mca_scoll_base_module_enable_1_0_0_fn_t scoll_module_enable;
|
|
};
|
|
typedef struct mca_scoll_base_module_1_0_0_t mca_scoll_base_module_1_0_0_t;
|
|
|
|
/** Per guidence in mca.h, use the unversioned struct name if you just
|
|
want to always keep up with the most recent version of the
|
|
interace. */
|
|
typedef struct mca_scoll_base_module_1_0_0_t mca_scoll_base_module_t;
|
|
OSHMEM_DECLSPEC OBJ_CLASS_DECLARATION(mca_scoll_base_module_t);
|
|
|
|
/* ******************************************************************** */
|
|
|
|
/*
|
|
* Macro for use in components that are of type coll
|
|
*/
|
|
#define MCA_SCOLL_BASE_VERSION_2_0_0 \
|
|
OSHMEM_MCA_BASE_VERSION_2_1_0("scoll", 1, 0, 0)
|
|
|
|
/* ******************************************************************** */
|
|
/*
|
|
* Collectives group cache structure
|
|
*
|
|
* Collectives gorup cache structure, used to find functions to
|
|
* implement collective algorithms and their associated modules.
|
|
*/
|
|
struct mca_scoll_base_group_scoll_t {
|
|
mca_scoll_base_module_barrier_fn_t scoll_barrier;
|
|
mca_scoll_base_module_1_0_0_t *scoll_barrier_module;
|
|
mca_scoll_base_module_broadcast_fn_t scoll_broadcast;
|
|
mca_scoll_base_module_1_0_0_t *scoll_broadcast_module;
|
|
mca_scoll_base_module_collect_fn_t scoll_collect;
|
|
mca_scoll_base_module_1_0_0_t *scoll_collect_module;
|
|
mca_scoll_base_module_reduce_fn_t scoll_reduce;
|
|
mca_scoll_base_module_1_0_0_t *scoll_reduce_module;
|
|
};
|
|
typedef struct mca_scoll_base_group_scoll_t mca_scoll_base_group_scoll_t;
|
|
|
|
#define PREVIOUS_SCOLL_FN(module, __api, group, ...) do { \
|
|
group->g_scoll.scoll_ ## __api ## _module = (mca_scoll_base_module_1_0_0_t*) module->previous_ ## __api ## _module; \
|
|
rc = module->previous_ ## __api (group, __VA_ARGS__); \
|
|
group->g_scoll.scoll_ ## __api ## _module = (mca_scoll_base_module_1_0_0_t*) module; \
|
|
} while(0)
|
|
|
|
END_C_DECLS
|
|
|
|
#endif /* OSHMEM_MCA_SCOLL_H */
|