From 2a7b67adff36a879a6d08912aed489942d807ebe Mon Sep 17 00:00:00 2001 From: Junrong Lin <33685709+ocss884@users.noreply.github.com> Date: Tue, 13 Jan 2026 23:39:12 +0800 Subject: [PATCH] [CI/NPU] Fix ascend CI issue (#16953) --- test/srt/ascend/test_ascend_piecewise_graph_prefill.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/srt/ascend/test_ascend_piecewise_graph_prefill.py b/test/srt/ascend/test_ascend_piecewise_graph_prefill.py index 83f094df2..7f13d13e8 100644 --- a/test/srt/ascend/test_ascend_piecewise_graph_prefill.py +++ b/test/srt/ascend/test_ascend_piecewise_graph_prefill.py @@ -38,8 +38,8 @@ class TestPiecewiseGraphPrefillCorrectness(CustomTestCase): 128, "--enable-piecewise-cuda-graph", "--piecewise-cuda-graph-tokens", - ] - + TOKENS_TO_CAPTURE, + *TOKENS_TO_CAPTURE, + ], ) @classmethod