Skip to content
This repository has been archived by the owner on Mar 21, 2024. It is now read-only.

Commit

Permalink
cub change cub/pull/305
Browse files Browse the repository at this point in the history
  • Loading branch information
@zasdfgbnm
zasdfgbnm committed Oct 6, 2021
1 parent ba52753 commit 1ce950e
Showing 1 changed file with 5 additions and 4 deletions.
9 changes: 5 additions & 4 deletions thrust/system/cuda/detail/scan.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -137,15 +137,16 @@ OutputIt exclusive_scan_n_impl(thrust::cuda_cub::execution_policy<Derived> &poli
InitValueT init,
ScanOp scan_op)
{
using InputValueT = cub::detail::InputValue<InitValueT>;
using Dispatch32 = cub::DispatchScan<InputIt,
OutputIt,
ScanOp,
InitValueT,
InputValueT,
thrust::detail::int32_t>;
using Dispatch64 = cub::DispatchScan<InputIt,
OutputIt,
ScanOp,
InitValueT,
InputValueT,
thrust::detail::int64_t>;

cudaStream_t stream = thrust::cuda_cub::stream(policy);
Expand All @@ -163,7 +164,7 @@ OutputIt exclusive_scan_n_impl(thrust::cuda_cub::execution_policy<Derived> &poli
first,
result,
scan_op,
init,
InputValueT(init),
num_items_fixed,
stream,
THRUST_DEBUG_SYNC_FLAG));
Expand All @@ -187,7 +188,7 @@ OutputIt exclusive_scan_n_impl(thrust::cuda_cub::execution_policy<Derived> &poli
first,
result,
scan_op,
init,
InputValueT(init),
num_items_fixed,
stream,
THRUST_DEBUG_SYNC_FLAG));
Expand Down

0 comments on commit 1ce950e

Please sign in to comment.