Forgot one

gevtushenko · May 23, 2023 · 837855f · 837855f
1 parent bcbb6d1
commit 837855f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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);