From e5888ce06a7975d6fed53023de02e4ae782fbada Mon Sep 17 00:00:00 2001 From: pavelkumbrasev Date: Mon, 15 Jul 2024 11:00:53 +0100 Subject: [PATCH] Apply review comment Signed-off-by: pavelkumbrasev --- include/oneapi/tbb/parallel_for_each.h | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/include/oneapi/tbb/parallel_for_each.h b/include/oneapi/tbb/parallel_for_each.h index 49867907bb..85c0269196 100644 --- a/include/oneapi/tbb/parallel_for_each.h +++ b/include/oneapi/tbb/parallel_for_each.h @@ -118,11 +118,11 @@ struct feeder_item_task: public task { using feeder_type = feeder_impl; template - feeder_item_task(ItemType&& input_item, feeder_type& feeder, small_object_allocator& alloc, wait_tree_vertex_interface* node) : + feeder_item_task(ItemType&& input_item, feeder_type& feeder, small_object_allocator& alloc, wait_tree_vertex_interface& wait_vertex) : item(std::forward(input_item)), my_feeder(feeder), my_allocator(alloc), - m_wait_tree_vertex(node) + m_wait_tree_vertex(r1::get_thread_reference_vertex(&wait_vertex)) { m_wait_tree_vertex->reserve(); } @@ -174,7 +174,7 @@ class feeder_impl : public feeder { void internal_add_copy_impl(std::true_type, const Item& item) { using feeder_task = feeder_item_task; small_object_allocator alloc; - auto task = alloc.new_object(item, *this, alloc, r1::get_thread_reference_vertex(&my_wait_context)); + auto task = alloc.new_object(item, *this, alloc, my_wait_context); spawn(*task, my_execution_context); } @@ -190,7 +190,7 @@ class feeder_impl : public feeder { void internal_add_move(Item&& item) override { using feeder_task = feeder_item_task; small_object_allocator alloc{}; - auto task = alloc.new_object(std::move(item), *this, alloc, r1::get_thread_reference_vertex(&my_wait_context)); + auto task = alloc.new_object(std::move(item), *this, alloc, my_wait_context); spawn(*task, my_execution_context); }