1 //---------------------------------------------------------------------------//
2 // Copyright (c) 2013 Kyle Lutz <kyle.r.lutz@gmail.com>
3 //
4 // Distributed under the Boost Software License, Version 1.0
5 // See accompanying file LICENSE_1_0.txt or copy at
6 // http://www.boost.org/LICENSE_1_0.txt
7 //
8 // See http://boostorg.github.com/compute for more information.
9 //---------------------------------------------------------------------------//
10
11 #define BOOST_TEST_MODULE TestInnerProduct
12 #include <boost/test/unit_test.hpp>
13
14 #include <boost/compute/system.hpp>
15 #include <boost/compute/algorithm/inner_product.hpp>
16 #include <boost/compute/container/vector.hpp>
17 #include <boost/compute/iterator/counting_iterator.hpp>
18
19 #include "context_setup.hpp"
20
21 namespace bc = boost::compute;
22
BOOST_AUTO_TEST_CASE(inner_product_int)23 BOOST_AUTO_TEST_CASE(inner_product_int)
24 {
25 int data1[] = { 1, 2, 3, 4 };
26 bc::vector<int> input1(data1, data1 + 4, queue);
27
28 int data2[] = { 10, 20, 30, 40 };
29 bc::vector<int> input2(data2, data2 + 4, queue);
30
31 int product = bc::inner_product(input1.begin(),
32 input1.end(),
33 input2.begin(),
34 0,
35 queue);
36 BOOST_CHECK_EQUAL(product, 300);
37 }
38
BOOST_AUTO_TEST_CASE(inner_product_counting_iterator)39 BOOST_AUTO_TEST_CASE(inner_product_counting_iterator)
40 {
41 BOOST_CHECK_EQUAL(
42 boost::compute::inner_product(
43 boost::compute::make_counting_iterator<int>(0),
44 boost::compute::make_counting_iterator<int>(100),
45 boost::compute::make_counting_iterator<int>(0),
46 0,
47 queue
48 ),
49 328350
50 );
51 }
52
53 BOOST_AUTO_TEST_SUITE_END()
54