From db31475e072cad2099acbab25c87bb50a5eb0503 Mon Sep 17 00:00:00 2001
From: "PoYen, Chen" <PoYen.Chen@amd.com>
Date: Tue, 6 Aug 2024 08:37:29 +0000
Subject: [PATCH] Unify origin

---
 .../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 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<Problem, decltype(gemm_0)>());
 
         auto randval_dram_window = dropout.MakeRandvalDramWindow<decltype(gemm_0)>(
-            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,