Forgot to set partial state

gevtushenko · May 9, 2023 · 5333190 · 5333190
1 parent 8f4008a
commit 5333190
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cub/agent/single_pass_scan_operators.cuh b/cub/agent/single_pass_scan_operators.cuh
@@ -571,6 +571,9 @@ struct ClusterTilePrefixCallbackOp
                 {
                   if (threadIdx.x == 0)
                   {
+                    T inclusive_prefix = scan_op(exclusive_prefix, block_aggregate);
+                    tile_status.SetPartial(tile_idx, inclusive_prefix);
+
                     do
                     {
                       LoadTileDescriptor(src_cta, tile_descriptor);