| 1 | // Copyright 2009-2021 Intel Corporation |
| 2 | // SPDX-License-Identifier: Apache-2.0 |
| 3 | |
| 4 | #pragma once |
| 5 | |
| 6 | #include <functional> |
| 7 | #include "parallel_reduce.h" |
| 8 | |
| 9 | namespace embree |
| 10 | { |
| 11 | |
| 12 | template<typename Index, class UnaryPredicate> |
| 13 | __forceinline bool parallel_any_of (Index first, Index last, UnaryPredicate pred) |
| 14 | { |
| 15 | bool ret = false; |
| 16 | |
| 17 | #if defined(TASKING_TBB) |
| 18 | #if TBB_INTERFACE_VERSION >= 12002 |
| 19 | tbb::task_group_context context; |
| 20 | tbb::parallel_for(tbb::blocked_range<size_t>{first, last}, [&ret,pred,&context](const tbb::blocked_range<size_t>& r) { |
| 21 | if (context.is_group_execution_cancelled()) return; |
| 22 | for (size_t i = r.begin(); i != r.end(); ++i) { |
| 23 | if (pred(i)) { |
| 24 | ret = true; |
| 25 | context.cancel_group_execution(); |
| 26 | } |
| 27 | } |
| 28 | }); |
| 29 | #else |
| 30 | tbb::parallel_for(tbb::blocked_range<size_t>{first, last}, [&ret,pred](const tbb::blocked_range<size_t>& r) { |
| 31 | if (tbb::task::self().is_cancelled()) return; |
| 32 | for (size_t i = r.begin(); i != r.end(); ++i) { |
| 33 | if (pred(i)) { |
| 34 | ret = true; |
| 35 | tbb::task::self().cancel_group_execution(); |
| 36 | } |
| 37 | } |
| 38 | }); |
| 39 | #endif |
| 40 | #else |
| 41 | ret = parallel_reduce (first, last, false, [pred](const range<size_t>& r)->bool { |
| 42 | bool localret = false; |
| 43 | for (auto i=r.begin(); i<r.end(); ++i) { |
| 44 | localret |= pred(i); |
| 45 | } |
| 46 | return localret; |
| 47 | }, |
| 48 | std::bit_or<bool>() |
| 49 | ); |
| 50 | #endif |
| 51 | |
| 52 | return ret; |
| 53 | } |
| 54 | |
| 55 | } // end namespace |
| 56 | |