Upgrade MPICH collective selector to 3.3.

[simgrid.git] / src / smpi / colls / barrier / barrier-mpich-smp.cpp

diff --git a/src/smpi/colls/barrier/barrier-mpich-smp.cpp b/src/smpi/colls/barrier/barrier-mpich-smp.cpp
new file mode 100644 (file)
index 0000000..6df4962
--- /dev/null
+++ b/src/smpi/colls/barrier/barrier-mpich-smp.cpp
@@ -0,0 +1,62 @@
+/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
+/*
+ *
+ *  (C) 2001 by Argonne National Laboratory.
+ *      See COPYRIGHT in top-level directory.
+ */
+
+
+#include "../coll_tuned_topo.hpp"
+#include "../colls_private.hpp"
+
+namespace simgrid{
+namespace smpi{
+int Coll_barrier_mpich_smp::barrier(MPI_Comm comm)
+{
+    int mpi_errno = MPI_SUCCESS;
+    int mpi_errno_ret = MPI_SUCCESS;
+    MPI_Comm shmem_comm = MPI_COMM_NULL, leader_comm = MPI_COMM_NULL;
+    int local_rank = -1;
+    
+    if(comm->get_leaders_comm()==MPI_COMM_NULL){
+      comm->init_smp();
+    }
+
+    shmem_comm = comm->get_intra_comm();
+    local_rank = shmem_comm->rank();
+    /* do the intranode barrier on all nodes */
+    if (shmem_comm != NULL) {
+        mpi_errno = Coll_barrier_mpich::barrier(shmem_comm);
+        if (mpi_errno) {
+          mpi_errno_ret+=mpi_errno;
+        }
+    }
+
+    leader_comm = comm->get_leaders_comm();
+    /* do the barrier across roots of all nodes */
+    if (leader_comm != NULL && local_rank == 0) {
+        mpi_errno = Coll_barrier_mpich::barrier(leader_comm);
+        if (mpi_errno) {
+          mpi_errno_ret+=mpi_errno;
+        }
+    }
+
+    /* release the local processes on each node with a 1-byte
+     * broadcast (0-byte broadcast just returns without doing
+     * anything) */
+    if (shmem_comm != NULL) {
+        int i = 0;
+        mpi_errno = Coll_bcast_mpich::bcast(&i, 1, MPI_BYTE, 0, shmem_comm);
+        if (mpi_errno) {
+          mpi_errno_ret+=mpi_errno;
+        }
+    }
+
+    if (mpi_errno_ret)
+        mpi_errno = mpi_errno_ret;
+    return mpi_errno;
+}
+
+}
+}
+