/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/find_extrema_with_atomics.hpp (4251B)
//---------------------------------------------------------------------------// // 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_FIND_EXTREMA_WITH_ATOMICS_HPP #define BOOST_COMPUTE_ALGORITHM_DETAIL_FIND_EXTREMA_WITH_ATOMICS_HPP #include #include #include #include #include #include namespace boost { namespace compute { namespace detail { template inline InputIterator find_extrema_with_atomics(InputIterator first, InputIterator last, Compare compare, const bool find_minimum, command_queue &queue) { typedef typename std::iterator_traits::value_type value_type; typedef typename std::iterator_traits::difference_type difference_type; const context &context = queue.get_context(); meta_kernel k("find_extrema"); atomic_cmpxchg atomic_cmpxchg_uint; k << "const uint gid = get_global_id(0);\n" << "uint old_index = *index;\n" << k.decl("old") << " = " << first[k.var("old_index")] << ";\n" << k.decl("new") << " = " << first[k.var("gid")] << ";\n" << k.decl("compare_result") << ";\n" << "#ifdef BOOST_COMPUTE_FIND_MAXIMUM\n" << "while(" << "(compare_result = " << compare(k.var("old"), k.var("new")) << ")" << " || (!(compare_result" << " || " << compare(k.var("new"), k.var("old")) << ") " "&& gid < old_index)){\n" << "#else\n" << // while condition explained for minimum case with less (<) // as comparison function: // while(new_value < old_value // OR (new_value == old_value AND new_index < old_index)) "while(" << "(compare_result = " << compare(k.var("new"), k.var("old")) << ")" << " || (!(compare_result" << " || " << compare(k.var("old"), k.var("new")) << ") " "&& gid < old_index)){\n" << "#endif\n" << " if(" << atomic_cmpxchg_uint(k.var("index"), k.var("old_index"), k.var("gid")) << " == old_index)\n" << " break;\n" << " else\n" << " old_index = *index;\n" << "old = " << first[k.var("old_index")] << ";\n" << "}\n"; size_t index_arg_index = k.add_arg(memory_object::global_memory, "index"); std::string options; if(!find_minimum){ options = "-DBOOST_COMPUTE_FIND_MAXIMUM"; } kernel kernel = k.compile(context, options); // setup index buffer scalar index(context); kernel.set_arg(index_arg_index, index.get_buffer()); // initialize index index.write(0, queue); // run kernel size_t count = iterator_range_size(first, last); queue.enqueue_1d_range_kernel(kernel, 0, count, 0); // read index and return iterator return first + static_cast(index.read(queue)); } } // end detail namespace } // end compute namespace } // end boost namespace #endif // BOOST_COMPUTE_ALGORITHM_DETAIL_FIND_EXTREMA_WITH_ATOMICS_HPP