youngkingdom/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Qi Yuhang
2025-10-22 11:11:14 +08:00
committed by GitHub
parent b1d6e2c9b3
commit b2ca083d2b
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
include/cutlass/gemm/kernel/sm90_tile_scheduler_stream_k.hpp
View File

@ -333,8 +333,10 @@ public:
if (continue_current_work(work_tile_info)) { if (continue_current_work(work_tile_info)) {
return false; 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( return not get_current_work_for_linear_idx(
unit_iter_start_, unit_iter_start,
current_work_linear_idx_ + ( current_work_linear_idx_ + (
uint64_t(gridDim.x) * uint64_t(gridDim.y) * uint64_t(gridDim.z) * uint64_t(advance_count) uint64_t(gridDim.x) * uint64_t(gridDim.y) * uint64_t(gridDim.z) * uint64_t(advance_count)
), ),