fix bench_one_batch by extending array with array not list (#28732)

Signed-off-by: P V R K Jyothendra Varma <polisetty.v.r.k.jyothendra.varma@intel.com>
This commit is contained in:
Polisetty V R K Jyothendra Varma
2026-06-21 08:28:41 +08:00
committed by GitHub
parent 8a3d6c3403
commit fbbf559de2
+1 -1
View File
@@ -393,7 +393,7 @@ def prepare_extend_inputs_for_correctness_test(
):
for i in range(len(reqs)):
req: Req = reqs[i]
req.full_untruncated_fill_ids += input_ids[i][bench_args.cut_len :]
req.full_untruncated_fill_ids.extend(input_ids[i][bench_args.cut_len :])
req.fill_len = len(req.full_untruncated_fill_ids)
if model_runner is not None:
# Use req.req_pool_idx instead of i to handle slot 0 padding correctly