From faa24679f6445f8425ffae442471ad84501f4bf8 Mon Sep 17 00:00:00 2001 From: HydraQYH Date: Fri, 10 Oct 2025 17:34:56 +0800 Subject: [PATCH] Fixed compilation error when using StreamK scheduler + PDL. --- include/cutlass/gemm/kernel/sm90_tile_scheduler_stream_k.hpp | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/include/cutlass/gemm/kernel/sm90_tile_scheduler_stream_k.hpp b/include/cutlass/gemm/kernel/sm90_tile_scheduler_stream_k.hpp index a298e06bf4..abefc9b6a6 100644 --- 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 @@ class PersistentTileSchedulerSm90StreamK { 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) ),