/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/count_if_with_ballot.hpp (2605B)
//---------------------------------------------------------------------------// // Copyright (c) 2013 Kyle Lutz // // 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_COUNT_IF_WITH_BALLOT_HPP #define BOOST_COMPUTE_ALGORITHM_DETAIL_COUNT_IF_WITH_BALLOT_HPP #include #include #include #include #include #include #include namespace boost { namespace compute { namespace detail { template inline size_t count_if_with_ballot(InputIterator first, InputIterator last, Predicate predicate, command_queue &queue) { size_t count = iterator_range_size(first, last); size_t block_size = 32; size_t block_count = count / block_size; if(block_count * block_size != count){ block_count++; } const ::boost::compute::context &context = queue.get_context(); ::boost::compute::vector counts(block_count, context); ::boost::compute::detail::nvidia_popcount popc; ::boost::compute::detail::nvidia_ballot ballot; meta_kernel k("count_if_with_ballot"); k << "const uint gid = get_global_id(0);\n" << "bool value = false;\n" << "if(gid < count)\n" << " value = " << predicate(first[k.var("gid")]) << ";\n" << "uint bits = " << ballot(k.var("value")) << ";\n" << "if(get_local_id(0) == 0)\n" << counts.begin()[k.var("get_group_id(0)") ] << " = " << popc(k.var("bits")) << ";\n"; k.add_set_arg("count", count); k.exec_1d(queue, 0, block_size * block_count, block_size); uint_ result; ::boost::compute::reduce( counts.begin(), counts.end(), &result, queue ); return result; } } // end detail namespace } // end compute namespace } // end boost namespace #endif // BOOST_COMPUTE_ALGORITHM_DETAIL_COUNT_IF_WITH_BALLOT_HPP