46 строки
978 B
C
46 строки
978 B
C
|
/*
|
||
|
|
* Copyright (c) 2007-2008 UT-Battelle, LLC
|
||
|
|
* $COPYRIGHT$
|
||
|
|
*
|
||
|
|
* Additional copyrights may follow
|
||
|
|
*
|
||
|
|
* $HEADER$
|
||
|
|
*/
|
||
|
|
/** @file */
|
||
|
|
|
||
|
|
#include "ompi_config.h"
|
||
|
|
|
||
|
|
#include "ompi/constants.h"
|
||
|
|
#include "coll_sm2.h"
|
||
|
|
#include "ompi/op/op.h"
|
||
|
|
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Shared memory blocking allreduce.
|
||
|
|
*/
|
||
|
|
int mca_coll_sm2_allreduce_intra(void *sbuf, void *rbuf, int count,
|
||
|
|
struct ompi_datatype_t *dtype,
|
||
|
|
struct ompi_op_t *op,
|
||
|
|
struct ompi_communicator_t *comm,
|
||
|
|
struct mca_coll_base_module_1_1_0_t *module)
|
||
|
|
{
|
||
|
|
/* local variables */
|
||
|
|
int ret;
|
||
|
|
char *sm_buffer;
|
||
|
|
mca_coll_sm2_module_t *sm_module;
|
||
|
|
|
||
|
|
sm_module=(mca_coll_sm2_module_t *) module;
|
||
|
|
|
||
|
|
/* get a pointer to the shared-memory working buffer */
|
||
|
|
|
||
|
|
if( 0 != (op->o_flags & OMPI_OP_FLAGS_COMMUTE)) {
|
||
|
|
/* Commutative Operation */
|
||
|
|
} else {
|
||
|
|
/* Non-Commutative Operation */
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
}
|