1 // Copyright (C) 2018 Steffen Hirschmann
2
3 // Use, modification and distribution is subject to the Boost Software
4 // License, Version 1.0. (See accompanying file LICENSE_1_0.txt or copy at
5 // http://www.boost.org/LICENSE_1_0.txt)
6
7 // Test any_source on serialized data
8 #include <vector>
9 #include <iostream>
10 #include <iterator>
11 #include <boost/mpi.hpp>
12 #include <boost/serialization/vector.hpp>
13
14 #define BOOST_TEST_MODULE mpi_non_blockin_any_source
15 #include <boost/test/included/unit_test.hpp>
16
17 namespace mpi = boost::mpi;
18
ok(bool b)19 std::string ok(bool b) {
20 return b ? "ok" : "ko";
21 }
22
BOOST_AUTO_TEST_CASE(non_blocking_any)23 BOOST_AUTO_TEST_CASE(non_blocking_any)
24 {
25 mpi::environment env;
26 mpi::communicator world;
27 int rank = world.rank();
28 #if BOOST_MPI_VERSION < 3
29 if (rank == 0) {
30 std::cout << "\nExpected failure with MPI standard < 3 ("
31 << BOOST_MPI_VERSION << "." << BOOST_MPI_SUBVERSION
32 << " detected)\n\n";
33 }
34 return;
35 #endif
36 if (rank == 0) {
37 std::vector<boost::mpi::request> req;
38 std::vector<std::vector<int> > data(world.size() - 1);
39 for (int i = 1; i < world.size(); ++i) {
40 req.push_back(world.irecv(mpi::any_source, 0, data[i - 1]));
41 }
42 boost::mpi::wait_all(req.begin(), req.end());
43 std::vector<bool> check(world.size()-1, false);
44 for (int i = 0; i < world.size() - 1; ++i) {
45 std::cout << "Process 0 received:" << std::endl;
46 std::copy(data[i].begin(), data[i].end(), std::ostream_iterator<int>(std::cout, " "));
47 std::cout << std::endl;
48 int idx = data[i].size();
49 BOOST_CHECK(std::equal_range(data[i].begin(), data[i].end(), idx)
50 == std::make_pair(data[i].begin(), data[i].end()));
51 check[idx-1] = true;
52 }
53 for(int i = 0; i < world.size() - 1; ++i) {
54 std::cout << "Received from " << i+1 << " is " << ok(check[i]) << '\n';
55 }
56 BOOST_CHECK(std::equal_range(check.begin(), check.end(), true)
57 == std::make_pair(check.begin(), check.end()));
58 } else {
59 std::vector<int> vec(rank, rank);
60 mpi::request req = world.isend(0, 0, vec);
61 req.wait();
62 }
63 }
64