diff --git a/cub/detail/choose_offset.cuh b/cub/detail/choose_offset.cuh index ed703e26c9..48a4fe1b21 100644 --- a/cub/detail/choose_offset.cuh +++ b/cub/detail/choose_offset.cuh @@ -50,10 +50,10 @@ struct ChooseOffsetT !std::is_same::type, bool>::value, "NumItemsT must be an integral type, but not bool"); - // Unsigned integer type for global offsets. + // Signed integer type for global offsets. using Type = typename std::conditional::type; + std::int32_t, + std::int64_t>::type; }; } // namespace detail