minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
khalilzhk
2026-01-12 20:16:30 +08:00
committed by GitHub
parent 53dca74f47
commit 9a628744fc
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/srt/ascend/test_ascend_piecewise_graph_prefill.py
+4 -4
View File
@@ -38,8 +38,8 @@ class TestPiecewiseGraphPrefillCorrectness(CustomTestCase):
128, 128,
"--enable-piecewise-cuda-graph", "--enable-piecewise-cuda-graph",
"--piecewise-cuda-graph-tokens", "--piecewise-cuda-graph-tokens",
TOKENS_TO_CAPTURE, ]
], + TOKENS_TO_CAPTURE,
) )
@classmethod @classmethod
@@ -79,8 +79,8 @@ class TestPiecewiseGraphPrefillBenchmark(CustomTestCase):
"ascend", "ascend",
"--enable-piecewise-cuda-graph", "--enable-piecewise-cuda-graph",
"--piecewise-cuda-graph-tokens", "--piecewise-cuda-graph-tokens",
TOKENS_TO_CAPTURE, ]
], + TOKENS_TO_CAPTURE,
) )
self.assertLess(prefill_latency, EXP_PREFILL_LATENCY) self.assertLess(prefill_latency, EXP_PREFILL_LATENCY)