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 |
}
|