123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263 |
- // 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();
- }
- }
|