From 3d3d73bee29099f11415bbe9185257ce2ebae97e Mon Sep 17 00:00:00 2001
From: "PoYen, Chen" <PoYen.Chen@amd.com>
Date: Sun, 18 Aug 2024 17:25:39 +0000
Subject: [PATCH] Fix wrong parameter name

---
 include/ck_tile/ops/fmha/kernel/fmha_fwd_appendkv_kernel.hpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/include/ck_tile/ops/fmha/kernel/fmha_fwd_appendkv_kernel.hpp b/include/ck_tile/ops/fmha/kernel/fmha_fwd_appendkv_kernel.hpp
index 7cfb27c926..0cc3e43bbb 100644
--- a/include/ck_tile/ops/fmha/kernel/fmha_fwd_appendkv_kernel.hpp
+++ b/include/ck_tile/ops/fmha/kernel/fmha_fwd_appendkv_kernel.hpp
@@ -234,9 +234,9 @@ struct FmhaFwdAppendKVKernel
     __host__ static constexpr auto GridSize(ck_tile::index_t batch_size,
                                             ck_tile::index_t nhead,
                                             ck_tile::index_t seqlen_q,
-                                            ck_tile::index_t hdim_v)
+                                            ck_tile::index_t seqlen_knew)
     {
-        return TilePartitioner::GridSize(batch_size, nhead, seqlen_q, hdim_v);
+        return TilePartitioner::GridSize(batch_size, nhead, seqlen_q, seqlen_knew);
     }
 
     __host__ static constexpr auto BlockSize() { return dim3(kBlockSize); }