Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
examples to demonstrate bugs in sendrecv
[simgrid.git] / examples / smpi / allreduce.c
1 #include <stdio.h>
2 #include <mpi.h>
3
4 /**
5  * MESSAGE PASSING INTERFACE TEST CASE SUITE
6  *
7  * Copyright IBM Corp. 1995
8  * 
9  * IBM Corp. hereby grants a non-exclusive license to use, copy, modify, and
10  *distribute this software for any purpose and without fee provided that the
11  *above copyright notice and the following paragraphs appear in all copies.
12
13  * IBM Corp. makes no representation that the test cases comprising this
14  * suite are correct or are an accurate representation of any standard.
15
16  * In no event shall IBM be liable to any party for direct, indirect, special
17  * incidental, or consequential damage arising out of the use of this software
18  * even if IBM Corp. has been advised of the possibility of such damage.
19
20  * IBM CORP. SPECIFICALLY DISCLAIMS ANY WARRANTIES INCLUDING, BUT NOT LIMITED
21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22  * PURPOSE.  THE SOFTWARE PROVIDED HEREUNDER IS ON AN "AS IS" BASIS AND IBM
23  * CORP. HAS NO OBLIGATION TO PROVIDE MAINTENANCE, SUPPORT, UPDATES,
24  * ENHANCEMENTS, OR MODIFICATIONS.
25  * ***************************************************************************
26  **/
27 int ibm_test(int rank, int size)
28 {
29           int success = 1;
30 #define MAXLEN  10000
31
32           int root, i, j, k;
33           int out[MAXLEN];
34           int in[MAXLEN];
35
36           for (j = 1; j <= MAXLEN; j *= 10) {
37                     for (i = 0; i < j; i++)
38                                 out[i] = i;
39
40                     MPI_Allreduce(out, in, j, MPI_INT, MPI_SUM, MPI_COMM_WORLD);
41                     MPI_Barrier(MPI_COMM_WORLD);
42
43                     if (rank == root) {
44                                 for (k = 0; k < j; k++) {
45                                           if (in[k] != k * size) {
46                                                     printf("bad answer (%d) at index %d of %d (should be %d)", in[k], k,
47                                                                           j, k * size);
48                                                     success = 0;
49                                                     break;
50                                           }
51                                 }
52                     }
53           }
54           return (success);
55 }
56
57
58
59
60 int main(int argc, char **argv)
61 {
62   int size, rank;
63
64
65   MPI_Init(&argc, &argv);
66   MPI_Comm_size(MPI_COMM_WORLD, &size);
67   MPI_Comm_rank(MPI_COMM_WORLD, &rank);
68
69
70   if (0 == rank)
71     printf("** IBM Test Result: ... \n");
72   if (!ibm_test(rank, size))
73     printf("\t[%d] failed.\n", rank);
74   else
75     printf("\t[%d] ok.\n", rank);
76
77   MPI_Finalize();
78   return 0;
79 }