We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b1d6e2c commit b2ca083Copy full SHA for b2ca083
include/cutlass/gemm/kernel/sm90_tile_scheduler_stream_k.hpp
@@ -333,8 +333,10 @@ class PersistentTileSchedulerSm90StreamK {
333
if (continue_current_work(work_tile_info)) {
334
return false;
335
}
336
+ // Create a copy to avoid unit_iter_start_ being modified
337
+ uint32_t unit_iter_start = unit_iter_start_;
338
return not get_current_work_for_linear_idx(
- unit_iter_start_,
339
+ unit_iter_start,
340
current_work_linear_idx_ + (
341
uint64_t(gridDim.x) * uint64_t(gridDim.y) * uint64_t(gridDim.z) * uint64_t(advance_count)
342
),
0 commit comments