Add test for mpi alltoallv

[simgrid.git] / teshsuite / smpi / alltoallv_coll.c

diff --git a/teshsuite/smpi/alltoallv.c b/teshsuite/smpi/alltoallv_coll.c
similarity index 52%
rename from teshsuite/smpi/alltoallv.c
rename to teshsuite/smpi/alltoallv_coll.c
index f2717b9..0b4c71b 100644 (file)
--- a/teshsuite/smpi/alltoallv.c
+++ b/teshsuite/smpi/alltoallv_coll.c
@@ -44,16 +44,16 @@
    <2> rdisp: (#3):  [0][2][4]
 
    after MPI_Alltoallv :
-   <0> rbuf: (#9):   [0][-1][-2][-3][-4][-5][-6][-7][-8]
-   <1> rbuf: (#9):   [1][101][201][-3][-4][-5][-6][-7][-8]
-   <2> rbuf: (#9):   [3][4][103][104][203][204][-6][-7][-8]
+   <0> rbuf: (#9):   [-1][-1][-1][-1][-1][-1][-1][-1][-1]
+   <1> rbuf: (#9):   [1][101][201][-1][-1][-1][-1][-1][-1]
+   <2> rbuf: (#9):   [3][4][103][104][203][204][-1][-1][-1]
 */
 
 
 static void print_buffer_int(void *buf, int len, char *msg, int rank)
 {
   int tmp, *v;
-  printf("**<%d> %s (#%d): ", rank, msg, len);
+  printf("[%d] %s (#%d): ", rank, msg, len);
   for (tmp = 0; tmp < len; tmp++) {
     v = buf;
     printf("[%d]", v[tmp]);
@@ -67,96 +67,53 @@ int main(int argc, char **argv)
 {
 
   MPI_Comm comm;
-  int *sbuf, *rbuf, *erbuf;
+  int *sbuf, *rbuf;
   int rank, size;
   int *sendcounts, *recvcounts, *rdispls, *sdispls;
-  int i, j, *p, err;
+  int i;
 
   MPI_Init(&argc, &argv);
-  err = 0;
 
   comm = MPI_COMM_WORLD;
 
   /* Create the buffer */
   MPI_Comm_size(comm, &size);
   MPI_Comm_rank(comm, &rank);
-  sbuf = (int *) malloc(size * size * sizeof(int));
-  rbuf = (int *) malloc(size * size * sizeof(int));
-  erbuf = (int *) malloc(size * size * sizeof(int));    // expected
-  if (!sbuf || !rbuf) {
-    fprintf(stderr, "Could not allocated buffers!\n");
-    MPI_Abort(comm, 1);
-  }
+  sbuf = (int *) xbt_malloc(size * size * sizeof(int));
+  rbuf = (int *) xbt_malloc(size * size * sizeof(int));
 
   /* Load up the buffers */
   for (i = 0; i < size * size; i++) {
     sbuf[i] = i + 100 * rank;
-    rbuf[i] = -i;
-    erbuf[i] = -i;
+    rbuf[i] = -1;
   }
 
   /* Create and load the arguments to alltoallv */
-  sendcounts = (int *) malloc(size * sizeof(int));
-  recvcounts = (int *) malloc(size * sizeof(int));
-  rdispls = (int *) malloc(size * sizeof(int));
-  sdispls = (int *) malloc(size * sizeof(int));
-  if (!sendcounts || !recvcounts || !rdispls || !sdispls) {
-    fprintf(stderr, "Could not allocate arg items!\n");
-    MPI_Abort(comm, 1);
-  }
+  sendcounts = (int *) xbt_malloc(size * sizeof(int));
+  recvcounts = (int *) xbt_malloc(size * sizeof(int));
+  rdispls = (int *) xbt_malloc(size * sizeof(int));
+  sdispls = (int *) xbt_malloc(size * sizeof(int));
   for (i = 0; i < size; i++) {
     sendcounts[i] = i;
     recvcounts[i] = rank;
     rdispls[i] = i * rank;
     sdispls[i] = (i * (i + 1)) / 2;
   }
-
-  /* debug */
-  /* 
-     print_buffer_int( sbuf, size*size, strdup("sbuf:"),rank);
-     print_buffer_int( sendcounts, size, strdup("scount:"),rank);
-     print_buffer_int( recvcounts, size, strdup("rcount:"),rank);
-     print_buffer_int( sdispls, size, strdup("sdisp:"),rank);
-     print_buffer_int( rdispls, size, strdup("rdisp:"),rank);
-   */
-
-
-  /* debug : erbuf */
-  /* debug
-     for (i=0; i<size; i++) {
-     for (j=0; j<rank; j++) {
-     *(erbuf+j+ rdispls[i]) = i * 100 + (rank*(rank+1))/2 + j; 
-     }
-     }
-   */
-
-
-  //print_buffer_int( erbuf, size*size, strdup("erbuf:"),rank);
-
+ 
+  print_buffer_int( sbuf, size*size, strdup("sbuf:"),rank);
+  print_buffer_int( sendcounts, size, strdup("scount:"),rank);
+  print_buffer_int( recvcounts, size, strdup("rcount:"),rank);
+  print_buffer_int( sdispls, size, strdup("sdisp:"),rank);
+  print_buffer_int( rdispls, size, strdup("rdisp:"),rank);
+   
   MPI_Alltoallv(sbuf, sendcounts, sdispls, MPI_INT,
                 rbuf, recvcounts, rdispls, MPI_INT, comm);
 
-  // debug: print_buffer_int( rbuf, size*size, strdup("rbuf:"),rank);
-
-
-  /* Check rbuf */
-  for (i = 0; i < size; i++) {
-    p = rbuf + rdispls[i];
-    for (j = 0; j < rank; j++) {
-      if (p[j] != i * 100 + (rank * (rank + 1)) / 2 + j) {
-        fprintf(stderr, "** Error: <%d> got %d expected %d for %dth\n",
-                rank, p[j], (i * (i + 1)) / 2 + j, j);
-        err++;
-      }
-    }
-  }
+  print_buffer_int( rbuf, size*size, strdup("rbuf:"),rank);
 
-  /* Summary */
-  if (err > 0) {
-    printf("<%d> Alltoallv test: failure (%d errors).\n", rank, err);
-  }
+  MPI_Barrier(MPI_COMM_WORLD);  
   if (0 == rank) {
-    printf("* Alltoallv TEST COMPLETE.\n");
+    printf("Alltoallv TEST COMPLETE.\n");
   }
   free(sdispls);
   free(rdispls);
@@ -165,7 +122,6 @@ int main(int argc, char **argv)
   free(rbuf);
   free(sbuf);
 
-  MPI_Barrier(MPI_COMM_WORLD);
   MPI_Finalize();
   return 0;
 }