1 /* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
4 * (C) 2001 by Argonne National Laboratory.
5 * See COPYRIGHT in top-level directory.
9 Exercise communicator routines.
11 This C version derived from a Fortran test program from ....
19 int test_communicators ( void );
20 int copy_fn ( MPI_Comm, int, void *, void *, void *, int * );
21 int delete_fn ( MPI_Comm, int, void *, void * );
23 #define FFLUSH fflush(stdout);
28 int main( int argc, char **argv )
31 MTest_Init( &argc, &argv );
33 errs = test_communicators();
34 MTest_Finalize( errs );
39 int copy_fn( MPI_Comm oldcomm, int keyval, void *extra_state,
40 void *attribute_val_in, void *attribute_val_out, int *flag)
42 /* Note that if (sizeof(int) < sizeof(void *), just setting the int
43 part of attribute_val_out may leave some dirty bits
45 *(MPI_Aint *)attribute_val_out = (MPI_Aint)attribute_val_in;
50 int delete_fn( MPI_Comm comm, int keyval, void *attribute_val,
54 MPI_Comm_rank( MPI_COMM_WORLD, &world_rank );
55 if ((MPI_Aint)attribute_val != (MPI_Aint)world_rank) {
56 printf( "incorrect attribute value %d\n", *(int*)attribute_val );
57 MPI_Abort(MPI_COMM_WORLD, 1005 );
62 int test_communicators( void )
64 MPI_Comm dup_comm_world, lo_comm, rev_comm, dup_comm,
65 split_comm, world_comm;
66 MPI_Group world_group, lo_group, rev_group;
69 int flag, world_rank, world_size, rank, size, n, key_1, key_3;
70 int color, key, result;
74 MPI_Comm_rank( MPI_COMM_WORLD, &world_rank );
75 MPI_Comm_size( MPI_COMM_WORLD, &world_size );
77 if (world_rank == 0) {
78 printf( "*** Communicators ***\n" ); fflush(stdout);
82 MPI_Comm_dup( MPI_COMM_WORLD, &dup_comm_world );
85 Exercise Comm_create by creating an equivalent to dup_comm_world
86 (sans attributes) and a half-world communicator.
90 if (world_rank == 0) {
91 printf( " Comm_create\n" ); fflush(stdout);
95 MPI_Comm_group( dup_comm_world, &world_group );
96 MPI_Comm_create( dup_comm_world, world_group, &world_comm );
97 MPI_Comm_rank( world_comm, &rank );
98 if (rank != world_rank) {
100 printf( "incorrect rank in world comm: %d\n", rank );
101 MPI_Abort(MPI_COMM_WORLD, 3001 );
107 ranges[0][1] = (world_size - n) - 1;
111 printf( "world rank = %d before range incl\n", world_rank );FFLUSH;
113 MPI_Group_range_incl(world_group, 1, ranges, &lo_group );
115 printf( "world rank = %d after range incl\n", world_rank );FFLUSH;
117 MPI_Comm_create(world_comm, lo_group, &lo_comm );
119 printf( "world rank = %d before group free\n", world_rank );FFLUSH;
121 MPI_Group_free( &lo_group );
124 printf( "world rank = %d after group free\n", world_rank );FFLUSH;
127 if (world_rank < (world_size - n)) {
128 MPI_Comm_rank(lo_comm, &rank );
129 if (rank == MPI_UNDEFINED) {
131 printf( "incorrect lo group rank: %d\n", rank ); fflush(stdout);
132 MPI_Abort(MPI_COMM_WORLD, 3002 );
135 /* printf( "lo in\n" );FFLUSH; */
136 MPI_Barrier(lo_comm );
137 /* printf( "lo out\n" );FFLUSH; */
141 if (lo_comm != MPI_COMM_NULL) {
143 printf( "rank : %d incorrect lo comm:\n", rank ); fflush(stdout);
144 MPI_Abort(MPI_COMM_WORLD, 3003 );
149 printf( "worldrank = %d\n", world_rank );FFLUSH;
151 MPI_Barrier(world_comm);
154 printf( "bar!\n" );FFLUSH;
157 Check Comm_dup by adding attributes to lo_comm & duplicating
160 if (world_rank == 0) {
161 printf( " Comm_dup\n" );
166 if (lo_comm != MPI_COMM_NULL) {
168 MPI_Keyval_create(copy_fn, delete_fn, &key_1, &value );
171 MPI_Keyval_create(MPI_NULL_COPY_FN, MPI_NULL_DELETE_FN,
174 /* This may generate a compilation warning; it is, however, an
175 easy way to cache a value instead of a pointer */
176 /* printf( "key1 = %x key3 = %x\n", key_1, key_3 ); */
177 MPI_Attr_put(lo_comm, key_1, (void *) (MPI_Aint) world_rank );
178 MPI_Attr_put(lo_comm, key_3, (void *)0 );
180 MPI_Comm_dup(lo_comm, &dup_comm );
182 /* Note that if sizeof(int) < sizeof(void *), we can't use
183 (void **)&value to get the value we passed into Attr_put. To avoid
184 problems (e.g., alignment errors), we recover the value into
185 a (void *) and cast to int. Note that this may generate warning
186 messages from the compiler. */
187 MPI_Attr_get(dup_comm, key_1, (void **)&vvalue, &flag );
188 value = (MPI_Aint)vvalue;
192 printf( "dup_comm key_1 not found on %d\n", world_rank );
194 MPI_Abort(MPI_COMM_WORLD, 3004 );
197 if (value != world_rank) {
199 printf( "dup_comm key_1 value incorrect: %ld, expected %d\n",
200 (long)value, world_rank );
202 MPI_Abort(MPI_COMM_WORLD, 3005 );
205 MPI_Attr_get(dup_comm, key_3, (void **)&vvalue, &flag );
206 value = (MPI_Aint)vvalue;
209 printf( "dup_comm key_3 found!\n" );
211 MPI_Abort(MPI_COMM_WORLD, 3008 );
213 MPI_Keyval_free(&key_1 );
214 MPI_Keyval_free(&key_3 );
217 Split the world into even & odd communicators with reversed ranks.
220 if (world_rank == 0) {
221 printf( " Comm_split\n" );
226 color = world_rank % 2;
227 key = world_size - world_rank;
229 MPI_Comm_split(dup_comm_world, color, key, &split_comm );
230 MPI_Comm_size(split_comm, &size );
231 MPI_Comm_rank(split_comm, &rank );
232 if (rank != ((size - world_rank/2) - 1)) {
234 printf( "incorrect split rank: %d\n", rank ); fflush(stdout);
235 MPI_Abort(MPI_COMM_WORLD, 3009 );
238 MPI_Barrier(split_comm );
240 Test each possible Comm_compare result
243 if (world_rank == 0) {
244 printf( " Comm_compare\n" );
249 MPI_Comm_compare(world_comm, world_comm, &result );
250 if (result != MPI_IDENT) {
252 printf( "incorrect ident result: %d\n", result );
253 MPI_Abort(MPI_COMM_WORLD, 3010 );
256 if (lo_comm != MPI_COMM_NULL) {
257 MPI_Comm_compare(lo_comm, dup_comm, &result );
258 if (result != MPI_CONGRUENT) {
260 printf( "incorrect congruent result: %d\n", result );
261 MPI_Abort(MPI_COMM_WORLD, 3011 );
265 ranges[0][0] = world_size - 1;
269 MPI_Group_range_incl(world_group, 1, ranges, &rev_group );
270 MPI_Comm_create(world_comm, rev_group, &rev_comm );
272 MPI_Comm_compare(world_comm, rev_comm, &result );
273 if (result != MPI_SIMILAR && world_size != 1) {
275 printf( "incorrect similar result: %d\n", result );
276 MPI_Abort(MPI_COMM_WORLD, 3012 );
279 if (lo_comm != MPI_COMM_NULL) {
280 MPI_Comm_compare(world_comm, lo_comm, &result );
281 if (result != MPI_UNEQUAL && world_size != 1) {
283 printf( "incorrect unequal result: %d\n", result );
284 MPI_Abort(MPI_COMM_WORLD, 3013 );
288 Free all communicators created
292 printf( " Comm_free\n" );
295 MPI_Comm_free( &world_comm );
296 MPI_Comm_free( &dup_comm_world );
298 MPI_Comm_free( &rev_comm );
299 MPI_Comm_free( &split_comm );
301 MPI_Group_free( &world_group );
302 MPI_Group_free( &rev_group );
304 if (lo_comm != MPI_COMM_NULL) {
305 MPI_Comm_free( &lo_comm );
306 MPI_Comm_free( &dup_comm );