[CI/NPU] Fix ascend CI issue (#16953)

This commit is contained in:
Junrong Lin
2026-01-13 23:39:12 +08:00
committed by GitHub
parent 1d811094f8
commit 2a7b67adff

View File

@@ -38,8 +38,8 @@ class TestPiecewiseGraphPrefillCorrectness(CustomTestCase):
128,
"--enable-piecewise-cuda-graph",
"--piecewise-cuda-graph-tokens",
]
+ TOKENS_TO_CAPTURE,
*TOKENS_TO_CAPTURE,
],
)
@classmethod