From 39cb8c33d14f7da68a14bb29d51f9d9fa947c98e Mon Sep 17 00:00:00 2001 From: Qianfeng Zhang Date: Sun, 2 Nov 2025 03:30:13 +0000 Subject: [PATCH] Use supplement_array_by_last_element(num_targets, ) in example --- .../ck_tile/18_hstu_attention/example_hstu_attention.cpp | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) 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 1354ebd50a..2052b6df86 100644 --- a/example/ck_tile/18_hstu_attention/example_hstu_attention.cpp +++ b/example/ck_tile/18_hstu_attention/example_hstu_attention.cpp @@ -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(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++)