From e31829384d0922cc83c2af39d9d93b4f9448c1c7 Mon Sep 17 00:00:00 2001 From: Qianfeng Zhang Date: Sun, 2 Nov 2025 03:54:51 +0000 Subject: [PATCH] Change in updating max_uih_seqlen in the example --- example/ck_tile/18_hstu_attention/example_hstu_attention.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/example/ck_tile/18_hstu_attention/example_hstu_attention.cpp b/example/ck_tile/18_hstu_attention/example_hstu_attention.cpp index 2052b6df86..da579605c1 100644 --- a/example/ck_tile/18_hstu_attention/example_hstu_attention.cpp +++ b/example/ck_tile/18_hstu_attention/example_hstu_attention.cpp @@ -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