Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
2a194ddd
Unverified
Commit
2a194ddd
authored
Mar 09, 2026
by
Woosuk Kwon
Committed by
GitHub
Mar 09, 2026
Browse files
[Model Runner V2] Add model_state inputs to CUDA graph capture (#36544)
Signed-off-by:
Woosuk Kwon
<
woosuk@inferact.ai
>
parent
203a7f27
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
0 deletions
+1
-0
vllm/v1/worker/gpu/cudagraph_utils.py
vllm/v1/worker/gpu/cudagraph_utils.py
+1
-0
No files found.
vllm/v1/worker/gpu/cudagraph_utils.py
View file @
2a194ddd
...
...
@@ -320,6 +320,7 @@ class ModelCudaGraphManager(CudaGraphManager):
model_inputs
=
{
"input_ids"
:
input_buffers
.
input_ids
[:
num_tokens
],
"positions"
:
input_buffers
.
positions
[:
num_tokens
],
**
model_state
.
prepare_dummy_inputs
(
num_reqs
,
num_tokens
),
}
model_output
=
model
(
**
model_inputs
)
if
self
.
use_aux_hidden_state_outputs
:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment