123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228 |
- // Copyright (C) 2005, 2006 Douglas Gregor <doug.gregor -at- gmail.com>.
- // 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)
- // A test of the scan() collective.
- #include <boost/mpi/collectives/scan.hpp>
- #include <boost/mpi/communicator.hpp>
- #include <boost/mpi/environment.hpp>
- #include <algorithm>
- #include <boost/serialization/string.hpp>
- #include <boost/iterator/counting_iterator.hpp>
- #include <boost/lexical_cast.hpp>
- #include <numeric>
- #define BOOST_TEST_MODULE mpi_scan_test
- #include <boost/test/included/unit_test.hpp>
- using boost::mpi::communicator;
- // A simple point class that we can build, add, compare, and
- // serialize.
- struct point
- {
- point() : x(0), y(0), z(0) { }
- point(int x, int y, int z) : x(x), y(y), z(z) { }
- int x;
- int y;
- int z;
- private:
- template<typename Archiver>
- void serialize(Archiver& ar, unsigned int /*version*/)
- {
- ar & x & y & z;
- }
- friend class boost::serialization::access;
- };
- std::ostream& operator<<(std::ostream& out, const point& p)
- {
- return out << p.x << ' ' << p.y << ' ' << p.z;
- }
- bool operator==(const point& p1, const point& p2)
- {
- return p1.x == p2.x && p1.y == p2.y && p1.z == p2.z;
- }
- bool operator!=(const point& p1, const point& p2)
- {
- return !(p1 == p2);
- }
- point operator+(const point& p1, const point& p2)
- {
- return point(p1.x + p2.x, p1.y + p2.y, p1.z + p2.z);
- }
- namespace boost { namespace mpi {
- template <>
- struct is_mpi_datatype<point> : public mpl::true_ { };
- } } // end namespace boost::mpi
- template<typename Generator, typename Op>
- void
- scan_test(const communicator& comm, Generator generator,
- const char* type_kind, Op op, const char* op_kind)
- {
- typedef typename Generator::result_type value_type;
- value_type value = generator(comm.rank());
- using boost::mpi::scan;
-
- if (comm.rank() == 0) {
- std::cout << "Prefix reducing to " << op_kind << " of " << type_kind
- << "...";
- std::cout.flush();
- }
- value_type result_value;
- scan(comm, value, result_value, op);
- value_type scan_result = scan(comm, value, op);
- BOOST_CHECK(scan_result == result_value);
- // Compute expected result
- std::vector<value_type> generated_values;
- for (int p = 0; p < comm.size(); ++p)
- generated_values.push_back(generator(p));
- std::vector<value_type> expected_results(comm.size());
- std::partial_sum(generated_values.begin(), generated_values.end(),
- expected_results.begin(), op);
- BOOST_CHECK(result_value == expected_results[comm.rank()]);
- if (comm.rank() == 0) std::cout << "Done." << std::endl;
- (comm.barrier)();
- }
- // Generates integers to test with scan()
- struct int_generator
- {
- typedef int result_type;
- int_generator(int base = 1) : base(base) { }
- int operator()(int p) const { return base + p; }
- private:
- int base;
- };
- // Generate points to test with scan()
- struct point_generator
- {
- typedef point result_type;
- point_generator(point origin) : origin(origin) { }
- point operator()(int p) const
- {
- return point(origin.x + 1, origin.y + 1, origin.z + 1);
- }
- private:
- point origin;
- };
- struct string_generator
- {
- typedef std::string result_type;
- std::string operator()(int p) const
- {
- std::string result = boost::lexical_cast<std::string>(p);
- result += " rosebud";
- if (p != 1) result += 's';
- return result;
- }
- };
- struct secret_int_bit_and
- {
- int operator()(int x, int y) const { return x & y; }
- };
- struct wrapped_int
- {
- wrapped_int() : value(0) { }
- explicit wrapped_int(int value) : value(value) { }
- template<typename Archive>
- void serialize(Archive& ar, unsigned int /* version */)
- {
- ar & value;
- }
- int value;
- };
- wrapped_int operator+(const wrapped_int& x, const wrapped_int& y)
- {
- return wrapped_int(x.value + y.value);
- }
- bool operator==(const wrapped_int& x, const wrapped_int& y)
- {
- return x.value == y.value;
- }
- // Generates wrapped_its to test with scan()
- struct wrapped_int_generator
- {
- typedef wrapped_int result_type;
- wrapped_int_generator(int base = 1) : base(base) { }
- wrapped_int operator()(int p) const { return wrapped_int(base + p); }
- private:
- int base;
- };
- namespace boost { namespace mpi {
- // Make std::plus<wrapped_int> commutative.
- template<>
- struct is_commutative<std::plus<wrapped_int>, wrapped_int>
- : mpl::true_ { };
- } } // end namespace boost::mpi
- BOOST_AUTO_TEST_CASE(scan_check)
- {
- using namespace boost::mpi;
- environment env;
- communicator comm;
- // Built-in MPI datatypes with built-in MPI operations
- scan_test(comm, int_generator(), "integers", std::plus<int>(), "sum");
- scan_test(comm, int_generator(), "integers", std::multiplies<int>(),
- "product");
- scan_test(comm, int_generator(), "integers", maximum<int>(),
- "maximum");
- scan_test(comm, int_generator(), "integers", minimum<int>(),
- "minimum");
- // User-defined MPI datatypes with operations that have the
- // same name as built-in operations.
- scan_test(comm, point_generator(point(0,0,0)), "points",
- std::plus<point>(), "sum");
- // Built-in MPI datatypes with user-defined operations
- scan_test(comm, int_generator(17), "integers", secret_int_bit_and(),
- "bitwise and");
- // Arbitrary types with user-defined, commutative operations.
- scan_test(comm, wrapped_int_generator(17), "wrapped integers",
- std::plus<wrapped_int>(), "sum");
- // Arbitrary types with (non-commutative) user-defined operations
- scan_test(comm, string_generator(), "strings",
- std::plus<std::string>(), "concatenation");
- }
|