annotate DEPENDENCIES/generic/include/boost/mpi/collectives/broadcast.hpp @ 125:34e428693f5d vext

Vext -> Repoint
author Chris Cannam
date Thu, 14 Jun 2018 11:15:39 +0100
parents 2665513ce2d3
children
rev   line source
Chris@16 1 // Copyright (C) 2005, 2006 Douglas Gregor <doug.gregor -at- gmail.com>.
Chris@16 2
Chris@16 3 // Use, modification and distribution is subject to the Boost Software
Chris@16 4 // License, Version 1.0. (See accompanying file LICENSE_1_0.txt or copy at
Chris@16 5 // http://www.boost.org/LICENSE_1_0.txt)
Chris@16 6
Chris@16 7 // Message Passing Interface 1.1 -- Section 4.4. Broadcast
Chris@16 8 #ifndef BOOST_MPI_BROADCAST_HPP
Chris@16 9 #define BOOST_MPI_BROADCAST_HPP
Chris@16 10
Chris@16 11 #include <boost/mpi/collectives_fwd.hpp>
Chris@16 12 #include <boost/mpi/exception.hpp>
Chris@16 13 #include <boost/mpi/datatype.hpp>
Chris@16 14 #include <boost/mpi/communicator.hpp>
Chris@16 15
Chris@16 16 namespace boost { namespace mpi {
Chris@16 17
Chris@16 18 /************************************************************************
Chris@16 19 * Specializations *
Chris@16 20 ************************************************************************/
Chris@16 21
Chris@16 22 /**
Chris@16 23 * INTERNAL ONLY
Chris@16 24 */
Chris@16 25 template<>
Chris@16 26 BOOST_MPI_DECL void
Chris@16 27 broadcast<const packed_oarchive>(const communicator& comm,
Chris@16 28 const packed_oarchive& oa,
Chris@16 29 int root);
Chris@16 30
Chris@16 31 /**
Chris@16 32 * INTERNAL ONLY
Chris@16 33 */
Chris@16 34 template<>
Chris@16 35 BOOST_MPI_DECL void
Chris@16 36 broadcast<packed_oarchive>(const communicator& comm, packed_oarchive& oa,
Chris@16 37 int root);
Chris@16 38
Chris@16 39 /**
Chris@16 40 * INTERNAL ONLY
Chris@16 41 */
Chris@16 42 template<>
Chris@16 43 BOOST_MPI_DECL void
Chris@16 44 broadcast<packed_iarchive>(const communicator& comm, packed_iarchive& ia,
Chris@16 45 int root);
Chris@16 46
Chris@16 47 /**
Chris@16 48 * INTERNAL ONLY
Chris@16 49 */
Chris@16 50 template<>
Chris@16 51 BOOST_MPI_DECL void
Chris@16 52 broadcast<const packed_skeleton_oarchive>(const communicator& comm,
Chris@16 53 const packed_skeleton_oarchive& oa,
Chris@16 54 int root);
Chris@16 55
Chris@16 56 /**
Chris@16 57 * INTERNAL ONLY
Chris@16 58 */
Chris@16 59 template<>
Chris@16 60 void
Chris@16 61 broadcast<packed_skeleton_oarchive>(const communicator& comm,
Chris@16 62 packed_skeleton_oarchive& oa, int root);
Chris@16 63
Chris@16 64 /**
Chris@16 65 * INTERNAL ONLY
Chris@16 66 */
Chris@16 67 template<>
Chris@16 68 void
Chris@16 69 broadcast<packed_skeleton_iarchive>(const communicator& comm,
Chris@16 70 packed_skeleton_iarchive& ia, int root);
Chris@16 71
Chris@16 72 /**
Chris@16 73 * INTERNAL ONLY
Chris@16 74 */
Chris@16 75 template<>
Chris@16 76 void broadcast<content>(const communicator& comm, content& c, int root);
Chris@16 77
Chris@16 78 /**
Chris@16 79 * INTERNAL ONLY
Chris@16 80 */
Chris@16 81 template<>
Chris@16 82 void broadcast<const content>(const communicator& comm, const content& c,
Chris@16 83 int root);
Chris@16 84
Chris@16 85 /************************************************************************
Chris@16 86 * broadcast() implementation *
Chris@16 87 ************************************************************************/
Chris@16 88 namespace detail {
Chris@16 89 // We're sending a type that has an associated MPI datatype, so
Chris@16 90 // we'll use MPI_Bcast to do all of the work.
Chris@16 91 template<typename T>
Chris@16 92 void
Chris@16 93 broadcast_impl(const communicator& comm, T* values, int n, int root,
Chris@16 94 mpl::true_)
Chris@16 95 {
Chris@16 96 BOOST_MPI_CHECK_RESULT(MPI_Bcast,
Chris@16 97 (values, n,
Chris@16 98 boost::mpi::get_mpi_datatype<T>(*values),
Chris@16 99 root, MPI_Comm(comm)));
Chris@16 100 }
Chris@16 101
Chris@16 102 // We're sending a type that does not have an associated MPI
Chris@16 103 // datatype, so we'll need to serialize it. Unfortunately, this
Chris@16 104 // means that we cannot use MPI_Bcast, so we'll just send from the
Chris@16 105 // root to everyone else.
Chris@16 106 template<typename T>
Chris@16 107 void
Chris@16 108 broadcast_impl(const communicator& comm, T* values, int n, int root,
Chris@16 109 mpl::false_)
Chris@16 110 {
Chris@16 111 if (comm.rank() == root) {
Chris@16 112 packed_oarchive oa(comm);
Chris@16 113 for (int i = 0; i < n; ++i)
Chris@16 114 oa << values[i];
Chris@16 115 broadcast(comm, oa, root);
Chris@16 116 } else {
Chris@16 117 packed_iarchive ia(comm);
Chris@16 118 broadcast(comm, ia, root);
Chris@16 119 for (int i = 0; i < n; ++i)
Chris@16 120 ia >> values[i];
Chris@16 121 }
Chris@16 122 }
Chris@16 123 } // end namespace detail
Chris@16 124
Chris@16 125 template<typename T>
Chris@16 126 void broadcast(const communicator& comm, T& value, int root)
Chris@16 127 {
Chris@16 128 detail::broadcast_impl(comm, &value, 1, root, is_mpi_datatype<T>());
Chris@16 129 }
Chris@16 130
Chris@16 131 template<typename T>
Chris@16 132 void broadcast(const communicator& comm, T* values, int n, int root)
Chris@16 133 {
Chris@16 134 detail::broadcast_impl(comm, values, n, root, is_mpi_datatype<T>());
Chris@16 135 }
Chris@16 136
Chris@16 137 } } // end namespace boost::mpi
Chris@16 138
Chris@16 139 // If the user has already included skeleton_and_content.hpp, include
Chris@16 140 // the code to broadcast skeletons and content.
Chris@16 141 #ifdef BOOST_MPI_SKELETON_AND_CONTENT_HPP
Chris@16 142 # include <boost/mpi/detail/broadcast_sc.hpp>
Chris@16 143 #endif
Chris@16 144
Chris@16 145 #endif // BOOST_MPI_BROADCAST_HPP