updated after merge

morrison-turnansky · morrison-turnansky · commit 4a53a64987b0 · 2025-10-15T00:28:39.000Z
Signed-off-by: morrison-turnansky &lt;mturnans@redhat.com&gt;
diff --git a/tests/compile/piecewise/test_toy_llama.py b/tests/compile/piecewise/test_toy_llama.py
@@ -356,13 +356,13 @@ def test_toy_llama(
     )
 
     compile_config_no_compile = CompilationConfig(
-        level=CompilationLevel.NO_COMPILATION,
+        level=CompilationMode.NONE,
         cudagraph_mode=CUDAGraphMode.NONE,
         backend="eager",
     )
 
     compile_config_no_split = CompilationConfig(
-        level=CompilationLevel.PIECEWISE,
+        level=CompilationMode.VLLM_COMPILE,
         use_inductor_graph_partition=use_inductor_graph_partition,
         cudagraph_mode=CUDAGraphMode.PIECEWISE,
         backend=backend,

Original file line number	Diff line number	Diff line change
`@@ -356,13 +356,13 @@ def test_toy_llama(`
`356`	`356`	`)`
`357`	`357`
`358`	`358`	`compile_config_no_compile = CompilationConfig(`
`359`		`- level=CompilationLevel.NO_COMPILATION,`
	`359`	`+ level=CompilationMode.NONE,`
`360`	`360`	`cudagraph_mode=CUDAGraphMode.NONE,`
`361`	`361`	`backend="eager",`
`362`	`362`	`)`
`363`	`363`
`364`	`364`	`compile_config_no_split = CompilationConfig(`
`365`		`- level=CompilationLevel.PIECEWISE,`
	`365`	`+ level=CompilationMode.VLLM_COMPILE,`
`366`	`366`	`use_inductor_graph_partition=use_inductor_graph_partition,`
`367`	`367`	`cudagraph_mode=CUDAGraphMode.PIECEWISE,`
`368`	`368`	`backend=backend,`