114 lines
3.0 KiB
Plaintext
114 lines
3.0 KiB
Plaintext
/*
|
|
[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
|