From a93c5e820f68b0cf417e62a79ff00ea124973cb8 Mon Sep 17 00:00:00 2001
From: "PoYen, Chen" <PoYen.Chen@amd.com>
Date: Sun, 18 Aug 2024 18:37:25 +0000
Subject: [PATCH] Rename parameter

---
 example/ck_tile/01_fmha/rotary.hpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/example/ck_tile/01_fmha/rotary.hpp b/example/ck_tile/01_fmha/rotary.hpp
index d8ceb99ecc..346f2a5e7e 100644
--- a/example/ck_tile/01_fmha/rotary.hpp
+++ b/example/ck_tile/01_fmha/rotary.hpp
@@ -24,7 +24,7 @@ enum class rope_enum
 
 template <typename DataType>
 std::tuple<ck_tile::HostTensor<DataType>, ck_tile::HostTensor<DataType>>
-generate_rotary_cos_sin(ck_tile::index_t seqlen_k,
+generate_rotary_cos_sin(ck_tile::index_t seqlen,
                         ck_tile::index_t rotary_dim,
                         std::optional<unsigned> seed = std::nullopt)
 {
@@ -38,7 +38,7 @@ generate_rotary_cos_sin(ck_tile::index_t seqlen_k,
     std::mt19937 random_engine(seed.has_value() ? *seed : std::random_device{}());
     std::uniform_real_distribution<float> generator(0.0f, 1.0f);
 
-    const ck_tile::index_t num_rows = seqlen_k * 2;
+    const ck_tile::index_t num_rows = seqlen * 2;
     const ck_tile::index_t num_cols = rotary_dim / 2;
 
     using std::begin, std::end;