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
|