[CI/NPU] Fix ascend CI issue (#16953)
This commit is contained in:
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user