示例#1
0
int MPI_Reduce_scatter(const void* sbuff, void* rbuff, const int rcnts[], 
                       MPI_Datatype dt, MPI_Op op, MPI_Comm comm)
{
    int g2g = 1;
    char call[COLLCHK_SM_STRLEN];

    sprintf(call, "REDUCE_SCATTER");

    /* Check if init has been called */
    g2g = CollChk_is_init();

    if(g2g) {
        /* check for call consistancy */
        CollChk_same_call(comm, call);
        /* check for MPI_IN_PLACE consistency if needed */
        CollChk_check_buff(comm, sbuff, call);
        /* check for same operation */
        CollChk_same_op(comm, op, call);

        /* check for same datatypes */
        CollChk_dtype_bcast(comm, dt, 1, 0, call);

        /* make the call */
        return PMPI_Reduce_scatter(sbuff, rbuff, rcnts, dt, op, comm);
    }
    else {
        /* init not called */
        return CollChk_err_han("MPI_Init() has not been called!",
                               COLLCHK_ERR_NOT_INIT, call, comm);
    }
}
示例#2
0
文件: allgatherv.c 项目: pkestene/mpe
int MPI_Allgatherv( const void* sbuff, int scnt, MPI_Datatype stype,
                    void* rbuff, const int *rcnts, const int *displs, MPI_Datatype rtype,
                    MPI_Comm comm )
{
    char            call[COLLCHK_SM_STRLEN];
    int             g2g = 1, r;
    int             are2buffs;

    sprintf( call, "ALLGATHERV" );

    /* Check if init has been called */
    g2g = CollChk_is_init();

    if( g2g ) {
        MPI_Comm_rank(comm, &r);

        /* check for call consistency */
        CollChk_same_call( comm, call );
        /* check MPI_IN_PLACE consistency */
        CollChk_check_buff( comm, sbuff, call );

        /* check data signature consistency */
#if defined( HAVE_MPI_IN_PLACE )
        are2buffs = ( sbuff != MPI_IN_PLACE );
#else
        are2buffs = 1;
#endif
        CollChk_dtype_allgatherv(comm, stype, scnt, rtype, rcnts,
                                 are2buffs, call);

        /* make the call */
        return PMPI_Allgatherv( sbuff, scnt, stype,
                                rbuff, rcnts, displs, rtype,
                                comm );
    }
    else {
        /* init not called */
        return CollChk_err_han( "MPI_Init() has not been called!",
                                COLLCHK_ERR_NOT_INIT, call, comm );
    }
}