1 /* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
3 * (C) 2001 by Argonne National Laboratory.
4 * See COPYRIGHT in top-level directory.
10 void addem(int *, int *, int *, MPI_Datatype *);
11 void assoc(int *, int *, int *, MPI_Datatype *);
13 void addem(int *invec, int *inoutvec, int *len, MPI_Datatype * dtype)
16 for (i = 0; i < *len; i++)
17 inoutvec[i] += invec[i];
20 #define BAD_ANSWER 100000
23 The operation is inoutvec[i] = invec[i] op inoutvec[i]
24 (see 4.9.4). The order is important.
26 Note that the computation is in process rank (in the communicator)
27 order, independent of the root.
29 void assoc(int *invec, int *inoutvec, int *len, MPI_Datatype * dtype)
32 for (i = 0; i < *len; i++) {
33 if (inoutvec[i] <= invec[i]) {
35 MPI_Comm_rank(MPI_COMM_WORLD, &rank);
36 fprintf(stderr, "[%d] inout[0] = %d, in[0] = %d\n", rank, inoutvec[0], invec[0]);
37 inoutvec[i] = BAD_ANSWER;
40 inoutvec[i] = invec[i];
44 int main(int argc, char **argv)
51 MPI_Op op_assoc, op_addem;
52 MPI_Comm comm = MPI_COMM_WORLD;
54 MPI_Init(&argc, &argv);
55 MPI_Op_create((MPI_User_function *) assoc, 0, &op_assoc);
56 MPI_Op_create((MPI_User_function *) addem, 1, &op_addem);
58 /* Run this for a variety of communicator sizes */
60 MPI_Comm_rank(comm, &rank);
61 MPI_Comm_size(comm, &size);
66 for (i = 0; i <= rank; i++)
69 MPI_Scan(&data, &result, 1, MPI_INT, MPI_SUM, comm);
70 if (result != correct_result) {
71 fprintf(stderr, "[%d] Error suming ints with scan\n", rank);
75 MPI_Scan(&data, &result, 1, MPI_INT, MPI_SUM, comm);
76 if (result != correct_result) {
77 fprintf(stderr, "[%d] Error summing ints with scan (2)\n", rank);
83 MPI_Scan(&data, &result, 1, MPI_INT, op_addem, comm);
84 if (result != correct_result) {
85 fprintf(stderr, "[%d] Error summing ints with scan (userop)\n", rank);
89 MPI_Scan(&data, &result, 1, MPI_INT, op_addem, comm);
90 if (result != correct_result) {
91 fprintf(stderr, "[%d] Error summing ints with scan (userop2)\n", rank);
96 MPI_Scan(&data, &result, 1, MPI_INT, op_assoc, comm);
97 if (result == BAD_ANSWER) {
98 fprintf(stderr, "[%d] Error scanning with non-commutative op\n", rank);
102 MPI_Op_free(&op_assoc);
103 MPI_Op_free(&op_addem);
107 printf("[%d] done with ERRORS(%d)!\n", rank, errors);
110 printf(" No Errors\n");