/usr/include/boost/compute/algorithm/detail
NameSizeModeActions
balanced_path.hpp62670644editdlrm
binary_find.hpp46260644editdlrm
compact.hpp21880644editdlrm
copy_on_device.hpp73820644editdlrm
copy_to_device.hpp67310644editdlrm
copy_to_host.hpp66580644editdlrm
count_if_with_ballot.hpp26050644editdlrm
count_if_with_reduce.hpp26480644editdlrm
count_if_with_threads.hpp42410644editdlrm
find_extrema.hpp24410644editdlrm
find_extrema_on_cpu.hpp53330644editdlrm
find_extrema_with_atomics.hpp42510644editdlrm
find_extrema_with_reduce.hpp186720644editdlrm
find_if_with_atomics.hpp85090644editdlrm
inplace_reduce.hpp48390644editdlrm
insertion_sort.hpp60300644editdlrm
merge_path.hpp38740644editdlrm
merge_sort_on_cpu.hpp146540644editdlrm
merge_sort_on_gpu.hpp220930644editdlrm
merge_with_merge_path.hpp73390644editdlrm
radix_sort.hpp155020644editdlrm
random_fill.hpp18800644editdlrm
reduce_by_key.hpp52430644editdlrm
reduce_by_key_with_scan.hpp237350644editdlrm
reduce_on_cpu.hpp39490644editdlrm
reduce_on_gpu.hpp103910644editdlrm
scan.hpp15450644editdlrm
scan_on_cpu.hpp70560644editdlrm
scan_on_gpu.hpp113200644editdlrm
search_all.hpp25790644editdlrm
serial_accumulate.hpp19900644editdlrm
serial_count_if.hpp22430644editdlrm
serial_find_extrema.hpp31510644editdlrm
serial_merge.hpp35040644editdlrm
serial_reduce.hpp21450644editdlrm
serial_reduce_by_key.hpp42400644editdlrm
serial_scan.hpp31840644editdlrm
Edit: /usr/include/boost/compute/algorithm/detail/merge_path.hpp (3874B)
//---------------------------------------------------------------------------// // Copyright (c) 2014 Roshan // // 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. //---------------------------------------------------------------------------// #ifndef BOOST_COMPUTE_ALGORITHM_DETAIL_MERGE_PATH_HPP #define BOOST_COMPUTE_ALGORITHM_DETAIL_MERGE_PATH_HPP #include #include #include #include #include #include #include namespace boost { namespace compute { namespace detail { /// /// \brief Merge Path kernel class /// /// Subclass of meta_kernel to break two sets into tiles according /// to their merge path /// class merge_path_kernel : public meta_kernel { public: unsigned int tile_size; merge_path_kernel() : meta_kernel("merge_path") { tile_size = 4; } template void set_range(InputIterator1 first1, InputIterator1 last1, InputIterator2 first2, InputIterator2 last2, OutputIterator1 result_a, OutputIterator2 result_b, Compare comp) { m_a_count = iterator_range_size(first1, last1); m_a_count_arg = add_arg("a_count"); m_b_count = iterator_range_size(first2, last2); m_b_count_arg = add_arg("b_count"); *this << "uint i = get_global_id(0);\n" << "uint target = (i+1)*" << tile_size << ";\n" << "uint start = max(convert_int(0),convert_int(target)-convert_int(b_count));\n" << "uint end = min(target,a_count);\n" << "uint a_index, b_index;\n" << "while(start("b_index")], first1[expr("a_index")]) << "))\n" << " start = a_index + 1;\n" << " else end = a_index;\n" << "}\n" << result_a[expr("i")] << " = start;\n" << result_b[expr("i")] << " = target - start;\n"; } template void set_range(InputIterator1 first1, InputIterator1 last1, InputIterator2 first2, InputIterator2 last2, OutputIterator1 result_a, OutputIterator2 result_b) { typedef typename std::iterator_traits::value_type value_type; ::boost::compute::less less_than; set_range(first1, last1, first2, last2, result_a, result_b, less_than); } event exec(command_queue &queue) { if((m_a_count + m_b_count)/tile_size == 0) { return event(); } set_arg(m_a_count_arg, uint_(m_a_count)); set_arg(m_b_count_arg, uint_(m_b_count)); return exec_1d(queue, 0, (m_a_count + m_b_count)/tile_size); } private: size_t m_a_count; size_t m_a_count_arg; size_t m_b_count; size_t m_b_count_arg; }; } //end detail namespace } //end compute namespace } //end boost namespace #endif // BOOST_COMPUTE_ALGORITHM_DETAIL_MERGE_PATH_HPP