diff --git a/cub/agent/single_pass_scan_operators.cuh b/cub/agent/single_pass_scan_operators.cuh index 5ba13b6ce..06a0fec4c 100644 --- a/cub/agent/single_pass_scan_operators.cuh +++ b/cub/agent/single_pass_scan_operators.cuh @@ -715,7 +715,7 @@ struct ReduceByKeyScanTileState do { - detail::delay_or_prevent_hoisting(delay); + detail::delay_or_prevent_hoisting(delay <<= 1); TxnWord alias = detail::load_relaxed(d_tile_descriptors + TILE_STATUS_PADDING + tile_idx); tile_descriptor = reinterpret_cast(alias);