ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-18 01:28:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix a bug in calculating max_num_target in run_group_hstu_forward()

Browse Source
This commit is contained in:
Qianfeng Zhang
2026-06-28 15:04:28 +00:00
parent 6cf17bc827
commit f2d13923d6
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_fwd.cpp
View File

@@ -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]);
};