#include "mpi.h"
#include "mpitest.h"
-//#define DEBUG
-int test_communicators ( void );
-int copy_fn ( MPI_Comm, int, void *, void *, void *, int * );
-int delete_fn ( MPI_Comm, int, void *, void * );
+/* #define DEBUG */
+int test_communicators(void);
+int copy_fn(MPI_Comm, int, void *, void *, void *, int *);
+int delete_fn(MPI_Comm, int, void *, void *);
#ifdef DEBUG
#define FFLUSH fflush(stdout);
#else
#define FFLUSH
#endif
-int main( int argc, char **argv )
+int main(int argc, char **argv)
{
int errs = 0;
- MTest_Init( &argc, &argv );
-
+ MTest_Init(&argc, &argv);
+
errs = test_communicators();
- MTest_Finalize( errs );
+ MTest_Finalize(errs);
MPI_Finalize();
return 0;
}
-int copy_fn( MPI_Comm oldcomm, int keyval, void *extra_state,
- void *attribute_val_in, void *attribute_val_out, int *flag)
+int copy_fn(MPI_Comm oldcomm, int keyval, void *extra_state,
+ void *attribute_val_in, void *attribute_val_out, int *flag)
{
/* Note that if (sizeof(int) < sizeof(void *), just setting the int
- part of attribute_val_out may leave some dirty bits
- */
- *(MPI_Aint *)attribute_val_out = (MPI_Aint)attribute_val_in;
+ * part of attribute_val_out may leave some dirty bits
+ */
+ *(MPI_Aint *) attribute_val_out = (MPI_Aint) attribute_val_in;
*flag = 1;
return MPI_SUCCESS;
}
-int delete_fn( MPI_Comm comm, int keyval, void *attribute_val,
- void *extra_state)
+int delete_fn(MPI_Comm comm, int keyval, void *attribute_val, void *extra_state)
{
int world_rank;
- MPI_Comm_rank( MPI_COMM_WORLD, &world_rank );
- if ((MPI_Aint)attribute_val != (MPI_Aint)world_rank) {
- printf( "incorrect attribute value %d\n", *(int*)attribute_val );
- MPI_Abort(MPI_COMM_WORLD, 1005 );
- exit(1005);
+ MPI_Comm_rank(MPI_COMM_WORLD, &world_rank);
+ if ((MPI_Aint) attribute_val != (MPI_Aint) world_rank) {
+ printf("incorrect attribute value %d\n", *(int *) attribute_val);
+ MPI_Abort(MPI_COMM_WORLD, 1005);
}
return MPI_SUCCESS;
}
-int test_communicators( void )
+int test_communicators(void)
{
- MPI_Comm dup_comm_world, lo_comm, rev_comm, dup_comm,
- split_comm, world_comm;
+ MPI_Comm dup_comm_world, lo_comm, rev_comm, dup_comm, split_comm, world_comm;
MPI_Group world_group, lo_group, rev_group;
void *vvalue;
int ranges[1][3];
int errs = 0;
MPI_Aint value;
- MPI_Comm_rank( MPI_COMM_WORLD, &world_rank );
- MPI_Comm_size( MPI_COMM_WORLD, &world_size );
+ MPI_Comm_rank(MPI_COMM_WORLD, &world_rank);
+ MPI_Comm_size(MPI_COMM_WORLD, &world_size);
#ifdef DEBUG
if (world_rank == 0) {
- printf( "*** Communicators ***\n" ); fflush(stdout);
+ printf("*** Communicators ***\n");
+ fflush(stdout);
}
#endif
- MPI_Comm_dup( MPI_COMM_WORLD, &dup_comm_world );
+ MPI_Comm_dup(MPI_COMM_WORLD, &dup_comm_world);
/*
- Exercise Comm_create by creating an equivalent to dup_comm_world
- (sans attributes) and a half-world communicator.
- */
+ * Exercise Comm_create by creating an equivalent to dup_comm_world
+ * (sans attributes) and a half-world communicator.
+ */
#ifdef DEBUG
if (world_rank == 0) {
- printf( " Comm_create\n" ); fflush(stdout);
+ printf(" Comm_create\n");
+ fflush(stdout);
}
#endif
- MPI_Comm_group( dup_comm_world, &world_group );
- MPI_Comm_create( dup_comm_world, world_group, &world_comm );
- MPI_Comm_rank( world_comm, &rank );
+ MPI_Comm_group(dup_comm_world, &world_group);
+ MPI_Comm_create(dup_comm_world, world_group, &world_comm);
+ MPI_Comm_rank(world_comm, &rank);
if (rank != world_rank) {
- errs++;
- printf( "incorrect rank in world comm: %d\n", rank );
- MPI_Abort(MPI_COMM_WORLD, 3001 );
- exit(3001);
+ errs++;
+ printf("incorrect rank in world comm: %d\n", rank);
+ MPI_Abort(MPI_COMM_WORLD, 3001);
}
n = world_size / 2;
ranges[0][2] = 1;
#ifdef DEBUG
- printf( "world rank = %d before range incl\n", world_rank );FFLUSH;
+ printf("world rank = %d before range incl\n", world_rank);
+ FFLUSH;
#endif
- MPI_Group_range_incl(world_group, 1, ranges, &lo_group );
+ MPI_Group_range_incl(world_group, 1, ranges, &lo_group);
#ifdef DEBUG
- printf( "world rank = %d after range incl\n", world_rank );FFLUSH;
+ printf("world rank = %d after range incl\n", world_rank);
+ FFLUSH;
#endif
- MPI_Comm_create(world_comm, lo_group, &lo_comm );
+ MPI_Comm_create(world_comm, lo_group, &lo_comm);
#ifdef DEBUG
- printf( "world rank = %d before group free\n", world_rank );FFLUSH;
+ printf("world rank = %d before group free\n", world_rank);
+ FFLUSH;
#endif
- MPI_Group_free( &lo_group );
+ MPI_Group_free(&lo_group);
#ifdef DEBUG
- printf( "world rank = %d after group free\n", world_rank );FFLUSH;
+ printf("world rank = %d after group free\n", world_rank);
+ FFLUSH;
#endif
if (world_rank < (world_size - n)) {
- MPI_Comm_rank(lo_comm, &rank );
- if (rank == MPI_UNDEFINED) {
- errs++;
- printf( "incorrect lo group rank: %d\n", rank ); fflush(stdout);
- MPI_Abort(MPI_COMM_WORLD, 3002 );
- exit(3002);
- }
- else {
- /* printf( "lo in\n" );FFLUSH; */
- MPI_Barrier(lo_comm );
- /* printf( "lo out\n" );FFLUSH; */
- }
+ MPI_Comm_rank(lo_comm, &rank);
+ if (rank == MPI_UNDEFINED) {
+ errs++;
+ printf("incorrect lo group rank: %d\n", rank);
+ fflush(stdout);
+ MPI_Abort(MPI_COMM_WORLD, 3002);
+ }
+ else {
+ /* printf("lo in\n");FFLUSH; */
+ MPI_Barrier(lo_comm);
+ /* printf("lo out\n");FFLUSH; */
+ }
}
else {
- if (lo_comm != MPI_COMM_NULL) {
- errs++;
- printf( "rank : %d incorrect lo comm:\n", rank ); fflush(stdout);
- MPI_Abort(MPI_COMM_WORLD, 3003 );
- exit(3003);
- }
+ if (lo_comm != MPI_COMM_NULL) {
+ errs++;
+ printf("incorrect lo comm:\n");
+ fflush(stdout);
+ MPI_Abort(MPI_COMM_WORLD, 3003);
+ }
}
#ifdef DEBUG
- printf( "worldrank = %d\n", world_rank );FFLUSH;
+ printf("worldrank = %d\n", world_rank);
+ FFLUSH;
#endif
MPI_Barrier(world_comm);
#ifdef DEBUG
- printf( "bar!\n" );FFLUSH;
+ printf("bar!\n");
+ FFLUSH;
#endif
/*
- Check Comm_dup by adding attributes to lo_comm & duplicating
- */
+ * Check Comm_dup by adding attributes to lo_comm & duplicating
+ */
#ifdef DEBUG
if (world_rank == 0) {
- printf( " Comm_dup\n" );
- fflush(stdout);
+ printf(" Comm_dup\n");
+ fflush(stdout);
}
#endif
-
+
if (lo_comm != MPI_COMM_NULL) {
- value = 9;
- MPI_Keyval_create(copy_fn, delete_fn, &key_1, &value );
- value = 8;
- value = 7;
- MPI_Keyval_create(MPI_NULL_COPY_FN, MPI_NULL_DELETE_FN,
- &key_3, &value );
-
- /* This may generate a compilation warning; it is, however, an
- easy way to cache a value instead of a pointer */
- /* printf( "key1 = %x key3 = %x\n", key_1, key_3 ); */
- MPI_Attr_put(lo_comm, key_1, (void *) (MPI_Aint) world_rank );
- MPI_Attr_put(lo_comm, key_3, (void *)0 );
-
- MPI_Comm_dup(lo_comm, &dup_comm );
-
- /* Note that if sizeof(int) < sizeof(void *), we can't use
- (void **)&value to get the value we passed into Attr_put. To avoid
- problems (e.g., alignment errors), we recover the value into
- a (void *) and cast to int. Note that this may generate warning
- messages from the compiler. */
- MPI_Attr_get(dup_comm, key_1, (void **)&vvalue, &flag );
- value = (MPI_Aint)vvalue;
-
- if (! flag) {
- errs++;
- printf( "dup_comm key_1 not found on %d\n", world_rank );
- fflush( stdout );
- MPI_Abort(MPI_COMM_WORLD, 3004 );
- exit(3004);
- }
-
- if (value != world_rank) {
- errs++;
- printf( "dup_comm key_1 value incorrect: %ld, expected %d\n",
- (long)value, world_rank );
- fflush( stdout );
- MPI_Abort(MPI_COMM_WORLD, 3005 );
- exit(3005);
- }
-
- MPI_Attr_get(dup_comm, key_3, (void **)&vvalue, &flag );
- value = (MPI_Aint)vvalue;
- if (flag) {
- errs++;
- printf( "dup_comm key_3 found!\n" );
- fflush( stdout );
- MPI_Abort(MPI_COMM_WORLD, 3008 );
- exit(3008);
- }
- MPI_Keyval_free(&key_1 );
- MPI_Keyval_free(&key_3 );
+ value = 9;
+ MPI_Keyval_create(copy_fn, delete_fn, &key_1, &value);
+ value = 8;
+ value = 7;
+ MPI_Keyval_create(MPI_NULL_COPY_FN, MPI_NULL_DELETE_FN, &key_3, &value);
+
+ /* This may generate a compilation warning; it is, however, an
+ * easy way to cache a value instead of a pointer */
+ /* printf("key1 = %x key3 = %x\n", key_1, key_3); */
+ MPI_Attr_put(lo_comm, key_1, (void *) (MPI_Aint) world_rank);
+ MPI_Attr_put(lo_comm, key_3, (void *) 0);
+
+ MPI_Comm_dup(lo_comm, &dup_comm);
+
+ /* Note that if sizeof(int) < sizeof(void *), we can't use
+ * (void **)&value to get the value we passed into Attr_put. To avoid
+ * problems (e.g., alignment errors), we recover the value into
+ * a (void *) and cast to int. Note that this may generate warning
+ * messages from the compiler. */
+ MPI_Attr_get(dup_comm, key_1, (void **) &vvalue, &flag);
+ value = (MPI_Aint) vvalue;
+
+ if (!flag) {
+ errs++;
+ printf("dup_comm key_1 not found on %d\n", world_rank);
+ fflush(stdout);
+ MPI_Abort(MPI_COMM_WORLD, 3004);
+ }
+
+ if (value != world_rank) {
+ errs++;
+ printf("dup_comm key_1 value incorrect: %ld, expected %d\n", (long) value, world_rank);
+ fflush(stdout);
+ MPI_Abort(MPI_COMM_WORLD, 3005);
+ }
+
+ MPI_Attr_get(dup_comm, key_3, (void **) &vvalue, &flag);
+ value = (MPI_Aint) vvalue;
+ if (flag) {
+ errs++;
+ printf("dup_comm key_3 found!\n");
+ fflush(stdout);
+ MPI_Abort(MPI_COMM_WORLD, 3008);
+ }
+ MPI_Keyval_free(&key_1);
+ MPI_Keyval_free(&key_3);
}
- /*
- Split the world into even & odd communicators with reversed ranks.
- */
+ /*
+ * Split the world into even & odd communicators with reversed ranks.
+ */
#ifdef DEBUG
if (world_rank == 0) {
- printf( " Comm_split\n" );
- fflush(stdout);
+ printf(" Comm_split\n");
+ fflush(stdout);
}
#endif
-
+
color = world_rank % 2;
- key = world_size - world_rank;
-
- MPI_Comm_split(dup_comm_world, color, key, &split_comm );
- MPI_Comm_size(split_comm, &size );
- MPI_Comm_rank(split_comm, &rank );
- if (rank != ((size - world_rank/2) - 1)) {
- errs++;
- printf( "incorrect split rank: %d\n", rank ); fflush(stdout);
- MPI_Abort(MPI_COMM_WORLD, 3009 );
- exit(3009);
+ key = world_size - world_rank;
+
+ MPI_Comm_split(dup_comm_world, color, key, &split_comm);
+ MPI_Comm_size(split_comm, &size);
+ MPI_Comm_rank(split_comm, &rank);
+ if (rank != ((size - world_rank / 2) - 1)) {
+ errs++;
+ printf("incorrect split rank: %d\n", rank);
+ fflush(stdout);
+ MPI_Abort(MPI_COMM_WORLD, 3009);
}
-
- MPI_Barrier(split_comm );
+
+ MPI_Barrier(split_comm);
/*
- Test each possible Comm_compare result
- */
+ * Test each possible Comm_compare result
+ */
#ifdef DEBUG
if (world_rank == 0) {
- printf( " Comm_compare\n" );
- fflush(stdout);
+ printf(" Comm_compare\n");
+ fflush(stdout);
}
#endif
-
- MPI_Comm_compare(world_comm, world_comm, &result );
+
+ MPI_Comm_compare(world_comm, world_comm, &result);
if (result != MPI_IDENT) {
- errs++;
- printf( "incorrect ident result: %d\n", result );
- MPI_Abort(MPI_COMM_WORLD, 3010 );
- exit(3010);
+ errs++;
+ printf("incorrect ident result: %d\n", result);
+ MPI_Abort(MPI_COMM_WORLD, 3010);
}
-
+
if (lo_comm != MPI_COMM_NULL) {
- MPI_Comm_compare(lo_comm, dup_comm, &result );
- if (result != MPI_CONGRUENT) {
- errs++;
- printf( "incorrect congruent result: %d\n", result );
- MPI_Abort(MPI_COMM_WORLD, 3011 );
- exit(3011);
- }
+ MPI_Comm_compare(lo_comm, dup_comm, &result);
+ if (result != MPI_CONGRUENT) {
+ errs++;
+ printf("incorrect congruent result: %d\n", result);
+ MPI_Abort(MPI_COMM_WORLD, 3011);
+ }
}
-
+
ranges[0][0] = world_size - 1;
ranges[0][1] = 0;
ranges[0][2] = -1;
- MPI_Group_range_incl(world_group, 1, ranges, &rev_group );
- MPI_Comm_create(world_comm, rev_group, &rev_comm );
+ MPI_Group_range_incl(world_group, 1, ranges, &rev_group);
+ MPI_Comm_create(world_comm, rev_group, &rev_comm);
- MPI_Comm_compare(world_comm, rev_comm, &result );
+ MPI_Comm_compare(world_comm, rev_comm, &result);
if (result != MPI_SIMILAR && world_size != 1) {
- errs++;
- printf( "incorrect similar result: %d\n", result );
- MPI_Abort(MPI_COMM_WORLD, 3012 );
- exit(3012);
+ errs++;
+ printf("incorrect similar result: %d\n", result);
+ MPI_Abort(MPI_COMM_WORLD, 3012);
}
-
+
if (lo_comm != MPI_COMM_NULL) {
- MPI_Comm_compare(world_comm, lo_comm, &result );
- if (result != MPI_UNEQUAL && world_size != 1) {
- errs++;
- printf( "incorrect unequal result: %d\n", result );
- MPI_Abort(MPI_COMM_WORLD, 3013 );
- exit(3013);
- }
+ MPI_Comm_compare(world_comm, lo_comm, &result);
+ if (result != MPI_UNEQUAL && world_size != 1) {
+ errs++;
+ printf("incorrect unequal result: %d\n", result);
+ MPI_Abort(MPI_COMM_WORLD, 3013);
+ }
}
/*
- Free all communicators created
- */
+ * Free all communicators created
+ */
#ifdef DEBUG
- if (world_rank == 0)
- printf( " Comm_free\n" );
+ if (world_rank == 0)
+ printf(" Comm_free\n");
#endif
-
- MPI_Comm_free( &world_comm );
- MPI_Comm_free( &dup_comm_world );
-
- MPI_Comm_free( &rev_comm );
- MPI_Comm_free( &split_comm );
-
- MPI_Group_free( &world_group );
- MPI_Group_free( &rev_group );
-
+
+ MPI_Comm_free(&world_comm);
+ MPI_Comm_free(&dup_comm_world);
+
+ MPI_Comm_free(&rev_comm);
+ MPI_Comm_free(&split_comm);
+
+ MPI_Group_free(&world_group);
+ MPI_Group_free(&rev_group);
+
if (lo_comm != MPI_COMM_NULL) {
- MPI_Comm_free( &lo_comm );
- MPI_Comm_free( &dup_comm );
+ MPI_Comm_free(&lo_comm);
+ MPI_Comm_free(&dup_comm);
}
-
+
return errs;
}
-
