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