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