From 3850028bca529b453bade8f7aa1fcd24b42ea4a6 Mon Sep 17 00:00:00 2001 From: "PoYen, Chen" Date: Tue, 20 Aug 2024 14:10:16 +0000 Subject: [PATCH] Make comment more clear --- example/ck_tile/01_fmha/utils.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/example/ck_tile/01_fmha/utils.hpp b/example/ck_tile/01_fmha/utils.hpp index c26098e39b..70a5844cde 100644 --- a/example/ck_tile/01_fmha/utils.hpp +++ b/example/ck_tile/01_fmha/utils.hpp @@ -150,7 +150,7 @@ decode_seqlen(mode_enum mode, if(1 < batch && use_kvcache) { - // we always use seqlen_k_max in first batch + // to keep the original s_k value, we always use seqlen_k_max in first batch randints(std::next(seqlen_ks.begin()), seqlen_ks.end(), seqlen_k_min,