++

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

@@ -522,7 +522,7 @@ struct __usm_or_buffer_accessor
 struct __result_and_scratch_storage_base
 {
     virtual ~__result_and_scratch_storage_base() = default;
-    template <typename _Event>
+    template <typename _T, typename _Event>
     virtual _T __wait_and_get_value(_Event&&, size_t) const = 0;
 };
 
@@ -684,8 +684,8 @@ struct __result_and_scratch_storage : __result_and_scratch_storage_base
         }
     }
 
-    template <typename _Event>
-    virtual _T
+    template <typename _Val, typename _Event>
+    virtual _Val
     __wait_and_get_value(_Event&& __event, size_t idx = 0) const
     {
         if (is_USM())
@@ -728,14 +728,14 @@ class __future : private std::tuple<_Args...>
     constexpr auto
     __wait_and_get_value(const __result_and_scratch_storage<_ExecutionPolicy, _T>& __storage)
     {
-        return __storage.__wait_and_get_value(__my_event);
+        return __storage.__wait_and_get_value<_T>(__my_event);
     }
 
     template <typename _ExecutionPolicy, typename _T>
     constexpr auto
     __wait_and_get_value(const std::shared_ptr<__result_and_scratch_storage_base>& __storage)
     {
-        return __storage.__wait_and_get_value(__my_event);
+        return __storage.__wait_and_get_value<_T>(__my_event);
     }
 
     template <typename _T>