123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207 |
- #ifndef BOOST_COMPUTE_ALGORITHM_SORT_HPP
- #define BOOST_COMPUTE_ALGORITHM_SORT_HPP
- #include <iterator>
- #include <boost/utility/enable_if.hpp>
- #include <boost/compute/system.hpp>
- #include <boost/compute/command_queue.hpp>
- #include <boost/compute/algorithm/detail/merge_sort_on_cpu.hpp>
- #include <boost/compute/algorithm/detail/merge_sort_on_gpu.hpp>
- #include <boost/compute/algorithm/detail/radix_sort.hpp>
- #include <boost/compute/algorithm/detail/insertion_sort.hpp>
- #include <boost/compute/algorithm/reverse.hpp>
- #include <boost/compute/container/mapped_view.hpp>
- #include <boost/compute/detail/iterator_range_size.hpp>
- #include <boost/compute/iterator/buffer_iterator.hpp>
- #include <boost/compute/type_traits/is_device_iterator.hpp>
- namespace boost {
- namespace compute {
- namespace detail {
- template<class T>
- inline void dispatch_gpu_sort(buffer_iterator<T> first,
- buffer_iterator<T> last,
- less<T>,
- command_queue &queue,
- typename boost::enable_if_c<
- is_radix_sortable<T>::value
- >::type* = 0)
- {
- size_t count = detail::iterator_range_size(first, last);
- if(count < 2){
-
- return;
- }
- else if(count <= 32){
- ::boost::compute::detail::serial_insertion_sort(first, last, queue);
- }
- else {
- ::boost::compute::detail::radix_sort(first, last, queue);
- }
- }
- template<class T>
- inline void dispatch_gpu_sort(buffer_iterator<T> first,
- buffer_iterator<T> last,
- greater<T> compare,
- command_queue &queue,
- typename boost::enable_if_c<
- is_radix_sortable<T>::value
- >::type* = 0)
- {
- size_t count = detail::iterator_range_size(first, last);
- if(count < 2){
-
- return;
- }
- else if(count <= 32){
- ::boost::compute::detail::serial_insertion_sort(
- first, last, compare, queue
- );
- }
- else {
-
- ::boost::compute::detail::radix_sort(first, last, false, queue);
- }
- }
- template<class Iterator, class Compare>
- inline void dispatch_gpu_sort(Iterator first,
- Iterator last,
- Compare compare,
- command_queue &queue)
- {
- size_t count = detail::iterator_range_size(first, last);
- if(count < 2){
-
- return;
- }
- else if(count <= 32){
- ::boost::compute::detail::serial_insertion_sort(
- first, last, compare, queue
- );
- }
- else {
- ::boost::compute::detail::merge_sort_on_gpu(
- first, last, compare, queue
- );
- }
- }
- template<class Iterator, class Compare>
- inline void dispatch_sort(Iterator first,
- Iterator last,
- Compare compare,
- command_queue &queue,
- typename boost::enable_if<
- is_device_iterator<Iterator>
- >::type* = 0)
- {
- if(queue.get_device().type() & device::gpu) {
- dispatch_gpu_sort(first, last, compare, queue);
- return;
- }
- ::boost::compute::detail::merge_sort_on_cpu(first, last, compare, queue);
- }
- template<class Iterator, class Compare>
- inline void dispatch_sort(Iterator first,
- Iterator last,
- Compare compare,
- command_queue &queue,
- typename boost::disable_if<
- is_device_iterator<Iterator>
- >::type* = 0)
- {
- typedef typename std::iterator_traits<Iterator>::value_type T;
- size_t size = static_cast<size_t>(std::distance(first, last));
-
- mapped_view<T> view(
- boost::addressof(*first), size, queue.get_context()
- );
-
- dispatch_sort(view.begin(), view.end(), compare, queue);
-
- view.map(queue);
- }
- }
- template<class Iterator, class Compare>
- inline void sort(Iterator first,
- Iterator last,
- Compare compare,
- command_queue &queue = system::default_queue())
- {
- ::boost::compute::detail::dispatch_sort(first, last, compare, queue);
- }
- template<class Iterator>
- inline void sort(Iterator first,
- Iterator last,
- command_queue &queue = system::default_queue())
- {
- typedef typename std::iterator_traits<Iterator>::value_type value_type;
- ::boost::compute::sort(
- first, last, ::boost::compute::less<value_type>(), queue
- );
- }
- }
- }
- #endif
|