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 {
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
You can’t perform that action at this time.
0 commit comments