X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/0f5e8daaa6e9f74521068aa75837200bcd182ea6..cc01c6a871dbbfd0cc84e26eb9f94aeb539e613e:/examples/smpi/scatter.c

diff --git a/examples/smpi/scatter.c b/examples/smpi/scatter.c
index 8833cea19f..e4a63037e5 100644
--- a/examples/smpi/scatter.c
+++ b/examples/smpi/scatter.c
@@ -24,106 +24,112 @@
 #include <stdio.h>
 #include <mpi.h>
 
-static int ibm_test(int rank, int size)  {
+static int ibm_test(int rank, int size)
+{
 
 #define MAXLEN  10000
 
-	  int success = 1;
-	  int root=0;
-        int i,j,k;
-	  int *out;
-	  int *in;
-        
-        
-        out = malloc(MAXLEN*64*sizeof(int));
-        in  = malloc(MAXLEN*sizeof(int));
-
-	  for(j=1;j<=MAXLEN;j*=10){
-                root=(root+1)%size;
-		    if (rank == root)
-				for(i=0;i<j*size;i++)  
-                                out[i] = i;
-
-		    MPI_Scatter(out,j,MPI_INT,in,j,MPI_INT,root,MPI_COMM_WORLD);
-
-		    for(k=0;k<j;k++) {
-				if(in[k] != k+rank*j) {
-					  fprintf(stderr,"task %d bad answer (%d) at index %d k of %d (should be %d)", rank,in[k],k,j,(k+rank*j));
-					  return( 0 ); 
-				}
-		    }
-	  }
-        free(out);
-        free(in);
-	  MPI_Barrier( MPI_COMM_WORLD );
-        return( success );
+  int success = 1;
+  int root = 0;
+  int i, j, k;
+  int *out;
+  int *in;
+
+
+  out = malloc(MAXLEN * 64 * sizeof(int));
+  in = malloc(MAXLEN * sizeof(int));
+
+  for (j = 1; j <= MAXLEN; j *= 10) {
+    root = (root + 1) % size;
+    if (rank == root)
+      for (i = 0; i < j * size; i++)
+        out[i] = i;
+
+    MPI_Scatter(out, j, MPI_INT, in, j, MPI_INT, root, MPI_COMM_WORLD);
+
+    for (k = 0; k < j; k++) {
+      if (in[k] != k + rank * j) {
+        fprintf(stderr,
+                "task %d bad answer (%d) at index %d k of %d (should be %d)",
+                rank, in[k], k, j, (k + rank * j));
+        return (0);
+      }
+    }
+  }
+  free(out);
+  free(in);
+  MPI_Barrier(MPI_COMM_WORLD);
+  return (success);
 }
 
 /**
  * small test: the root sends a single distinct double to other processes
  **/
-static int small_test( int rank, int size ) {
-        int success=1;
-        int retval;
-        int sendcount=1; // one double to each process
-        int recvcount=1;
-        int i;
-        double *sndbuf =NULL;
-        double rcvd;
-        int root=0; // arbitrary choice 
-
-        // on root, initialize sendbuf
-        if (root == rank )  {
-                sndbuf = malloc( size * sizeof(double));
-                for (i=0; i< size;i++) {
-                        sndbuf[i] = (double) i;
-                }
-        }
-
-        retval=MPI_Scatter(sndbuf, sendcount, MPI_DOUBLE, &rcvd,recvcount,MPI_DOUBLE,root,MPI_COMM_WORLD);
-        if (retval != MPI_SUCCESS) {
-                fprintf(stderr,"(%s:%d) MPI_Scatter() returned retval=%d\n",__FILE__,__LINE__,retval);
-                return 0;
-        }
-
-        // verification
-        if ((double)rank != rcvd) {
-                fprintf(stderr,"[%d] has %lf instead of %d\n",rank,rcvd,rank);
-                success=0;
-        }
-        return(success);
+static int small_test(int rank, int size)
+{
+  int success = 1;
+  int retval;
+  int sendcount = 1;            // one double to each process
+  int recvcount = 1;
+  int i;
+  double *sndbuf = NULL;
+  double rcvd;
+  int root = 0;                 // arbitrary choice 
+
+  // on root, initialize sendbuf
+  if (root == rank) {
+    sndbuf = malloc(size * sizeof(double));
+    for (i = 0; i < size; i++) {
+      sndbuf[i] = (double) i;
+    }
+  }
+
+  retval =
+      MPI_Scatter(sndbuf, sendcount, MPI_DOUBLE, &rcvd, recvcount,
+                  MPI_DOUBLE, root, MPI_COMM_WORLD);
+  if (retval != MPI_SUCCESS) {
+    fprintf(stderr, "(%s:%d) MPI_Scatter() returned retval=%d\n", __FILE__,
+            __LINE__, retval);
+    return 0;
+  }
+  // verification
+  if ((double) rank != rcvd) {
+    fprintf(stderr, "[%d] has %lf instead of %d\n", rank, rcvd, rank);
+    success = 0;
+  }
+  return (success);
 }
 
 
 
 int main(int argc, char **argv)
 {
-        int size, rank;
+  int size, rank;
 
 
-        MPI_Init(&argc, &argv);
-        MPI_Comm_size(MPI_COMM_WORLD, &size);
-        MPI_Comm_rank(MPI_COMM_WORLD, &rank);
+  MPI_Init(&argc, &argv);
+  MPI_Comm_size(MPI_COMM_WORLD, &size);
+  MPI_Comm_rank(MPI_COMM_WORLD, &rank);
 
-        /* test 1 */
-        if (0 == rank)
-                printf("** Small Test Result: ... \n");
-        if (! small_test( rank, size ))
-                printf("\t[%d] failed.\n", rank);
-        else
-                printf("\t[%d] ok.\n", rank);
+  /* test 1 */
+  if (0 == rank)
+    printf("** Small Test Result: ... \n");
+  if (!small_test(rank, size))
+    printf("\t[%d] failed.\n", rank);
+  else
+    printf("\t[%d] ok.\n", rank);
 
 
-        MPI_Barrier( MPI_COMM_WORLD );
+  MPI_Barrier(MPI_COMM_WORLD);
 
-        /* test 2 */
-        if (0 == rank)
-                printf("** IBM Test Result: ... \n");
-        if (!ibm_test(rank, size))
-                printf("\t[%d] failed.\n", rank);
-        else
-                printf("\t[%d] ok.\n", rank);
+  /* test 2 */
+  if (0 == rank)
+    printf("** IBM Test Result: ... \n");
+  if (!ibm_test(rank, size))
+    printf("\t[%d] failed.\n", rank);
+  else
+    printf("\t[%d] ok.\n", rank);
 
-        MPI_Finalize();
-        return 0;
+  MPI_Finalize();
+  return 0;
 }