teshsuite/smpi/mpich3-test/coll/scantst.c

   1 /* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
   2 /*
   3  *  (C) 2001 by Argonne National Laboratory.
   4  *      See COPYRIGHT in top-level directory.
   5  */
   6 #include "mpi.h"
   7 #include <stdio.h>
   8 #include "mpitest.h"
   9
  10 void addem(int *, int *, int *, MPI_Datatype *);
  11 void assoc(int *, int *, int *, MPI_Datatype *);
  12
  13 void addem(int *invec, int *inoutvec, int *len, MPI_Datatype * dtype)
  14 {
  15     int i;
  16     for (i = 0; i < *len; i++)
  17         inoutvec[i] += invec[i];
  18 }
  19
  20 #define BAD_ANSWER 100000
  21
  22 /*
  23     The operation is inoutvec[i] = invec[i] op inoutvec[i]
  24     (see 4.9.4).  The order is important.
  25
  26     Note that the computation is in process rank (in the communicator)
  27     order, independant of the root.
  28  */
  29 void assoc(int *invec, int *inoutvec, int *len, MPI_Datatype * dtype)
  30 {
  31     int i;
  32     for (i = 0; i < *len; i++) {
  33         if (inoutvec[i] <= invec[i]) {
  34             int rank;
  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;
  38         }
  39         else
  40             inoutvec[i] = invec[i];
  41     }
  42 }
  43
  44 int main(int argc, char **argv)
  45 {
  46     int rank, size, i;
  47     int data;
  48     int errors = 0;
  49     int result = -100;
  50     int correct_result;
  51     MPI_Op op_assoc, op_addem;
  52     MPI_Comm comm = MPI_COMM_WORLD;
  53
  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);
  57
  58     /* Run this for a variety of communicator sizes */
  59
  60     MPI_Comm_rank(comm, &rank);
  61     MPI_Comm_size(comm, &size);
  62
  63     data = rank;
  64
  65     correct_result = 0;
  66     for (i = 0; i <= rank; i++)
  67         correct_result += i;
  68
  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);
  72         errors++;
  73     }
  74
  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);
  78         errors++;
  79     }
  80
  81     data = rank;
  82     result = -100;
  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);
  86         errors++;
  87     }
  88
  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);
  92         errors++;
  93     }
  94     result = -100;
  95     data = 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);
  99         errors++;
 100     }
 101
 102     MPI_Op_free(&op_assoc);
 103     MPI_Op_free(&op_addem);
 104
 105     MPI_Finalize();
 106     if (errors)
 107         printf("[%d] done with ERRORS(%d)!\n", rank, errors);
 108     else {
 109         if (rank == 0)
 110             printf(" No Errors\n");
 111     }
 112
 113     return errors;
 114 }