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
|
#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));
}
|
