summaryrefslogtreecommitdiff
path: root/boost/numeric/odeint/external/mpi/mpi_state.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'boost/numeric/odeint/external/mpi/mpi_state.hpp')
-rw-r--r--boost/numeric/odeint/external/mpi/mpi_state.hpp113
1 files changed, 113 insertions, 0 deletions
diff --git a/boost/numeric/odeint/external/mpi/mpi_state.hpp b/boost/numeric/odeint/external/mpi/mpi_state.hpp
new file mode 100644
index 0000000000..aba179c09c
--- /dev/null
+++ b/boost/numeric/odeint/external/mpi/mpi_state.hpp
@@ -0,0 +1,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