Create a dummy function that calls all ROMIO public symbols.
[mpich.git] / test / mpi / topo / neighb_coll.c
1 /* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
2 /*
3  *  (C) 2012 by Argonne National Laboratory.
4  *      See COPYRIGHT in top-level directory.
5  */
6 #include <stdio.h>
7 #include <stdlib.h>
8 #include <string.h>
9
10 #include <mpi.h>
11 #include "mpitest.h"
12
13 #if !defined(USE_STRICT_MPI) && defined(MPICH)
14 #define TEST_NEIGHB_COLL 1
15 #endif
16
17 /* assert-like macro that bumps the err count and emits a message */
18 #define check(x_)                                                                 \
19     do {                                                                          \
20         if (!(x_)) {                                                              \
21             ++errs;                                                               \
22             if (errs < 10) {                                                      \
23                 fprintf(stderr, "check failed: (%s), line %d\n", #x_, __LINE__); \
24             }                                                                     \
25         }                                                                         \
26     } while (0)
27
28 int main(int argc, char *argv[])
29 {
30     int errs = 0;
31     int wrank, wsize;
32     int periods[1] = { 0 };
33     MPI_Comm cart, dgraph, graph;
34
35     MPI_Init(&argc, &argv);
36     MPI_Comm_rank(MPI_COMM_WORLD, &wrank);
37     MPI_Comm_size(MPI_COMM_WORLD, &wsize);
38
39 #if defined(TEST_NEIGHB_COLL)
40     /* a basic test for the 10 (5 patterns x {blocking,nonblocking}) MPI-3
41      * neighborhood collective routines */
42
43     /* (wrap)--> 0 <--> 1 <--> ... <--> p-1 <--(wrap) */
44     MPI_Cart_create(MPI_COMM_WORLD, 1, &wsize, periods, /*reorder=*/0, &cart);
45
46     /* allgather */
47     {
48         int sendbuf[1] = { wrank };
49         int recvbuf[2] = { 0xdeadbeef, 0xdeadbeef };
50
51         /* should see one send to each neighbor (rank-1 and rank+1) and one receive
52          * each from same */
53         MPI_Neighbor_allgather(sendbuf, 1, MPI_INT, recvbuf, 1, MPI_INT, cart);
54
55         if (wrank == 0)
56             check(recvbuf[0] == 0xdeadbeef);
57         else
58             check(recvbuf[0] == wrank - 1);
59
60         if (wrank == wsize - 1)
61             check(recvbuf[1] == 0xdeadbeef);
62         else
63             check(recvbuf[1] == wrank + 1);
64     }
65
66     /* allgatherv */
67     {
68         int sendbuf[1]    = { wrank };
69         int recvbuf[2]    = { 0xdeadbeef, 0xdeadbeef };
70         int recvcounts[2] = { 1, 1 };
71         int displs[2]     = { 1, 0};
72
73         /* should see one send to each neighbor (rank-1 and rank+1) and one receive
74          * each from same, but put them in opposite slots in the buffer */
75         MPI_Neighbor_allgatherv(sendbuf, 1, MPI_INT, recvbuf, recvcounts, displs, MPI_INT, cart);
76
77         if (wrank == 0)
78             check(recvbuf[1] == 0xdeadbeef);
79         else
80             check(recvbuf[1] == wrank - 1);
81
82         if (wrank == wsize - 1)
83             check(recvbuf[0] == 0xdeadbeef);
84         else
85             check(recvbuf[0] == wrank + 1);
86     }
87
88     /* alltoall */
89     {
90         int sendbuf[2]    = { -(wrank+1), wrank+1 };
91         int recvbuf[2]    = { 0xdeadbeef, 0xdeadbeef };
92
93         /* should see one send to each neighbor (rank-1 and rank+1) and one
94          * receive each from same */
95         MPI_Neighbor_alltoall(sendbuf, 1, MPI_INT, recvbuf, 1, MPI_INT, cart);
96
97         if (wrank == 0)
98             check(recvbuf[0] == 0xdeadbeef);
99         else
100             check(recvbuf[0] == wrank);
101
102         if (wrank == wsize - 1)
103             check(recvbuf[1] == 0xdeadbeef);
104         else
105             check(recvbuf[1] == -(wrank + 2));
106     }
107
108     /* alltoallv */
109     {
110         int sendbuf[2]    = { -(wrank+1), wrank+1 };
111         int recvbuf[2]    = { 0xdeadbeef, 0xdeadbeef };
112         int sendcounts[2] = { 1, 1 };
113         int recvcounts[2] = { 1, 1 };
114         int sdispls[2]    = { 0, 1 };
115         int rdispls[2]    = { 1, 0 };
116
117         /* should see one send to each neighbor (rank-1 and rank+1) and one receive
118          * each from same, but put them in opposite slots in the buffer */
119         MPI_Neighbor_alltoallv(sendbuf, sendcounts, sdispls, MPI_INT,
120                                 recvbuf, recvcounts, rdispls, MPI_INT,
121                                 cart);
122
123         if (wrank == 0)
124             check(recvbuf[1] == 0xdeadbeef);
125         else
126             check(recvbuf[1] == wrank);
127
128         if (wrank == wsize - 1)
129             check(recvbuf[0] == 0xdeadbeef);
130         else
131             check(recvbuf[0] == -(wrank + 2));
132     }
133
134     /* alltoallw */
135     {
136         int sendbuf[2]            = { -(wrank+1), wrank+1 };
137         int recvbuf[2]            = { 0xdeadbeef, 0xdeadbeef };
138         int sendcounts[2]         = { 1, 1 };
139         int recvcounts[2]         = { 1, 1 };
140         MPI_Aint sdispls[2]       = { 0, sizeof(int) };
141         MPI_Aint rdispls[2]       = { sizeof(int), 0 };
142         MPI_Datatype sendtypes[2] = { MPI_INT, MPI_INT };
143         MPI_Datatype recvtypes[2] = { MPI_INT, MPI_INT };
144
145         /* should see one send to each neighbor (rank-1 and rank+1) and one receive
146          * each from same, but put them in opposite slots in the buffer */
147         MPI_Neighbor_alltoallw(sendbuf, sendcounts, sdispls, sendtypes,
148                                 recvbuf, recvcounts, rdispls, recvtypes,
149                                 cart);
150
151         if (wrank == 0)
152             check(recvbuf[1] == 0xdeadbeef);
153         else
154             check(recvbuf[1] == wrank);
155
156         if (wrank == wsize - 1)
157             check(recvbuf[0] == 0xdeadbeef);
158         else
159             check(recvbuf[0] == -(wrank + 2));
160     }
161
162
163     MPI_Comm_free(&cart);
164 #endif /* defined(TEST_NEIGHB_COLL) */
165
166     MPI_Reduce((wrank == 0 ? MPI_IN_PLACE : &errs), &errs, 1, MPI_INT, MPI_SUM, 0, MPI_COMM_WORLD);
167     if (wrank == 0) {
168         if (errs) {
169             printf("found %d errors\n", errs);
170         }
171         else {
172             printf(" No errors\n");
173         }
174     }
175     MPI_Finalize();
176
177     return 0;
178 }
179