ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-17 19:40:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change in updating max_uih_seqlen in the example

Browse Source
This commit is contained in:
Qianfeng Zhang
2025-11-02 03:54:51 +00:00
parent 39cb8c33d1
commit e31829384d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
example/ck_tile/18_hstu_attention/example_hstu_attention.cpp
View File

@@ -290,7 +290,7 @@ bool run(const ck_tile::ArgParser& arg_parser)
// only consider num_batch values even if more values are provided by the user
for(int i = 0; i < num_batch; i++)
{
max_uih_seqlen = max(max_uih_seqlen, seq_lengths_q[i]);
max_uih_seqlen = max(max_uih_seqlen, max(seq_lengths_q[i], seq_lengths_kv[i]));
};
}
else