Skip to content

Commit 2171386

Browse files
committed
[oneDPL][ranges][merge][dpcpp] + out_of_bounds fix
1 parent 324fe55 commit 2171386

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

include/oneapi/dpl/pstl/hetero/dpcpp/parallel_backend_sycl_merge.h

+2-1
Original file line numberDiff line numberDiff line change
@@ -164,8 +164,9 @@ struct __parallel_merge_submitter<_IdType, __internal::__optional_kernel_name<_N
164164
auto __id = __item_id.get_linear_id();
165165
const _IdType __i_elem = __id * __chunk;
166166

167+
const auto __n_merge = std::min<_IdType>(__chunk, __n - __i_elem);
167168
const auto __start = __find_start_point(__rng1, __rng2, __i_elem, __n1, __n2, __comp);
168-
__serial_merge(__rng1, __rng2, __rng3, __start.first, __start.second, __i_elem, __chunk, __n1, __n2,
169+
__serial_merge(__rng1, __rng2, __rng3, __start.first, __start.second, __i_elem, __n_merge, __n1, __n2,
169170
__comp);
170171

171172
if(__id == __steps - 1) //the last WI does additional work

0 commit comments

Comments
 (0)