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
|
// Copyright (C) 2018 Steffen Hirschmann
// Use, modification and distribution is subject to 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)
// Test any_source on serialized data
#include <vector>
#include <iostream>
#include <iterator>
#include <boost/mpi.hpp>
#include <boost/serialization/vector.hpp>
#define BOOST_TEST_MODULE mpi_non_blockin_any_source
#include <boost/test/included/unit_test.hpp>
namespace mpi = boost::mpi;
std::string ok(bool b) {
return b ? "ok" : "ko";
}
BOOST_AUTO_TEST_CASE(non_blocking_any)
{
mpi::environment env;
mpi::communicator world;
int rank = world.rank();
#if BOOST_MPI_VERSION < 3
if (rank == 0) {
std::cout << "\nExpected failure with MPI standard < 3 ("
<< BOOST_MPI_VERSION << "." << BOOST_MPI_SUBVERSION
<< " detected)\n\n";
}
return;
#endif
if (rank == 0) {
std::vector<boost::mpi::request> req;
std::vector<std::vector<int> > data(world.size() - 1);
for (int i = 1; i < world.size(); ++i) {
req.push_back(world.irecv(mpi::any_source, 0, data[i - 1]));
}
boost::mpi::wait_all(req.begin(), req.end());
std::vector<bool> check(world.size()-1, false);
for (int i = 0; i < world.size() - 1; ++i) {
std::cout << "Process 0 received:" << std::endl;
std::copy(data[i].begin(), data[i].end(), std::ostream_iterator<int>(std::cout, " "));
std::cout << std::endl;
int idx = data[i].size();
BOOST_CHECK(std::equal_range(data[i].begin(), data[i].end(), idx)
== std::make_pair(data[i].begin(), data[i].end()));
check[idx-1] = true;
}
for(int i = 0; i < world.size() - 1; ++i) {
std::cout << "Received from " << i+1 << " is " << ok(check[i]) << '\n';
}
BOOST_CHECK(std::equal_range(check.begin(), check.end(), true)
== std::make_pair(check.begin(), check.end()));
} else {
std::vector<int> vec(rank, rank);
mpi::request req = world.isend(0, 0, vec);
req.wait();
}
}
|
