[oneDPL][ranges][merge] support size limit for output; + fixes for std::min usage

MikeDvorskiy · MikeDvorskiy · commit 5b078adc3473 · 2024-11-29T18:24:13.000+01:00
diff --git a/include/oneapi/dpl/pstl/hetero/algorithm_ranges_impl_hetero.h b/include/oneapi/dpl/pstl/hetero/algorithm_ranges_impl_hetero.h
@@ -54,7 +54,8 @@ template <typename _BackendTag, typename _ExecutionPolicy, typename _Function, t
 void
 __pattern_walk_n(__hetero_tag<_BackendTag>, _ExecutionPolicy&& __exec, _Function __f, _Ranges&&... __rngs)
 {
-    auto __n = std::min({__rngs.size()...});
+    using _Size = std::make_unsigned_t<std::common_type_t<oneapi::dpl::__internal::__difference_t<_Ranges>...>>;
+    auto __n = std::min({_Size(__rngs.size())...});
     if (__n > 0)
     {
         oneapi::dpl::__par_backend_hetero::__parallel_for(_BackendTag{}, ::std::forward<_ExecutionPolicy>(__exec),

Original file line number	Diff line number	Diff line change
`@@ -54,7 +54,8 @@ template <typename _BackendTag, typename _ExecutionPolicy, typename _Function, t`
`54`	`54`	`void`
`55`	`55`	`__pattern_walk_n(__hetero_tag<_BackendTag>, _ExecutionPolicy&& __exec, _Function __f, _Ranges&&... __rngs)`
`56`	`56`	`{`
`57`		`- auto __n = std::min({__rngs.size()...});`
	`57`	`+ using _Size = std::make_unsigned_t<std::common_type_t<oneapi::dpl::__internal::__difference_t<_Ranges>...>>;`
	`58`	`+ auto __n = std::min({_Size(__rngs.size())...});`
`58`	`59`	`if (__n > 0)`
`59`	`60`	`{`
`60`	`61`	`oneapi::dpl::__par_backend_hetero::__parallel_for(_BackendTag{}, ::std::forward<_ExecutionPolicy>(__exec),`