[CI] Add tests for cudagraph (#27391)

Signed-off-by: zjy0516 <riverclouds.zhu@qq.com>
2025-10-25 10:37:33 +08:00
parent 83f478bb19
commit 29c9cb8007
4 changed files with 54 additions and 18 deletions
--- a/tests/v1/cudagraph/test_cudagraph_mode.py
+++ b/tests/v1/cudagraph/test_cudagraph_mode.py
@@ -109,9 +109,9 @@ combo_cases_2 = [
@pytest.mark.parametrize(
    "backend_name,cudagraph_mode,compilation_mode,supported", combo_cases_2
 )
-def test_cudagraph_compilation_combo(combo_case):
-    backend_name, cudagraph_mode, compilation_mode, supported = combo_case
-
+def test_cudagraph_compilation_combo(
+    backend_name, cudagraph_mode, compilation_mode, supported
+):
    env_vars = backend_configs[backend_name].env_vars

    with temporary_environ(env_vars), ExitStack() as stack: