From 837855f2671ea4817fb86d534c0fbc36e0492ccc Mon Sep 17 00:00:00 2001
From: Georgy Evtushenko <evtushenko.georgy@gmail.com>
Date: Tue, 23 May 2023 19:18:44 +0400
Subject: [PATCH] Forgot one

---
 cub/agent/single_pass_scan_operators.cuh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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<ValueT, KeyT, true>
 
         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<TileDescriptor&>(alias);