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 b060228272..6f47a1410b 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 @@ -274,7 +274,7 @@ struct BlockFmhaFwdSplitKVPipelineQRKSVS Policy::template MakeBiasDramTileDistribution()); auto randval_dram_window = dropout.MakeRandvalDramWindow( - randval_dram_block_window_tmp, seqlen_k_start); + randval_dram_block_window_tmp, adjusted_seqlen_k_start); auto [i_block1, v_dram_window] = v_tile_navigator.make_tile_window( v_dram_block_window_tmp,