Skip to content

Commit b2ca083

Browse files
authored
Fixed compilation error when using StreamK scheduler + PDL. (#2686)
1 parent b1d6e2c commit b2ca083

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

include/cutlass/gemm/kernel/sm90_tile_scheduler_stream_k.hpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -333,8 +333,10 @@ class PersistentTileSchedulerSm90StreamK {
333333
if (continue_current_work(work_tile_info)) {
334334
return false;
335335
}
336+
// Create a copy to avoid unit_iter_start_ being modified
337+
uint32_t unit_iter_start = unit_iter_start_;
336338
return not get_current_work_for_linear_idx(
337-
unit_iter_start_,
339+
unit_iter_start,
338340
current_work_linear_idx_ + (
339341
uint64_t(gridDim.x) * uint64_t(gridDim.y) * uint64_t(gridDim.z) * uint64_t(advance_count)
340342
),

0 commit comments

Comments
 (0)