Add mpich3 test suite, to replace older one.

[simgrid.git] / teshsuite / smpi / mpich3-test / pt2pt / sendrecv2.c

diff --git a/teshsuite/smpi/mpich3-test/pt2pt/sendrecv2.c b/teshsuite/smpi/mpich3-test/pt2pt/sendrecv2.c
new file mode 100644 (file)
index 0000000..f4845c6
--- /dev/null
+++ b/teshsuite/smpi/mpich3-test/pt2pt/sendrecv2.c
@@ -0,0 +1,136 @@
+/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
+/*
+ *
+ *  (C) 2003 by Argonne National Laboratory.
+ *      See COPYRIGHT in top-level directory.
+ */
+#include "mpi.h"
+#include "mpitestconf.h"
+#include <stdio.h>
+#include <stdlib.h>
+#ifdef HAVE_STRING_H
+#include <string.h>
+#endif
+
+static int verbose = 0;
+
+static int parse_args(int argc, char **argv);
+
+int main( int argc, char *argv[] )
+{
+    int i, j, errs = 0;
+    int rank, size;
+    MPI_Datatype newtype;
+    char *buf = NULL;
+
+    MPI_Init(&argc, &argv);
+    parse_args(argc, argv);
+
+    MPI_Comm_rank(MPI_COMM_WORLD, &rank);
+    MPI_Comm_size(MPI_COMM_WORLD, &size);
+
+    if (size < 2) {
+       if (verbose) fprintf(stderr, "comm size must be > 1\n");
+       errs++;
+       goto fn_exit;
+    }
+
+    buf = malloc(64 * 129);
+    if (buf == NULL) {
+       if (verbose) fprintf(stderr, "error allocating buffer\n");
+       errs++;
+       goto fn_exit;
+    }
+
+    for (i = 8; i < 64; i += 4) {
+       MPI_Type_vector(i, 128, 129, MPI_CHAR, &newtype);
+
+       MPI_Type_commit(&newtype);
+       memset(buf, 0, 64*129);
+
+       if (rank == 0) {
+           /* init buffer */
+           for (j=0; j < i; j++) {
+               int k;
+               for (k=0; k < 129; k++) {
+                   buf[129*j + k] = (char) j;
+               }
+           }
+
+           /* send */
+           MPI_Send(buf, 1, newtype, 1, i, MPI_COMM_WORLD);
+       }
+       else if (rank == 1) {
+           /* recv */
+           MPI_Recv(buf, 1, newtype, 0, i, MPI_COMM_WORLD, MPI_STATUS_IGNORE);
+
+           /* check buffer */
+           for (j=0; j < i; j++) {
+               int k;
+               for (k=0; k < 129; k++) {
+                   if (k < 128 && buf[129*j + k] != (char) j) {
+                       if (verbose) fprintf(stderr,
+                                            "(i=%d, pos=%d) should be %d but is %d\n",
+                                            i, 129*j + k, j, (int) buf[129*j + k]);
+                       errs++;
+                   }
+                   else if (k == 128 && buf[129*j + k] != (char) 0) {
+                       if (verbose) fprintf(stderr,
+                                            "(i=%d, pos=%d) should be %d but is %d\n",
+                                            i, 129*j + k, 0, (int) buf[129*j + k]);
+                       errs++;
+                   }
+               }
+           }
+       }
+
+       MPI_Type_free(&newtype);
+    }
+
+    if (rank == 0) {
+       int recv_errs = 0;
+
+       MPI_Recv(&recv_errs, 1, MPI_INT, 1, 0, MPI_COMM_WORLD,
+                MPI_STATUS_IGNORE);
+       if (recv_errs) {
+           if (verbose) fprintf(stderr, "%d errors reported from receiver\n",
+                                recv_errs);
+           errs += recv_errs;
+       }
+    }
+    else if (rank == 1) {
+       MPI_Send(&errs, 1, MPI_INT, 0, 0, MPI_COMM_WORLD);
+    }
+       
+ fn_exit:
+
+    free(buf);
+    /* print message and exit */
+    if (errs) {
+       if (rank == 0) fprintf(stderr, "Found %d errors\n", errs);
+    }
+    else {
+       if (rank == 0) printf(" No Errors\n");
+    }
+    MPI_Finalize();
+    return 0;
+}
+
+static int parse_args(int argc, char **argv)
+{
+    /*
+    int ret;
+
+    while ((ret = getopt(argc, argv, "v")) >= 0)
+    {
+       switch (ret) {
+           case 'v':
+               verbose = 1;
+               break;
+       }
+    }
+    */
+    if (argc > 1 && strcmp(argv[1], "-v") == 0)
+       verbose = 1;
+    return 0;
+}