diff --git a/thrust/thrust/system/cuda/detail/async/exclusive_scan.h b/thrust/thrust/system/cuda/detail/async/exclusive_scan.h index fbdd7a4b142..264a1e4e253 100644 --- a/thrust/thrust/system/cuda/detail/async/exclusive_scan.h +++ b/thrust/thrust/system/cuda/detail/async/exclusive_scan.h @@ -73,8 +73,8 @@ unique_eager_event async_exclusive_scan_n( execution_policy& policy, ForwardIt first, Size n, OutputIt out, InitialValueType init, BinaryOp op) { using InputValueT = cub::detail::InputValue; - using Dispatch32 = cub::DispatchScan; - using Dispatch64 = cub::DispatchScan; + using Dispatch32 = cub::DispatchScan; + using Dispatch64 = cub::DispatchScan; InputValueT init_value(init); @@ -85,7 +85,7 @@ unique_eager_event async_exclusive_scan_n( cudaError_t status; size_t tmp_size = 0; { - THRUST_INDEX_TYPE_DISPATCH2( + THRUST_UNSIGNED_INDEX_TYPE_DISPATCH2( status, Dispatch32::Dispatch, Dispatch64::Dispatch, diff --git a/thrust/thrust/system/cuda/detail/async/inclusive_scan.h b/thrust/thrust/system/cuda/detail/async/inclusive_scan.h index ffb87860efb..7f5d9e80570 100644 --- a/thrust/thrust/system/cuda/detail/async/inclusive_scan.h +++ b/thrust/thrust/system/cuda/detail/async/inclusive_scan.h @@ -68,8 +68,8 @@ unique_eager_event async_inclusive_scan_n(execution_policy& policy, ForwardIt first, Size n, OutputIt out, BinaryOp op) { using AccumT = typename thrust::iterator_traits::value_type; - using Dispatch32 = cub::DispatchScan; - using Dispatch64 = cub::DispatchScan; + using Dispatch32 = cub::DispatchScan; + using Dispatch64 = cub::DispatchScan; auto const device_alloc = get_async_device_allocator(policy); unique_eager_event ev; @@ -78,7 +78,7 @@ async_inclusive_scan_n(execution_policy& policy, ForwardIt first, cudaError_t status; size_t tmp_size = 0; { - THRUST_INDEX_TYPE_DISPATCH2( + THRUST_UNSIGNED_INDEX_TYPE_DISPATCH2( status, Dispatch32::Dispatch, Dispatch64::Dispatch, diff --git a/thrust/thrust/system/cuda/detail/scan.h b/thrust/thrust/system/cuda/detail/scan.h index 96cc914132a..890e02ba198 100644 --- a/thrust/thrust/system/cuda/detail/scan.h +++ b/thrust/thrust/system/cuda/detail/scan.h @@ -128,7 +128,7 @@ _CCCL_HOST_DEVICE OutputIt inclusive_scan_n_impl( OutputIt, ScanOp, InputValueT, - std::int32_t, + std::uint32_t, AccumT, cub::detail::scan::policy_hub, ForceInclusive>; @@ -137,7 +137,7 @@ _CCCL_HOST_DEVICE OutputIt inclusive_scan_n_impl( OutputIt, ScanOp, InputValueT, - std::int64_t, + std::uint64_t, AccumT, cub::detail::scan::policy_hub, ForceInclusive>; @@ -154,7 +154,7 @@ _CCCL_HOST_DEVICE OutputIt inclusive_scan_n_impl( // Determine temporary storage requirements: size_t tmp_size = 0; { - THRUST_INDEX_TYPE_DISPATCH2( + THRUST_UNSIGNED_INDEX_TYPE_DISPATCH2( status, Dispatch32::Dispatch, Dispatch64::Dispatch, @@ -170,7 +170,7 @@ _CCCL_HOST_DEVICE OutputIt inclusive_scan_n_impl( { // Allocate temporary storage: thrust::detail::temporary_array tmp{policy, tmp_size}; - THRUST_INDEX_TYPE_DISPATCH2( + THRUST_UNSIGNED_INDEX_TYPE_DISPATCH2( status, Dispatch32::Dispatch, Dispatch64::Dispatch,