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

Use supplement_array_by_last_element(num_targets, ) in example

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

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

@@ -266,13 +266,7 @@ bool run(const ck_tile::ArgParser& arg_parser)
if(!num_targets.empty())
{
// supplement num_targets using the last input value if user-provided lengths not enough
if(static_cast<int>(num_targets.size()) < num_batch)
{
auto last_val = num_targets.back();
for(int i = num_targets.size(); i < num_batch; i++)
num_targets.push_back(last_val);
};
supplement_array_by_last_element(num_targets, num_batch);
// only consider num_batch values even if more values are provided by the user
for(int i = 0; i < num_batch; i++)