From 40c885f007ba5cd4b37dd0bf8b22a81b7fbb1bd4 Mon Sep 17 00:00:00 2001 From: "PoYen, Chen" Date: Tue, 11 Jun 2024 06:50:02 +0000 Subject: [PATCH] Fix wrong loop counter step logic --- .../fmha/pipeline/block_fmha_fwd_splitkv_pipeline_qr_ks_vs.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_splitkv_pipeline_qr_ks_vs.hpp b/include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_splitkv_pipeline_qr_ks_vs.hpp index 778e021d72..f768551755 100644 --- a/include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_splitkv_pipeline_qr_ks_vs.hpp +++ b/include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_splitkv_pipeline_qr_ks_vs.hpp @@ -562,7 +562,7 @@ struct BlockFmhaFwdSplitKVPipelineQRKSVS } }; - while(++i_total_loops < num_total_loop - 1) + while(i_total_loops++ < num_total_loop - 1) { loop_body(std::false_type{}); }