Fixed compilation error when using StreamK scheduler + PDL. (#2686)

2025-10-22 11:11:14 +08:00
parent b1d6e2c9b3
commit b2ca083d2b
1 changed files with 3 additions and 1 deletions
--- a/include/cutlass/gemm/kernel/sm90_tile_scheduler_stream_k.hpp
+++ b/include/cutlass/gemm/kernel/sm90_tile_scheduler_stream_k.hpp
@ -333,8 +333,10 @@ public:
    if (continue_current_work(work_tile_info)) {
      return false;
    }
+    // Create a copy to avoid unit_iter_start_ being modified
+    uint32_t unit_iter_start = unit_iter_start_;
    return not get_current_work_for_linear_idx(
-        unit_iter_start_,
+        unit_iter_start,
        current_work_linear_idx_ + (
          uint64_t(gridDim.x) * uint64_t(gridDim.y) * uint64_t(gridDim.z) * uint64_t(advance_count)
          ),