Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Upgrade coll mpich testlist to new mpich
[simgrid.git] / teshsuite / smpi / mpich3-test / coll / redscat2.c
index f77bad8..4c14537 100644 (file)
@@ -3,7 +3,7 @@
  *  (C) 2001 by Argonne National Laboratory.
  *      See COPYRIGHT in top-level directory.
  */
-/* 
+/*
  * Test of reduce scatter.
  *
  * Checks that non-commutative operations are not commuted and that
 int err = 0;
 
 /* left(x,y) ==> x */
-void left(void *a, void *b, int *count, MPI_Datatype *type);
-void left(void *a, void *b, int *count, MPI_Datatype *type)
+void left(void *a, void *b, int *count, MPI_Datatype * type);
+void left(void *a, void *b, int *count, MPI_Datatype * type)
 {
     int *in = a;
     int *inout = b;
     int i;
 
-    for (i = 0; i < *count; ++i)
-    {
+    for (i = 0; i < *count; ++i) {
         if (in[i] > inout[i])
             ++err;
         inout[i] = in[i];
@@ -36,15 +35,14 @@ void left(void *a, void *b, int *count, MPI_Datatype *type)
 }
 
 /* right(x,y) ==> y */
-void right(void *a, void *b, int *count, MPI_Datatype *type);
-void right(void *a, void *b, int *count, MPI_Datatype *type)
+void right(void *a, void *b, int *count, MPI_Datatype * type);
+void right(void *a, void *b, int *count, MPI_Datatype * type)
 {
     int *in = a;
     int *inout = b;
     int i;
 
-    for (i = 0; i < *count; ++i)
-    {
+    for (i = 0; i < *count; ++i) {
         if (in[i] > inout[i])
             ++err;
         inout[i] = inout[i];
@@ -53,76 +51,79 @@ void right(void *a, void *b, int *count, MPI_Datatype *type)
 
 /* Just performs a simple sum but can be marked as non-commutative to
    potentially tigger different logic in the implementation. */
-void nc_sum(void *a, void *b, int *count, MPI_Datatype *type);
-void nc_sum(void *a, void *b, int *count, MPI_Datatype *type)
+void nc_sum(void *a, void *b, int *count, MPI_Datatype * type);
+void nc_sum(void *a, void *b, int *count, MPI_Datatype * type)
 {
     int *in = a;
     int *inout = b;
     int i;
 
-    for (i = 0; i < *count; ++i)
-    {
+    for (i = 0; i < *count; ++i) {
         inout[i] = in[i] + inout[i];
     }
 }
 
 #define MAX_BLOCK_SIZE 256
 
-int main( int argc, char **argv )
+int main(int argc, char **argv)
 {
-    int      *sendbuf, *recvcounts;
-    int      block_size;
-    int      *recvbuf;
-    int      size, rank, i;
+    int *sendbuf, *recvcounts;
+    int block_size;
+    int *recvbuf;
+    int size, rank, i;
     MPI_Comm comm;
     MPI_Op left_op, right_op, nc_sum_op;
 
-    MTest_Init( &argc, &argv );
+    MTest_Init(&argc, &argv);
     comm = MPI_COMM_WORLD;
 
-    MPI_Comm_size( comm, &size );
-    MPI_Comm_rank( comm, &rank );
+    MPI_Comm_size(comm, &size);
+    MPI_Comm_rank(comm, &rank);
 
-    MPI_Op_create(&left, 0/*non-commutative*/, &left_op);
-    MPI_Op_create(&right, 0/*non-commutative*/, &right_op);
-    MPI_Op_create(&nc_sum, 0/*non-commutative*/, &nc_sum_op);
+    MPI_Op_create(&left, 0 /*non-commutative */ , &left_op);
+    MPI_Op_create(&right, 0 /*non-commutative */ , &right_op);
+    MPI_Op_create(&nc_sum, 0 /*non-commutative */ , &nc_sum_op);
 
     for (block_size = 1; block_size < MAX_BLOCK_SIZE; block_size *= 2) {
-        sendbuf = (int *) malloc( block_size * size * sizeof(int) );
-        recvbuf = malloc( block_size * sizeof(int) );
+        sendbuf = (int *) malloc(block_size * size * sizeof(int));
+        recvbuf = malloc(block_size * sizeof(int));
 
-        for (i=0; i<(size*block_size); i++) 
+        for (i = 0; i < (size * block_size); i++)
             sendbuf[i] = rank + i;
-        for (i=0; i<block_size; i++)
+        for (i = 0; i < block_size; i++)
             recvbuf[i] = 0xdeadbeef;
-        recvcounts = (int *)malloc( size * sizeof(int) );
-        for (i=0; i<size; i++) 
+        recvcounts = (int *) malloc(size * sizeof(int));
+        for (i = 0; i < size; i++)
             recvcounts[i] = block_size;
 
-        MPI_Reduce_scatter( sendbuf, recvbuf, recvcounts, MPI_INT, left_op, comm );
+        MPI_Reduce_scatter(sendbuf, recvbuf, recvcounts, MPI_INT, left_op, comm);
         for (i = 0; i < block_size; ++i)
-            if (recvbuf[i] != (rank * block_size + i)) ++err;
+            if (recvbuf[i] != (rank * block_size + i))
+                ++err;
 
-        MPI_Reduce_scatter( sendbuf, recvbuf, recvcounts, MPI_INT, right_op, comm );
+        MPI_Reduce_scatter(sendbuf, recvbuf, recvcounts, MPI_INT, right_op, comm);
         for (i = 0; i < block_size; ++i)
-            if (recvbuf[i] != ((size - 1) + (rank * block_size) + i)) ++err;
+            if (recvbuf[i] != ((size - 1) + (rank * block_size) + i))
+                ++err;
 
-        MPI_Reduce_scatter( sendbuf, recvbuf, recvcounts, MPI_INT, nc_sum_op, comm );
+        MPI_Reduce_scatter(sendbuf, recvbuf, recvcounts, MPI_INT, nc_sum_op, comm);
         for (i = 0; i < block_size; ++i) {
             int x = rank * block_size + i;
-            if (recvbuf[i] != (size*x + (size-1)*size/2)) ++err;
+            if (recvbuf[i] != (size * x + (size - 1) * size / 2))
+                ++err;
         }
 
         free(recvbuf);
         free(sendbuf);
+        free(recvcounts);
     }
 
     MPI_Op_free(&left_op);
     MPI_Op_free(&right_op);
     MPI_Op_free(&nc_sum_op);
 
-    MTest_Finalize( err );
-    MPI_Finalize( );
+    MTest_Finalize(err);
+    MPI_Finalize();
 
     return err;
 }