diff --git a/include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_pagedkv_pipeline_qr_ks_vs.hpp b/include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_pagedkv_pipeline_qr_ks_vs.hpp index b8213f2110..894d22e5f1 100644 --- a/include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_pagedkv_pipeline_qr_ks_vs.hpp +++ b/include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_pagedkv_pipeline_qr_ks_vs.hpp @@ -240,7 +240,7 @@ struct BlockFmhaFwdPagedKVPipelineQRKSVS }(); clear_tile(o_acc); - if((__builtin_isinf_sign(sink_v) >= 0) && i_split == 0) + if(__builtin_isinf_sign(sink_v) >= 0) { set_tile(m, init_m_val); set_tile(l, SMPLComputeDataType{1.0f});