diff --git a/include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_splitkv_combine_pipeline.hpp b/include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_splitkv_combine_pipeline.hpp index 1acb5b9074..d456717ff7 100644 --- a/include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_splitkv_combine_pipeline.hpp +++ b/include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_splitkv_combine_pipeline.hpp @@ -172,11 +172,11 @@ struct BlockFmhaFwdSplitKVCombinePipeline #if defined(PRINT_LSE_ACCUM) DEBUG_STMTS { - printf("\n"); + printf("(%2d, %2d)\n", num_splits, kM0); for(index_t row = 0; row < num_splits; ++row) { printf("[POYENC][DEVICE] lse_acc[%2d] = ", row); - for(index_t col = 0; col < real_seqlen_q; ++col) + for(index_t col = 0; col < kM0; ++col) { printf("%11.7f", lse_acc_lds_ptr[row + col * kMaxSplits]); }