Revert "[BugFix] Fix assert in build_for_cudagraph_capture (vllm-project#29893)"

minosfuture · minosfuture · commit 3a5fa1afae2b · 2025-12-04T01:18:20.000Z
This reverts commit 5cdd664.
diff --git a/vllm/v1/worker/gpu_model_runner.py b/vllm/v1/worker/gpu_model_runner.py
@@ -4003,7 +4003,7 @@ def _dummy_run(
                 num_reqs=num_reqs_padded,
                 max_query_len=max_query_len,
                 ubatch_slices=ubatch_slices,
-                for_cudagraph_capture=is_graph_capturing,
+                for_cudagraph_capture=True,
             )
 
         with self.maybe_dummy_run_with_lora(

Original file line number	Diff line number	Diff line change
`@@ -4003,7 +4003,7 @@ def _dummy_run(`
`4003`	`4003`	`num_reqs=num_reqs_padded,`
`4004`	`4004`	`max_query_len=max_query_len,`
`4005`	`4005`	`ubatch_slices=ubatch_slices,`
`4006`		`- for_cudagraph_capture=is_graph_capturing,`
	`4006`	`+ for_cudagraph_capture=True,`
`4007`	`4007`	`)`
`4008`	`4008`
`4009`	`4009`	`with self.maybe_dummy_run_with_lora(`