ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-06 15:54:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Rename option s_k_new to s_knew

Browse Source
This commit is contained in:
PoYen, Chen
2024-08-07 15:31:54 +00:00
parent cef9da0a76
commit 655b13b059
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
example/ck_tile/01_fmha/fmha_fwd.cpp
View File

@@ -52,10 +52,10 @@ auto create_args(int argc, char* argv[])
"total_seqlen_q = seqlen_q * batch, and seqlen_q per batch may vary\n"
"also with \"-s=s0,s1,s2...\" comma seperated int to set per batch seqlen(group-mode)")
.insert("s_k", "-1", "seqlen_k (including new key/value), -1 means equal to s")
.insert("s_k_new",
.insert("s_knew",
"0",
"seqlen_k for new key/value, 0 means not to use this at all; "
"-1 to choose s_k_new in [1, s] randomly.")
"-1 to choose s_knew in [1, s] randomly.")
.insert("s_kpad",
"-1",
"seqlen_k stride between 2 tokens, currently used in group-mode only\n"
@@ -293,7 +293,7 @@ bool run(const ck_tile::ArgParser& arg_parser)
seed.reset();
}
ck_tile::index_t seqlen_knew = arg_parser.get_int("s_k_new");
ck_tile::index_t seqlen_knew = arg_parser.get_int("s_knew");
#if !CK_TILE_FMHA_FWD_APPENDKV_API
if(seqlen_knew != 0)
{