From 6785f99c18cdb937ec9ae59ea7fa93cada39b36a Mon Sep 17 00:00:00 2001 From: jakpiase Date: Tue, 2 Dec 2025 11:38:06 +0100 Subject: [PATCH] [CK_TILE] Fix for comp pipeline v4 (#3307) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * Fix for gemm_pipeline_ag_bg_cr_comp_v4 * Update hotloop condition Co-authored-by: Bartłomiej Kocot * fix formating --------- Co-authored-by: Bartłomiej Kocot [ROCm/composable_kernel commit: 5d67d82a0bb6dbf5f82f3b4ba2e9188eb838b927] --- .../ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_comp_v4.hpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_comp_v4.hpp b/include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_comp_v4.hpp index f22d2f599a..d448cdbb93 100644 --- a/include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_comp_v4.hpp +++ b/include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_comp_v4.hpp @@ -23,7 +23,8 @@ struct BaseGemmPipelineAgBgCrCompV4 CK_TILE_HOST_DEVICE static constexpr bool BlockHasHotloop(index_t num_loop) { - return num_loop > PrefetchStages; + constexpr index_t HotLoopGlobalReads = 2; + return num_loop >= (HotLoopGlobalReads + PrefetchStages); } CK_TILE_HOST_DEVICE static constexpr TailNumber GetBlockLoopTailNum(index_t num_loop)