File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -96,12 +96,12 @@ struct WarpReverseScan {
96
96
97
97
// / Whether the logical warp size and the PTX warp size coincide
98
98
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 ()
100
100
// While in cub, it's defined as a macro that takes a redundant unused argument.
101
101
#ifndef USE_ROCM
102
102
#define WARP_THREADS CUB_WARP_THREADS (0 )
103
103
#else
104
- #define WARP_THREADS HIPCUB_DEVICE_WARP_THREADS
104
+ #define WARP_THREADS HIPCUB_HOST_WARP_THREADS
105
105
#endif
106
106
static constexpr bool IS_ARCH_WARP = (LOGICAL_WARP_THREADS == WARP_THREADS);
107
107
// / The number of warp scan steps
You can’t perform that action at this time.
0 commit comments