Add thrust::async::incluisve_scan with init value

NVIDIA · Jul 17, 2024 · 703c3b4 · 703c3b4
1 parent fa8c7e3
commit 703c3b4
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/thrust/thrust/async/scan.h b/thrust/thrust/async/scan.h
@@ -112,6 +112,27 @@ struct inclusive_scan_fn final
       THRUST_FWD(out),
       thrust::plus<>{}))
 
+      template <typename DerivedPolicy,
+                typename ForwardIt,
+                typename Sentinel,
+                typename OutputIt,
+                typename InitialValueType,
+                typename BinaryOp>
+      auto operator()(thrust::detail::execution_policy_base<DerivedPolicy> const& exec,
+                      ForwardIt&& first,
+                      Sentinel&& last,
+                      OutputIt&& out,
+                      InitialValueType&& init,
+                      BinaryOp&& op) const
+    // ADL dispatch.
+    THRUST_RETURNS(async_inclusive_scan(
+      thrust::detail::derived_cast(thrust::detail::strip_const(exec)),
+      THRUST_FWD(first),
+      THRUST_FWD(last),
+      THRUST_FWD(out),
+      THRUST_FWD(init),
+      THRUST_FWD(op)))
+
       template <typename ForwardIt,
                 typename Sentinel,
                 typename OutputIt,