Skip to content

Commit 9c228a8

Browse files
committed
[CUB] uint32_t alias
1 parent ec8926d commit 9c228a8

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

cub/cub/block/block_radix_rank.cuh

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -719,22 +719,23 @@ public:
719719
// Each warp will strip-mine its section of input, one strip at a time
720720

721721
volatile DigitCounterT* digit_counters[KEYS_PER_THREAD];
722-
::cuda::std::uint32_t warp_id = linear_tid >> LOG_WARP_THREADS;
723-
::cuda::std::uint32_t lane_mask_lt = ::cuda::ptx::get_sreg_lanemask_lt();
722+
using ::cuda::std::uint32_t;
723+
uint32_t warp_id = linear_tid >> LOG_WARP_THREADS;
724+
uint32_t lane_mask_lt = ::cuda::ptx::get_sreg_lanemask_lt();
724725

725726
_CCCL_PRAGMA_UNROLL_FULL()
726727
for (int ITEM = 0; ITEM < KEYS_PER_THREAD; ++ITEM)
727728
{
728729
// My digit
729-
::cuda::std::uint32_t digit = digit_extractor.Digit(keys[ITEM]);
730+
uint32_t digit = digit_extractor.Digit(keys[ITEM]);
730731

731732
if (IS_DESCENDING)
732733
{
733734
digit = RADIX_DIGITS - digit - 1;
734735
}
735736

736737
// Mask of peers who have same digit as me
737-
::cuda::std::uint32_t peer_mask =
738+
uint32_t peer_mask =
738739
detail::warp_in_block_matcher_t<RADIX_BITS, PARTIAL_WARP_THREADS, WARPS - 1>::match_any(digit, warp_id);
739740

740741
// Pointer to smem digit counter for this key

0 commit comments

Comments
 (0)