Fix review comments: __current_iters_per_work_item is not required

Signed-off-by: Sergey Kopienko <[email protected]>
uxlfoundation · Aug 9, 2024 · 4922f46 · 4922f46
1 parent b2af0e8
commit 4922f46
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/include/oneapi/dpl/pstl/hetero/dpcpp/parallel_backend_sycl.h b/include/oneapi/dpl/pstl/hetero/dpcpp/parallel_backend_sycl.h
@@ -1186,9 +1186,9 @@ struct __parallel_find_or_nd_range_tuner<oneapi::dpl::__internal::__device_backe
 
                 if (__iters_per_work_item < __desired_iters_per_work_item)
                 {
-                    auto __k = __desired_iters_per_work_item / __iters_per_work_item;
-                    __k = std::pow(2, std::ceil(std::log2(__k)));
-                    __n_groups = (std::size_t)std::ceil(__n_groups / __k);
+                    auto __k = oneapi::dpl::__internal::__dpl_bit_ceil(
+                        (std::size_t)std::ceil(__desired_iters_per_work_item / __iters_per_work_item));
+                    __n_groups = oneapi::dpl::__internal::__dpl_ceiling_div(__n_groups, __k);
 
                     assert(oneapi::dpl::__internal::__dpl_ceiling_div(__rng_n, __n_groups * __wgroup_size) <=
                            __desired_iters_per_work_item);