96 lines
2.6 KiB
C++
96 lines
2.6 KiB
C++
#include <vector>
|
|
#include <iostream>
|
|
#include <iterator>
|
|
#include <typeinfo>
|
|
|
|
#include <boost/mpi.hpp>
|
|
#include <boost/serialization/vector.hpp>
|
|
#include <boost/core/demangle.hpp>
|
|
|
|
//#include "debugger.cpp"
|
|
|
|
#define BOOST_TEST_MODULE mpi_nonblocking
|
|
#include <boost/test/included/unit_test.hpp>
|
|
|
|
namespace mpi = boost::mpi;
|
|
|
|
template<typename T>
|
|
bool test(mpi::communicator const& comm, std::vector<T> const& ref, bool iswap, bool alloc)
|
|
{
|
|
|
|
int rank = comm.rank();
|
|
if (rank == 0) {
|
|
std::cout << "Testing with type " << boost::core::demangle(typeid(T).name()) << '\n';
|
|
if (iswap) {
|
|
std::cout << "Blockin send, non blocking receive.\n";
|
|
} else {
|
|
std::cout << "Non blockin send, blocking receive.\n";
|
|
}
|
|
if (alloc) {
|
|
std::cout << "Explicitly allocate space for the receiver.\n";
|
|
} else {
|
|
std::cout << "Do not explicitly allocate space for the receiver.\n";
|
|
}
|
|
}
|
|
if (rank == 0) {
|
|
std::vector<T> data;
|
|
if (alloc) {
|
|
data.resize(ref.size());
|
|
}
|
|
if (iswap) {
|
|
mpi::request req = comm.irecv(1, 0, data);
|
|
req.wait();
|
|
} else {
|
|
comm.recv(1, 0, data);
|
|
}
|
|
std::cout << "Process 0 received " << data.size() << " elements :" << std::endl;
|
|
std::copy(data.begin(), data.end(), std::ostream_iterator<T>(std::cout, " "));
|
|
std::cout << std::endl;
|
|
std::cout << "While expecting " << ref.size() << " elements :" << std::endl;
|
|
std::copy(ref.begin(), ref.end(), std::ostream_iterator<T>(std::cout, " "));
|
|
std::cout << std::endl;
|
|
return (data == ref);
|
|
} else {
|
|
if (rank == 1) {
|
|
std::vector<T> vec = ref;
|
|
if (iswap) {
|
|
comm.send(0, 0, vec);
|
|
} else {
|
|
mpi::request req = comm.isend(0, 0, vec);
|
|
req.wait();
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(non_blocking)
|
|
{
|
|
mpi::environment env;
|
|
mpi::communicator world;
|
|
|
|
BOOST_TEST_REQUIRE(world.size() > 1);
|
|
|
|
std::vector<int> integers(13); // don't assume we're lucky
|
|
for(int i = 0; i < int(integers.size()); ++i) {
|
|
integers[i] = i;
|
|
}
|
|
|
|
std::vector<std::string> strings(13); // don't assume we're lucky
|
|
for(int i = 0; i < int(strings.size()); ++i) {
|
|
std::ostringstream fmt;
|
|
fmt << "S" << i;
|
|
strings[i] = fmt.str();
|
|
}
|
|
|
|
BOOST_CHECK(test(world, integers, true, true));
|
|
BOOST_CHECK(test(world, integers, true, false));
|
|
BOOST_CHECK(test(world, strings, true, true));
|
|
BOOST_CHECK(test(world, strings, true, false));
|
|
|
|
BOOST_CHECK(test(world, integers, false, true));
|
|
BOOST_CHECK(test(world, integers, false, false));
|
|
BOOST_CHECK(test(world, strings, false, true));
|
|
BOOST_CHECK(test(world, strings, false, false));
|
|
}
|