[CI] fix piecewise graph test case on ascend (#16933)

2026-01-12 20:16:30 +08:00
parent 53dca74f47
commit 9a628744fc
1 changed files with 4 additions and 4 deletions
--- 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
@@ -79,8 +79,8 @@ class TestPiecewiseGraphPrefillBenchmark(CustomTestCase):
                "ascend",
                "--enable-piecewise-cuda-graph",
                "--piecewise-cuda-graph-tokens",
-                TOKENS_TO_CAPTURE,
-            ],
+            ]
+            + TOKENS_TO_CAPTURE,
        )
        self.assertLess(prefill_latency, EXP_PREFILL_LATENCY)