Skip to content

Commit 9ed6ba1

Browse files
author
Sudharshan Govindan
committed
Changed device to Host warp threads
1 parent 673c80b commit 9ed6ba1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

csrc/selective_scan/reverse_scan.cuh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -96,12 +96,12 @@ struct WarpReverseScan {
9696

9797
/// Whether the logical warp size and the PTX warp size coincide
9898

99-
// In hipcub, warp_threads is defined as HIPCUB_WARP_THREADS ::rocprim::warp_size()
99+
// In hipcub, warp_threads is defined as HIPCUB_HOST_WARP_THREADS ::rocprim::host_warp_size()
100100
// While in cub, it's defined as a macro that takes a redundant unused argument.
101101
#ifndef USE_ROCM
102102
#define WARP_THREADS CUB_WARP_THREADS(0)
103103
#else
104-
#define WARP_THREADS HIPCUB_DEVICE_WARP_THREADS
104+
#define WARP_THREADS HIPCUB_HOST_WARP_THREADS
105105
#endif
106106
static constexpr bool IS_ARCH_WARP = (LOGICAL_WARP_THREADS == WARP_THREADS);
107107
/// The number of warp scan steps

0 commit comments

Comments
 (0)