From 2ab441befef05b3c5ffdb23f981f13d035f4cd3c Mon Sep 17 00:00:00 2001
From: Ronald <ronaldautomobile@163.com>
Date: Thu, 8 Jan 2026 14:56:44 +0800
Subject: [PATCH] [platform] add dp_metadata arg to
 set_additional_forward_context (#31942)

Signed-off-by: Ronald1995 <ronaldautomobile@163.com>
---
 vllm/forward_context.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/vllm/forward_context.py b/vllm/forward_context.py
index d1223ad83fb..1ef8e5403e9 100644
--- a/vllm/forward_context.py
+++ b/vllm/forward_context.py
@@ -319,6 +319,7 @@ def set_forward_context(
         attn_metadata=attn_metadata,
         vllm_config=vllm_config,
         virtual_engine=virtual_engine,
+        dp_metadata=dp_metadata,
         num_tokens=num_tokens,
         num_tokens_across_dp=num_tokens_across_dp,
         cudagraph_runtime_mode=cudagraph_runtime_mode,