use cudagraph for mpk

NorthmanPKU · NorthmanPKU · commit 935674200b1d · 2025-11-30T19:54:29.000Z
diff --git a/vllm/config/vllm.py b/vllm/config/vllm.py
@@ -690,10 +690,9 @@ def has_blocked_weights():
                         )
                         self.compilation_config.cudagraph_mode = CUDAGraphMode.PIECEWISE
 
-            # disable cudagraph when enforce eager execution or mirage backend is used
+            # disable cudagraph when enforce eager execution
             disable_cuda_graph = (
                 (self.model_config is not None and self.model_config.enforce_eager)
-                or (self.compilation_config.backend == "mirage_byname")
             )
             if disable_cuda_graph:
                 logger.info("Cudagraph is disabled under eager mode")

Original file line number	Diff line number	Diff line change
`@@ -690,10 +690,9 @@ def has_blocked_weights():`
`690`	`690`	`)`
`691`	`691`	`self.compilation_config.cudagraph_mode = CUDAGraphMode.PIECEWISE`
`692`	`692`
`693`		`- # disable cudagraph when enforce eager execution or mirage backend is used`
	`693`	`+ # disable cudagraph when enforce eager execution`
`694`	`694`	`disable_cuda_graph = (`
`695`	`695`	`(self.model_config is not None and self.model_config.enforce_eager)`
`696`		`- or (self.compilation_config.backend == "mirage_byname")`
`697`	`696`	`)`
`698`	`697`	`if disable_cuda_graph:`
`699`	`698`	`logger.info("Cudagraph is disabled under eager mode")`