From f2d13923d651ac014376d245f7007ac7f48613ce Mon Sep 17 00:00:00 2001 From: Qianfeng Zhang Date: Sun, 28 Jun 2026 15:04:28 +0000 Subject: [PATCH] Fix a bug in calculating max_num_target in run_group_hstu_forward() --- .../ck_tile/18_hstu_attention/example_hstu_attention_fwd.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/example/ck_tile/18_hstu_attention/example_hstu_attention_fwd.cpp b/example/ck_tile/18_hstu_attention/example_hstu_attention_fwd.cpp index e1f7c0edfc..5eb780a688 100644 --- a/example/ck_tile/18_hstu_attention/example_hstu_attention_fwd.cpp +++ b/example/ck_tile/18_hstu_attention/example_hstu_attention_fwd.cpp @@ -806,7 +806,7 @@ bool run_group_hstu_forward(const ck_tile::ArgParser& arg_parser, int num_group) { for(int i_batch = 0; i_batch < num_batch_per_group; i_batch++) { - int i_global_batch = i_grp * num_batch_per_group * i_batch; + int i_global_batch = i_grp * num_batch_per_group + i_batch; max_num_target = max(max_num_target, num_targets[i_global_batch]); };