boost/mpi
2
0
Fork 0
mirror of https://github.com/boostorg/mpi.git synced 2026-02-25 16:32:22 +00:00
Code Issues Projects Releases Wiki Activity

Add allocator support to vector send recv api.

Browse Source 
Added test case to send recv a vector of udt with an
overload of get_mpi_datatype.  Added test to Jamfile
for nightly testing.
This commit is contained in:
K. Noel Belcourt
2016-07-22 12:14:58 -06:00
parent 333508581f
commit e2ebac2b4d
3 changed files with 123 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

68
include/boost/mpi/communicator.hpp
View File

@@ -1,4 +1,5 @@
// Copyright (C) 2005, 2006 Douglas Gregor <doug.gregor -at- gmail.com>.
// Copyright (C) 2016 K. Noel Belcourt <kbelco -at- sandia.gov>.
// Use, modification and distribution is subject to the Boost Software
// License, Version 1.0. (See accompanying file LICENSE_1_0.txt or copy at
@@ -23,6 +24,7 @@
#include <utility>
#include <iterator>
#include <stdexcept> // for std::range_error
#include <vector>
// For (de-)serializing sends and receives
#include <boost/mpi/packed_oarchive.hpp>
@@ -272,14 +274,16 @@ class BOOST_MPI_DECL communicator
template<typename T>
void send(int dest, int tag, const T& value) const;
template<typename T>
void send(int dest, int tag, const std::vector<T>& value) const;
template<typename T, typename A>
void send(int dest, int tag, const std::vector<T,A>& value) const;
template<typename T>
void send_vector(int dest, int tag, const std::vector<T>& value, mpl::true_) const;
template<typename T, typename A>
void send_vector(int dest, int tag, const std::vector<T,A>& value,
mpl::true_) const;
template<typename T>
void send_vector(int dest, int tag, const std::vector<T>& value, mpl::false_) const;
template<typename T, typename A>
void send_vector(int dest, int tag, const std::vector<T,A>& value,
mpl::false_) const;
/**
* @brief Send the skeleton of an object.
@@ -393,16 +397,16 @@ class BOOST_MPI_DECL communicator
template<typename T>
status recv(int source, int tag, T& value) const;
template<typename T>
status recv(int source, int tag, std::vector<T>& value) const;
template<typename T, typename A>
status recv(int source, int tag, std::vector<T,A>& value) const;
template<typename T>
status recv_vector(int source, int tag, std::vector<T>& value
, mpl::true_) const;
template<typename T, typename A>
status recv_vector(int source, int tag, std::vector<T,A>& value,
mpl::true_) const;
template<typename T>
status recv_vector(int source, int tag, std::vector<T>& value
, mpl::false_) const;
template<typename T, typename A>
status recv_vector(int source, int tag, std::vector<T,A>& value,
mpl::false_) const;
/**
* @brief Receive a skeleton from a remote process.
@@ -1207,26 +1211,26 @@ communicator::array_send_impl(int dest, int tag, const T* values, int n,
send(dest, tag, oa);
}
template<typename T>
void communicator::send_vector(int dest, int tag, const std::vector<T>& value
, mpl::true_ true_type) const
template<typename T, typename A>
void communicator::send_vector(int dest, int tag,
const std::vector<T,A>& value, mpl::true_ true_type) const
{
// send the vector size
typename std::vector<T>::size_type size = value.size();
typename std::vector<T,A>::size_type size = value.size();
send(dest, tag, size);
// send the data
this->array_send_impl(dest, tag, value.data(), size, true_type);
}
template<typename T>
void communicator::send_vector(int dest, int tag, const std::vector<T>& value
, mpl::false_ false_type) const
template<typename T, typename A>
void communicator::send_vector(int dest, int tag,
const std::vector<T,A>& value, mpl::false_ false_type) const
{
this->send_impl(dest, tag, value, false_type);
}
template<typename T>
void communicator::send(int dest, int tag, const std::vector<T>& value) const
template<typename T, typename A>
void communicator::send(int dest, int tag, const std::vector<T,A>& value) const
{
send_vector(dest, tag, value, is_mpi_datatype<T>());
}
@@ -1313,12 +1317,12 @@ communicator::array_recv_impl(int source, int tag, T* values, int n,
return stat;
}
template<typename T>
status communicator::recv_vector(int source, int tag, std::vector<T>& value
, mpl::true_ true_type) const
template<typename T, typename A>
status communicator::recv_vector(int source, int tag,
std::vector<T,A>& value, mpl::true_ true_type) const
{
// receive the vector size
typename std::vector<T>::size_type size = 0;
typename std::vector<T,A>::size_type size = 0;
recv(source, tag, size);
// size the vector
value.resize(size);
@@ -1326,15 +1330,15 @@ status communicator::recv_vector(int source, int tag, std::vector<T>& value
return this->array_recv_impl(source, tag, value.data(), size, true_type);
}
template<typename T>
status communicator::recv_vector(int source, int tag, std::vector<T>& value
, mpl::false_ false_type) const
template<typename T, typename A>
status communicator::recv_vector(int source, int tag,
std::vector<T,A>& value, mpl::false_ false_type) const
{
return this->recv_impl(source, tag, value, false_type);
}
template<typename T>
status communicator::recv(int source, int tag, std::vector<T>& value) const
template<typename T, typename A>
status communicator::recv(int source, int tag, std::vector<T,A>& value) const
{
return recv_vector(source, tag, value, is_mpi_datatype<T>());
}