123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103 |
- #ifndef BOOST_COMPUTE_ALGORITHM_INNER_PRODUCT_HPP
- #define BOOST_COMPUTE_ALGORITHM_INNER_PRODUCT_HPP
- #include <boost/static_assert.hpp>
- #include <boost/compute/system.hpp>
- #include <boost/compute/functional.hpp>
- #include <boost/compute/command_queue.hpp>
- #include <boost/compute/algorithm/accumulate.hpp>
- #include <boost/compute/container/vector.hpp>
- #include <boost/compute/iterator/transform_iterator.hpp>
- #include <boost/compute/iterator/zip_iterator.hpp>
- #include <boost/compute/functional/detail/unpack.hpp>
- #include <boost/compute/type_traits/is_device_iterator.hpp>
- namespace boost {
- namespace compute {
- template<class InputIterator1, class InputIterator2, class T>
- inline T inner_product(InputIterator1 first1,
- InputIterator1 last1,
- InputIterator2 first2,
- T init,
- command_queue &queue = system::default_queue())
- {
- BOOST_STATIC_ASSERT(is_device_iterator<InputIterator1>::value);
- BOOST_STATIC_ASSERT(is_device_iterator<InputIterator2>::value);
- typedef typename std::iterator_traits<InputIterator1>::value_type input_type;
- ptrdiff_t n = std::distance(first1, last1);
- return ::boost::compute::accumulate(
- ::boost::compute::make_transform_iterator(
- ::boost::compute::make_zip_iterator(
- boost::make_tuple(first1, first2)
- ),
- detail::unpack(multiplies<input_type>())
- ),
- ::boost::compute::make_transform_iterator(
- ::boost::compute::make_zip_iterator(
- boost::make_tuple(last1, first2 + n)
- ),
- detail::unpack(multiplies<input_type>())
- ),
- init,
- queue
- );
- }
- template<class InputIterator1,
- class InputIterator2,
- class T,
- class BinaryAccumulateFunction,
- class BinaryTransformFunction>
- inline T inner_product(InputIterator1 first1,
- InputIterator1 last1,
- InputIterator2 first2,
- T init,
- BinaryAccumulateFunction accumulate_function,
- BinaryTransformFunction transform_function,
- command_queue &queue = system::default_queue())
- {
- BOOST_STATIC_ASSERT(is_device_iterator<InputIterator1>::value);
- BOOST_STATIC_ASSERT(is_device_iterator<InputIterator2>::value);
- typedef typename std::iterator_traits<InputIterator1>::value_type value_type;
- size_t count = detail::iterator_range_size(first1, last1);
- vector<value_type> result(count, queue.get_context());
- transform(first1,
- last1,
- first2,
- result.begin(),
- transform_function,
- queue);
- return ::boost::compute::accumulate(result.begin(),
- result.end(),
- init,
- accumulate_function,
- queue);
- }
- }
- }
- #endif
|