Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
add mpich3 topo tests
[simgrid.git] / teshsuite / smpi / mpich3-test / topo / dgraph_unwgt.c
1 #include <stdio.h>
2 #include <stdlib.h>
3 #include <string.h>
4 #include "mpi.h"
5 #include "mpitest.h"
6
7 #define RING_NUM_NEIGHBORS   2
8
9 static int validate_dgraph(MPI_Comm dgraph_comm)
10 {
11     int         comm_topo;
12     int         src_sz, dest_sz;
13     int         wgt_flag, ierr;
14     int         srcs[RING_NUM_NEIGHBORS], dests[RING_NUM_NEIGHBORS];
15     int        *src_wgts, *dest_wgts;
16
17     int         world_rank, world_size;
18     int         idx, nbr_sep;
19
20     comm_topo = MPI_UNDEFINED;
21     MPI_Topo_test(dgraph_comm, &comm_topo);
22     switch (comm_topo) {
23         case MPI_DIST_GRAPH :
24             break;
25         default:
26             fprintf(stderr, "dgraph_comm is NOT of type MPI_DIST_GRAPH\n");
27             return 0;
28     }
29
30     ierr = MPI_Dist_graph_neighbors_count(dgraph_comm,
31                                           &src_sz, &dest_sz, &wgt_flag);
32     if (ierr != MPI_SUCCESS) {
33         fprintf(stderr, "MPI_Dist_graph_neighbors_count() fails!\n");
34         return 0;
35     }
36 /*
37     else
38         fprintf(stderr, "MPI_Dist_graph_neighbors_count() succeeds!\n");
39 */
40
41     if (wgt_flag) {
42         fprintf(stderr, "dgraph_comm is NOT created with MPI_UNWEIGHTED\n");
43         return 0;
44     }
45 /*
46     else
47         fprintf(stderr, "dgraph_comm is created with MPI_UNWEIGHTED\n");
48 */
49     if (src_sz != RING_NUM_NEIGHBORS || dest_sz != RING_NUM_NEIGHBORS) {
50         fprintf(stderr, "source or destination edge array is not of size %d.\n",
51                          RING_NUM_NEIGHBORS);
52         fprintf(stderr, "src_sz = %d, dest_sz = %d\n", src_sz, dest_sz);
53         return 0;
54     }
55
56     /*
57        src_wgts and dest_wgts could be anything, e.g. NULL, since
58        MPI_Dist_graph_neighbors_count() returns MPI_UNWEIGHTED.
59        Since this program has a Fortran77 version, and standard Fortran77
60        has no pointer and NULL, so use MPI_UNWEIGHTED for the weighted arrays.
61     */
62     src_wgts  = MPI_UNWEIGHTED;
63     dest_wgts = MPI_UNWEIGHTED;
64     ierr = MPI_Dist_graph_neighbors(dgraph_comm,
65                                     src_sz, srcs, src_wgts,
66                                     dest_sz, dests, dest_wgts);
67     if (ierr != MPI_SUCCESS) {
68         fprintf(stderr, "MPI_Dist_graph_neighbors() fails!\n");
69         return 0;
70     }
71 /*
72     else
73         fprintf(stderr, "MPI_Dist_graph_neighbors() succeeds!\n");
74 */
75
76     /*
77        Check if the neighbors returned from MPI are really
78        the nearest neighbors within a ring.
79     */
80     MPI_Comm_size(MPI_COMM_WORLD, &world_size);
81     MPI_Comm_rank(MPI_COMM_WORLD, &world_rank);
82  
83     for (idx=0; idx < src_sz; idx++) {
84         nbr_sep = abs(srcs[idx] - world_rank);
85         if ( nbr_sep != 1 && nbr_sep != (world_size-1) ) {
86             fprintf(stderr, "srcs[%d]=%d is NOT a neighbor of my rank %d.\n",
87                             idx, srcs[idx], world_rank);
88             return 0;
89         }  
90     }
91     for (idx=0; idx < dest_sz; idx++) {
92         nbr_sep = abs(dests[idx] - world_rank);
93         if ( nbr_sep != 1 && nbr_sep != (world_size-1) ) {
94             fprintf(stderr, "dests[%d]=%d is NOT a neighbor of my rank %d.\n",
95                             idx, dests[idx], world_rank);
96             return 0;
97         }  
98     }
99
100     /*
101     fprintf(stderr, "dgraph_comm is of type MPI_DIST_GRAPH "
102                     "of a bidirectional ring.\n");
103     */
104     return 1;
105 }
106
107 /*
108    Specify a distributed graph of a bidirectional ring of the MPI_COMM_WORLD,
109    i.e. everyone only talks to left and right neighbors. 
110 */
111 int main(int argc, char *argv[])
112 {
113     MPI_Comm    dgraph_comm;
114     int         world_size, world_rank, ierr;
115     int         errs = 0;
116
117     int         src_sz, dest_sz;
118     int         degs[1];
119     int         srcs[RING_NUM_NEIGHBORS], dests[RING_NUM_NEIGHBORS];
120     
121     MTest_Init(&argc, &argv);
122     MPI_Comm_size(MPI_COMM_WORLD, &world_size);
123     MPI_Comm_rank(MPI_COMM_WORLD, &world_rank);
124
125     degs[0]  = 2;
126     srcs[0]  = world_rank;
127     dests[0] = world_rank-1 <  0          ? world_size-1 : world_rank-1 ;
128     dests[1] = world_rank+1 >= world_size ?            0 : world_rank+1 ;
129     ierr = MPI_Dist_graph_create(MPI_COMM_WORLD, 1, srcs, degs, dests,
130                                  MPI_UNWEIGHTED, MPI_INFO_NULL, 1,
131                                  &dgraph_comm);
132     if ( ierr != MPI_SUCCESS )  {
133         fprintf(stderr, "MPI_Dist_graph_create() fails!\n");
134         MPI_Abort(MPI_COMM_WORLD, 1);
135         return 1;
136     }
137     if (!validate_dgraph(dgraph_comm)) {
138         fprintf(stderr, "MPI_Dist_graph_create() does NOT create "
139                         "a bidirectional ring graph!\n");
140         MPI_Abort(MPI_COMM_WORLD, 1);
141         return 1;
142     }
143     MPI_Comm_free(&dgraph_comm);
144     
145     src_sz   = 2;
146     srcs[0]  = world_rank-1 <  0          ? world_size-1 : world_rank-1 ;
147     srcs[1]  = world_rank+1 >= world_size ?            0 : world_rank+1 ;
148     dest_sz  = 2;
149     dests[0] = world_rank-1 <  0          ? world_size-1 : world_rank-1 ;
150     dests[1] = world_rank+1 >= world_size ?            0 : world_rank+1 ;
151     ierr = MPI_Dist_graph_create_adjacent(MPI_COMM_WORLD,
152                                           src_sz, srcs, MPI_UNWEIGHTED,
153                                           dest_sz, dests, MPI_UNWEIGHTED,
154                                           MPI_INFO_NULL, 1, &dgraph_comm);
155     if ( ierr != MPI_SUCCESS ) {
156         fprintf(stderr, "MPI_Dist_graph_create_adjacent() fails!\n");
157         MPI_Abort(MPI_COMM_WORLD, 1);
158         return 1;
159     }
160     if (!validate_dgraph(dgraph_comm)) {
161         fprintf(stderr, "MPI_Dist_graph_create_adjacent() does NOT create "
162                         "a bidirectional ring graph!\n");
163         MPI_Abort(MPI_COMM_WORLD, 1);
164         return 1;
165     }
166     MPI_Comm_free(&dgraph_comm);
167
168     MTest_Finalize(errs);
169     MPI_Finalize();
170     return 0;
171 }