Pack __pattern_walk3_async calls into __except_handler

oneapi-src · Mar 20, 2024 · 456e8c5 · 456e8c5
1 parent 540e8db
commit 456e8c5
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/include/oneapi/dpl/internal/async_impl/glue_async_impl.h b/include/oneapi/dpl/internal/async_impl/glue_async_impl.h
@@ -67,10 +67,13 @@ transform_async(_ExecutionPolicy&& __exec, _ForwardIterator1 __first1, _ForwardI
     const auto __dispatch_tag = oneapi::dpl::__internal::__select_backend(__exec, __first1, __first2, __result);
 
     wait_for_all(::std::forward<_Events>(__dependencies)...);
-    auto ret_val = oneapi::dpl::__internal::__pattern_walk3_async(
-        __dispatch_tag, ::std::forward<_ExecutionPolicy>(__exec), __first1, __last1, __first2, __result,
-        oneapi::dpl::__internal::__transform_functor<_BinaryOperation>(::std::move(__op)));
-    return ret_val;
+
+    return __internal::__except_handler([&]() {
+        auto ret_val = oneapi::dpl::__internal::__pattern_walk3_async(
+            __dispatch_tag, ::std::forward<_ExecutionPolicy>(__exec), __first1, __last1, __first2, __result,
+            oneapi::dpl::__internal::__transform_functor<_BinaryOperation>(::std::move(__op)));
+        return ret_val;
+    }
 }
 
 // [async.copy]