diff --git a/thrust/system/cuda/detail/scan.h b/thrust/system/cuda/detail/scan.h index 4f9628319d..6e266a8dbd 100644 --- a/thrust/system/cuda/detail/scan.h +++ b/thrust/system/cuda/detail/scan.h @@ -137,15 +137,16 @@ OutputIt exclusive_scan_n_impl(thrust::cuda_cub::execution_policy &poli InitValueT init, ScanOp scan_op) { + using InputValueT = cub::detail::InputValue; using Dispatch32 = cub::DispatchScan; using Dispatch64 = cub::DispatchScan; cudaStream_t stream = thrust::cuda_cub::stream(policy); @@ -163,7 +164,7 @@ OutputIt exclusive_scan_n_impl(thrust::cuda_cub::execution_policy &poli first, result, scan_op, - init, + InputValueT(init), num_items_fixed, stream, THRUST_DEBUG_SYNC_FLAG)); @@ -187,7 +188,7 @@ OutputIt exclusive_scan_n_impl(thrust::cuda_cub::execution_policy &poli first, result, scan_op, - init, + InputValueT(init), num_items_fixed, stream, THRUST_DEBUG_SYNC_FLAG));