7 #ifndef PXR_BASE_WORK_LOOPS_H
8 #define PXR_BASE_WORK_LOOPS_H
15 #include <tbb/blocked_range.h>
16 #include <tbb/parallel_for.h>
17 #include <tbb/parallel_for_each.h>
18 #include <tbb/task_group.h>
38 std::forward<Fn>(fn)(0, n);
56 template <
typename Fn>
66 class Work_ParallelForN_TBB
69 Work_ParallelForN_TBB(Fn &fn) : _fn(fn) { }
71 void operator()(
const tbb::blocked_range<size_t> &
r)
const {
77 std::forward<Fn>(_fn)(r.begin(), r.end());
86 tbb::task_group_context ctx(tbb::task_group_context::isolated);
88 Work_ParallelForN_TBB(callback),
110 template <
typename Fn>
130 template <
typename InputIterator,
typename Fn>
133 InputIterator
first, InputIterator
last, Fn &&fn)
135 tbb::task_group_context ctx(tbb::task_group_context::isolated);
136 tbb::parallel_for_each(first, last, std::forward<Fn>(fn), ctx);
141 #endif // PXR_BASE_WORK_LOOPS_H
PXR_NAMESPACE_OPEN_SCOPE void WorkSerialForN(size_t n, Fn &&fn)
void WorkParallelForN(size_t n, Fn &&callback, size_t grainSize)
OIIO_UTIL_API void parallel_for(int32_t begin, int32_t end, function_view< void(int32_t)> task, paropt opt=0)
WORK_API bool WorkHasConcurrency()
void WorkParallelForEach(InputIterator first, InputIterator last, Fn &&fn)
__hostdev__ uint64_t last(uint32_t i) const
PXR_NAMESPACE_CLOSE_SCOPE PXR_NAMESPACE_OPEN_SCOPE
#define PXR_NAMESPACE_CLOSE_SCOPE