Skip to content

GitLab

Unverified Commit 2ab441be authored by Ronald's avatar Ronald Committed by GitHub
Browse files

[platform] add dp_metadata arg to set_additional_forward_context (#31942) 


Signed-off-by: default avatarRonald1995 <ronaldautomobile@163.com>
parent 9572f74f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
vllm/forward_context.py
View file @ 2ab441be
...@@ -319,6 +319,7 @@ def set_forward_context( ...@@ -319,6 +319,7 @@ def set_forward_context(
attn_metadata=attn_metadata, attn_metadata=attn_metadata,
vllm_config=vllm_config, vllm_config=vllm_config,
virtual_engine=virtual_engine, virtual_engine=virtual_engine,
dp_metadata=dp_metadata,
num_tokens=num_tokens, num_tokens=num_tokens,
num_tokens_across_dp=num_tokens_across_dp, num_tokens_across_dp=num_tokens_across_dp,
cudagraph_runtime_mode=cudagraph_runtime_mode, cudagraph_runtime_mode=cudagraph_runtime_mode,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment