From f574b10ba5b5f8a5317f5a3ce0504355822383cd Mon Sep 17 00:00:00 2001 From: khalil2ji3mp6 Date: Mon, 12 Jan 2026 14:51:57 +0800 Subject: [PATCH 1/2] [CI] fix piecewise graph test case on ascend --- test/srt/ascend/test_ascend_piecewise_graph_prefill.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/test/srt/ascend/test_ascend_piecewise_graph_prefill.py b/test/srt/ascend/test_ascend_piecewise_graph_prefill.py index 9e43ca60f74b..d9dd8ca7755e 100644 --- a/test/srt/ascend/test_ascend_piecewise_graph_prefill.py +++ b/test/srt/ascend/test_ascend_piecewise_graph_prefill.py @@ -38,8 +38,7 @@ def setUpClass(cls): 128, "--enable-piecewise-cuda-graph", "--piecewise-cuda-graph-tokens", - TOKENS_TO_CAPTURE, - ], + ] + TOKENS_TO_CAPTURE ) @classmethod @@ -79,8 +78,7 @@ def test_latency(self): "ascend", "--enable-piecewise-cuda-graph", "--piecewise-cuda-graph-tokens", - TOKENS_TO_CAPTURE, - ], + ] + TOKENS_TO_CAPTURE ) self.assertLess(prefill_latency, EXP_PREFILL_LATENCY) From 9908a0ed3034892b2c02b9436ad8c9e7fd4a52aa Mon Sep 17 00:00:00 2001 From: khalil2ji3mp6 Date: Mon, 12 Jan 2026 14:59:01 +0800 Subject: [PATCH 2/2] fix --- test/srt/ascend/test_ascend_piecewise_graph_prefill.py | 6 ++++-- 1 file changed, 4 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 d9dd8ca7755e..83f094df2aac 100644 --- a/test/srt/ascend/test_ascend_piecewise_graph_prefill.py +++ b/test/srt/ascend/test_ascend_piecewise_graph_prefill.py @@ -38,7 +38,8 @@ def setUpClass(cls): 128, "--enable-piecewise-cuda-graph", "--piecewise-cuda-graph-tokens", - ] + TOKENS_TO_CAPTURE + ] + + TOKENS_TO_CAPTURE, ) @classmethod @@ -78,7 +79,8 @@ def test_latency(self): "ascend", "--enable-piecewise-cuda-graph", "--piecewise-cuda-graph-tokens", - ] + TOKENS_TO_CAPTURE + ] + + TOKENS_TO_CAPTURE, ) self.assertLess(prefill_latency, EXP_PREFILL_LATENCY)