diff --git a/cub/agent/single_pass_scan_operators.cuh b/cub/agent/single_pass_scan_operators.cuh index 199e8cd10..e20462648 100644 --- 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);