1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
/*
[auto_generated]
boost/numeric/odeint/external/mpi/mpi_state.hpp
[begin_description]
A generic split state, storing partial data on each node.
[end_description]
Copyright 2013 Karsten Ahnert
Copyright 2013 Mario Mulansky
Copyright 2013 Pascal Germroth
Distributed under the Boost Software License, Version 1.0.
(See accompanying file LICENSE_1_0.txt or
copy at http://www.boost.org/LICENSE_1_0.txt)
*/
#ifndef BOOST_NUMERIC_ODEINT_EXTERNAL_MPI_MPI_STATE_HPP_INCLUDED
#define BOOST_NUMERIC_ODEINT_EXTERNAL_MPI_MPI_STATE_HPP_INCLUDED
#include <vector>
#include <algorithm>
#include <boost/mpi.hpp>
#include <boost/numeric/odeint/util/copy.hpp>
#include <boost/numeric/odeint/util/split.hpp>
#include <boost/numeric/odeint/util/resize.hpp>
#include <boost/numeric/odeint/util/same_size.hpp>
#include <boost/numeric/odeint/algebra/algebra_dispatcher.hpp>
#include <boost/numeric/odeint/external/mpi/mpi_nested_algebra.hpp>
namespace boost {
namespace numeric {
namespace odeint {
/** \brief A container which has its contents distributed among the nodes.
*/
template< class InnerState >
struct mpi_state
{
typedef InnerState value_type;
// the node's local data.
InnerState m_data;
boost::mpi::communicator world;
mpi_state() {}
mpi_state(boost::mpi::communicator comm) : world(comm) {}
inline InnerState &operator()() { return m_data; }
inline const InnerState &operator()() const { return m_data; }
};
template< class InnerState >
struct is_resizeable< mpi_state< InnerState > >
: is_resizeable< InnerState > { };
template< class InnerState1 , class InnerState2 >
struct same_size_impl< mpi_state< InnerState1 > , mpi_state< InnerState2 > >
{
static bool same_size( const mpi_state< InnerState1 > &x , const mpi_state< InnerState2 > &y )
{
const bool local = boost::numeric::odeint::same_size(x(), y());
return boost::mpi::all_reduce(x.world, local, mpi::bitwise_and<bool>());
}
};
template< class InnerState1 , class InnerState2 >
struct resize_impl< mpi_state< InnerState1 > , mpi_state< InnerState2 > >
{
static void resize( mpi_state< InnerState1 > &x , const mpi_state< InnerState2 > &y )
{
// resize local parts on each node.
boost::numeric::odeint::resize(x(), y());
}
};
/** \brief Copy data between mpi_states of same size. */
template< class InnerState1 , class InnerState2 >
struct copy_impl< mpi_state< InnerState1 > , mpi_state< InnerState2 > >
{
static void copy( const mpi_state< InnerState1 > &from , mpi_state< InnerState2 > &to )
{
// copy local parts on each node.
boost::numeric::odeint::copy(from(), to());
}
};
/** \brief Use `mpi_algebra` for `mpi_state`. */
template< class InnerState >
struct algebra_dispatcher< mpi_state< InnerState > >
{
typedef mpi_nested_algebra<
typename algebra_dispatcher< InnerState >::algebra_type
> algebra_type;
};
}
}
}
#endif
|
