Add build scripts for Jenkins test jobs
[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, cart);
121
122         if (wrank == 0)
123             check(recvbuf[1] == 0xdeadbeef);
124         else
125             check(recvbuf[1] == wrank);
126
127         if (wrank == wsize - 1)
128             check(recvbuf[0] == 0xdeadbeef);
129         else
130             check(recvbuf[0] == -(wrank + 2));
131     }
132
133     /* alltoallw */
134     {
135         int sendbuf[2] = { -(wrank + 1), wrank + 1 };
136         int recvbuf[2] = { 0xdeadbeef, 0xdeadbeef };
137         int sendcounts[2] = { 1, 1 };
138         int recvcounts[2] = { 1, 1 };
139         MPI_Aint sdispls[2] = { 0, sizeof(int) };
140         MPI_Aint rdispls[2] = { sizeof(int), 0 };
141         MPI_Datatype sendtypes[2] = { MPI_INT, MPI_INT };
142         MPI_Datatype recvtypes[2] = { MPI_INT, MPI_INT };
143
144         /* should see one send to each neighbor (rank-1 and rank+1) and one receive
145          * each from same, but put them in opposite slots in the buffer */
146         MPI_Neighbor_alltoallw(sendbuf, sendcounts, sdispls, sendtypes,
147                                recvbuf, recvcounts, rdispls, recvtypes, cart);
148
149         if (wrank == 0)
150             check(recvbuf[1] == 0xdeadbeef);
151         else
152             check(recvbuf[1] == wrank);
153
154         if (wrank == wsize - 1)
155             check(recvbuf[0] == 0xdeadbeef);
156         else
157             check(recvbuf[0] == -(wrank + 2));
158     }
159
160
161     MPI_Comm_free(&cart);
162 #endif /* defined(TEST_NEIGHB_COLL) */
163
164     MPI_Reduce((wrank == 0 ? MPI_IN_PLACE : &errs), &errs, 1, MPI_INT, MPI_SUM, 0, MPI_COMM_WORLD);
165     if (wrank == 0) {
166         if (errs) {
167             printf("found %d errors\n", errs);
168         }
169         else {
170             printf(" No errors\n");
171         }
172     }
173     MPI_Finalize();
174
175     return 0;
176 }