123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298 |
- //---------------------------------------------------------------------------//
- // Copyright (c) 2013 Kyle Lutz <kyle.r.lutz@gmail.com>
- //
- // Distributed under 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
- //
- // See http://boostorg.github.com/compute for more information.
- //---------------------------------------------------------------------------//
- #define BOOST_TEST_MODULE TestReduce
- #include <boost/test/unit_test.hpp>
- #include <boost/compute/lambda.hpp>
- #include <boost/compute/system.hpp>
- #include <boost/compute/functional.hpp>
- #include <boost/compute/algorithm/reduce.hpp>
- #include <boost/compute/container/vector.hpp>
- #include <boost/compute/iterator/constant_iterator.hpp>
- #include <boost/compute/iterator/counting_iterator.hpp>
- #include <boost/compute/iterator/transform_iterator.hpp>
- #include <boost/compute/types/complex.hpp>
- #include "check_macros.hpp"
- #include "context_setup.hpp"
- namespace compute = boost::compute;
- BOOST_AUTO_TEST_CASE(reduce_int)
- {
- int data[] = { 1, 5, 9, 13, 17 };
- compute::vector<int> vector(data, data + 5, queue);
- int sum;
- compute::reduce(vector.begin(), vector.end(), &sum, compute::plus<int>(), queue);
- BOOST_CHECK_EQUAL(sum, 45);
- int product;
- compute::reduce(vector.begin(), vector.end(), &product, compute::multiplies<int>(), queue);
- BOOST_CHECK_EQUAL(product, 9945);
- }
- BOOST_AUTO_TEST_CASE(reduce_empty_vector)
- {
- compute::vector<short> vector(context);
- short sum = 0;
- compute::reduce(vector.begin(), vector.end(), &sum, queue);
- BOOST_CHECK_EQUAL(sum, short(0));
- }
- BOOST_AUTO_TEST_CASE(reduce_doctest)
- {
- int data[] = { 1, 2, 3, 4 };
- boost::compute::vector<int> vec(data, data + 4, queue);
- //! [sum_int]
- int sum = 0;
- boost::compute::reduce(vec.begin(), vec.end(), &sum, queue);
- //! [sum_int]
- BOOST_CHECK_EQUAL(sum, 10);
- }
- BOOST_AUTO_TEST_CASE(reduce_twos)
- {
- using compute::uint_;
- compute::vector<uint_> vector(8, context);
- compute::fill(vector.begin(), vector.end(), uint_(2), queue);
- uint_ sum;
- compute::reduce(vector.begin(), vector.end(), &sum, compute::plus<uint_>(), queue);
- BOOST_CHECK_EQUAL(sum, uint_(16));
- uint_ product;
- compute::reduce(vector.begin(), vector.end(), &product, compute::multiplies<uint_>(), queue);
- BOOST_CHECK_EQUAL(product, uint_(256));
- }
- BOOST_AUTO_TEST_CASE(reduce_on_device)
- {
- int data[] = { 1, 2, 3, 4, 5, 6, 7, 8 };
- compute::vector<int> input(data, data + 8, queue);
- compute::vector<int> result(2, context);
- compute::reduce(input.begin(), input.begin() + 4, result.begin(), queue);
- compute::reduce(input.begin() + 4, input.end(), result.end() - 1, queue);
- CHECK_RANGE_EQUAL(int, 2, result, (10, 26));
- }
- BOOST_AUTO_TEST_CASE(reduce_int_min_max)
- {
- int data[] = { 11, 5, 92, 13, 42 };
- compute::vector<int> vector(data, data + 5, queue);
- int min_value;
- compute::reduce(
- vector.begin(),
- vector.end(),
- &min_value,
- compute::min<int>(),
- queue
- );
- BOOST_CHECK_EQUAL(min_value, 5);
- int max_value;
- compute::reduce(
- vector.begin(),
- vector.end(),
- &max_value,
- compute::max<int>(),
- queue
- );
- BOOST_CHECK_EQUAL(max_value, 92);
- }
- BOOST_AUTO_TEST_CASE(reduce_int2)
- {
- std::vector<compute::int2_> data;
- for(int i = 0; i < 6; i++){
- compute::int2_ value;
- value[0] = i + 1;
- value[1] = 2 * i + 1;
- data.push_back(value);
- }
- compute::vector<compute::int2_> vector(data.begin(), data.end(), queue);
- compute::int2_ sum;
- compute::reduce(
- vector.begin(),
- vector.end(),
- &sum,
- queue
- );
- BOOST_CHECK_EQUAL(sum, compute::int2_(21, 36));
- }
- BOOST_AUTO_TEST_CASE(reduce_pinned_vector)
- {
- int data[] = { 2, 5, 8, 11, 15 };
- std::vector<int> vector(data, data + 5);
- compute::buffer buffer(context,
- vector.size() * sizeof(int),
- compute::buffer::read_only | compute::buffer::use_host_ptr,
- &vector[0]);
- int sum;
- compute::reduce(
- compute::make_buffer_iterator<int>(buffer, 0),
- compute::make_buffer_iterator<int>(buffer, 5),
- &sum,
- compute::plus<int>()
- );
- BOOST_CHECK_EQUAL(sum, 41);
- }
- BOOST_AUTO_TEST_CASE(reduce_constant_iterator)
- {
- int result;
- compute::reduce(
- compute::make_constant_iterator(1, 0),
- compute::make_constant_iterator(1, 5),
- &result,
- queue
- );
- BOOST_CHECK_EQUAL(result, 5);
- compute::reduce(
- compute::make_constant_iterator(3, 0),
- compute::make_constant_iterator(3, 5),
- &result,
- queue
- );
- BOOST_CHECK_EQUAL(result, 15);
- compute::reduce(
- compute::make_constant_iterator(2, 0),
- compute::make_constant_iterator(2, 5),
- &result,
- compute::multiplies<int>(),
- queue
- );
- BOOST_CHECK_EQUAL(result, 32);
- }
- BOOST_AUTO_TEST_CASE(reduce_counting_iterator)
- {
- int result;
- compute::reduce(
- compute::make_counting_iterator(1),
- compute::make_counting_iterator(10),
- &result,
- queue
- );
- BOOST_CHECK_EQUAL(result, 45);
- compute::reduce(
- compute::make_counting_iterator(1),
- compute::make_counting_iterator(5),
- &result,
- compute::multiplies<int>(),
- queue
- );
- BOOST_CHECK_EQUAL(result, 24);
- }
- BOOST_AUTO_TEST_CASE(reduce_transform_iterator)
- {
- using ::boost::compute::_1;
- int data[] = { 1, 3, 5, 7, 9 };
- compute::vector<int> vector(data, data + 5, queue);
- int sum;
- compute::reduce(
- compute::make_transform_iterator(vector.begin(), _1 + 1),
- compute::make_transform_iterator(vector.end(), _1 + 1),
- &sum,
- queue
- );
- BOOST_CHECK_EQUAL(sum, 30);
- compute::reduce(
- compute::make_transform_iterator(vector.begin(), _1 > 4),
- compute::make_transform_iterator(vector.end(), _1 > 4),
- &sum,
- compute::plus<int>(),
- queue
- );
- BOOST_CHECK_EQUAL(sum, 3);
- compute::reduce(
- compute::make_transform_iterator(vector.begin(), _1 * _1),
- compute::make_transform_iterator(vector.end(), _1 * _1),
- &sum,
- queue
- );
- BOOST_CHECK_EQUAL(sum, 165);
- }
- BOOST_AUTO_TEST_CASE(reduce_complex)
- {
- std::vector<std::complex<float> > data;
- data.push_back(std::complex<float>(1, 2));
- data.push_back(std::complex<float>(2, 4));
- data.push_back(std::complex<float>(3, 6));
- data.push_back(std::complex<float>(4, 8));
- compute::vector<std::complex<float> > vector(data.size(), context);
- compute::copy(data.begin(), data.end(), vector.begin(), queue);
- std::complex<float> result;
- compute::reduce(
- vector.begin(), vector.end(), &result, queue
- );
- BOOST_CHECK(result == std::complex<float>(10, 20));
- compute::reduce(
- vector.begin(), vector.end(), &result, compute::plus<std::complex<float> >(), queue
- );
- BOOST_CHECK(result == std::complex<float>(10, 20));
- compute::reduce(
- vector.begin(), vector.end(), &result, compute::multiplies<std::complex<float> >(), queue
- );
- BOOST_CHECK(result == std::complex<float>(-168, -576));
- }
- BOOST_AUTO_TEST_CASE(reduce_uchar_to_float)
- {
- compute::vector<compute::uchar_> data(context);
- data.push_back(250, queue);
- data.push_back(250, queue);
- float sum = 0;
- compute::reduce(data.begin(), data.end(), &sum, compute::plus<float>(), queue);
- BOOST_CHECK_EQUAL(sum, 500);
- }
- // Test case for https://github.com/boostorg/compute/issues/746
- BOOST_AUTO_TEST_CASE(buffer_reference_count_test)
- {
- using compute::uint_;
- compute::vector<uint_> vector(8, context);
- const compute::buffer& b = vector.get_buffer();
- uint_ rc1 = b.get_info<CL_MEM_REFERENCE_COUNT>();
- {
- compute::fill(vector.begin(), vector.end(), uint_(2), queue);
- uint_ product;
- compute::reduce(vector.begin(), vector.end(), &product, compute::multiplies<uint_>(), queue);
- BOOST_CHECK_EQUAL(product, uint_(256));
- }
- uint_ rc2 = b.get_info<CL_MEM_REFERENCE_COUNT>();
- BOOST_CHECK_EQUAL(rc1, rc2);
- }
- BOOST_AUTO_TEST_SUITE_END()
|