Blame test/mpi/coll/coll9.c

Packit Service c5cf8c
/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
Packit Service c5cf8c
/*
Packit Service c5cf8c
 *  (C) 2001 by Argonne National Laboratory.
Packit Service c5cf8c
 *      See COPYRIGHT in top-level directory.
Packit Service c5cf8c
 */
Packit Service c5cf8c
#include "mpi.h"
Packit Service c5cf8c
#include <stdio.h>
Packit Service c5cf8c
#include "mpitest.h"
Packit Service c5cf8c
Packit Service c5cf8c
void addem(int *, int *, int *, MPI_Datatype *);
Packit Service c5cf8c
Packit Service c5cf8c
void addem(int *invec, int *inoutvec, int *len, MPI_Datatype * dtype)
Packit Service c5cf8c
{
Packit Service c5cf8c
    int i;
Packit Service c5cf8c
    for (i = 0; i < *len; i++)
Packit Service c5cf8c
        inoutvec[i] += invec[i];
Packit Service c5cf8c
}
Packit Service c5cf8c
Packit Service c5cf8c
int main(int argc, char **argv)
Packit Service c5cf8c
{
Packit Service c5cf8c
    int rank, size, i;
Packit Service c5cf8c
    int data;
Packit Service c5cf8c
    int errors = 0;
Packit Service c5cf8c
    int result = -100;
Packit Service c5cf8c
    int correct_result;
Packit Service c5cf8c
    MPI_Op op;
Packit Service c5cf8c
Packit Service c5cf8c
    MTest_Init(&argc, &argv);
Packit Service c5cf8c
    MPI_Comm_rank(MPI_COMM_WORLD, &rank;;
Packit Service c5cf8c
    MPI_Comm_size(MPI_COMM_WORLD, &size);
Packit Service c5cf8c
Packit Service c5cf8c
    data = rank;
Packit Service c5cf8c
    MPI_Op_create((MPI_User_function *) addem, 1, &op);
Packit Service c5cf8c
    MPI_Reduce(&data, &result, 1, MPI_INT, op, 0, MPI_COMM_WORLD);
Packit Service c5cf8c
    MPI_Bcast(&result, 1, MPI_INT, 0, MPI_COMM_WORLD);
Packit Service c5cf8c
    MPI_Op_free(&op);
Packit Service c5cf8c
    correct_result = 0;
Packit Service c5cf8c
    for (i = 0; i < size; i++)
Packit Service c5cf8c
        correct_result += i;
Packit Service c5cf8c
    if (result != correct_result)
Packit Service c5cf8c
        errors++;
Packit Service c5cf8c
Packit Service c5cf8c
    MTest_Finalize(errors);
Packit Service c5cf8c
    return MTestReturnValue(errors);
Packit Service c5cf8c
}