Upgrade coll mpich testlist to new mpich

[simgrid.git] / teshsuite / smpi / mpich3-test / coll / bcast.c

diff --git a/teshsuite/smpi/mpich3-test/coll/bcast.c b/teshsuite/smpi/mpich3-test/coll/bcast.c
new file mode 100644 (file)
index 0000000..f465d01
--- /dev/null
+++ b/teshsuite/smpi/mpich3-test/coll/bcast.c
@@ -0,0 +1,93 @@
+/* -*- 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 <stdio.h>
+#include "mpitest.h"
+
+/*
+static char MTEST_Descrip[] = "Test of broadcast with various roots and datatypes";
+*/
+
+int main(int argc, char *argv[])
+{
+    int errs = 0, err;
+    int rank, size, root;
+    int minsize = 2, count;
+    MPI_Comm comm;
+    MTestDatatype sendtype, recvtype;
+
+    MTest_Init(&argc, &argv);
+
+    /* The following illustrates the use of the routines to
+     * run through a selection of communicators and datatypes.
+     * Use subsets of these for tests that do not involve combinations
+     * of communicators, datatypes, and counts of datatypes */
+    while (MTestGetIntracommGeneral(&comm, minsize, 1)) {
+        if (comm == MPI_COMM_NULL)
+            continue;
+
+#if defined BCAST_COMM_WORLD_ONLY
+        if (comm != MPI_COMM_WORLD) {
+            MTestFreeComm(&comm);
+            continue;
+        }
+#endif /* BCAST_COMM_WORLD_ONLY */
+
+        /* Determine the sender and receiver */
+        MPI_Comm_rank(comm, &rank);
+        MPI_Comm_size(comm, &size);
+
+        /* To improve reporting of problems about operations, we
+         * change the error handler to errors return */
+        MPI_Errhandler_set(comm, MPI_ERRORS_RETURN);
+
+        MTEST_DATATYPE_FOR_EACH_COUNT(count) {
+
+            /* To shorten test time, only run the default version of datatype tests
+             * for comm world and run the minimum version for other communicators. */
+#if defined BCAST_MIN_DATATYPES_ONLY
+            MTestInitMinDatatypes();
+#endif /* BCAST_MIN_DATATYPES_ONLY */
+
+            while (MTestGetDatatypes(&sendtype, &recvtype, count)) {
+                for (root = 0; root < size; root++) {
+                    if (rank == root) {
+                        sendtype.InitBuf(&sendtype);
+                        err = MPI_Bcast(sendtype.buf, sendtype.count,
+                                        sendtype.datatype, root, comm);
+                        if (err) {
+                            errs++;
+                            MTestPrintError(err);
+                        }
+                    }
+                    else {
+                        recvtype.InitBuf(&recvtype);
+                        err = MPI_Bcast(recvtype.buf, recvtype.count,
+                                        recvtype.datatype, root, comm);
+                        if (err) {
+                            errs++;
+                            fprintf(stderr, "Error with communicator %s and datatype %s\n",
+                                    MTestGetIntracommName(), MTestGetDatatypeName(&recvtype));
+                            MTestPrintError(err);
+                        }
+                        err = MTestCheckRecv(0, &recvtype);
+                        if (err) {
+                            errs += errs;
+                        }
+                    }
+                }
+                MTestFreeDatatype(&recvtype);
+                MTestFreeDatatype(&sendtype);
+            }
+        }
+        MTestFreeComm(&comm);
+    }
+
+    MTest_Finalize(errs);
+    MPI_Finalize();
+    return 0;
+}