From e97ff8061344d1dbf954ad3c3174ab3b576aaf33 Mon Sep 17 00:00:00 2001 From: "Roberto L. Castro" <38211239+LopezCastroRoberto@users.noreply.github.com> Date: Wed, 16 Sep 2026 16:24:59 +0200 Subject: [PATCH] [BugFix][PCP] Handle missing DP metadata in one-sided EP (#54016) Signed-off-by: LopezCastroRoberto Co-authored-by: Lucas Wilkinson Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com> --- .../prepare_finalize/flashinfer_nvlink_one_sided.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/vllm/model_executor/layers/fused_moe/prepare_finalize/flashinfer_nvlink_one_sided.py b/vllm/model_executor/layers/fused_moe/prepare_finalize/flashinfer_nvlink_one_sided.py index 0b57afb606..d7a84c931b 100644 --- a/vllm/model_executor/layers/fused_moe/prepare_finalize/flashinfer_nvlink_one_sided.py +++ b/vllm/model_executor/layers/fused_moe/prepare_finalize/flashinfer_nvlink_one_sided.py @@ -13,9 +13,10 @@ from vllm.model_executor.layers.fused_moe.utils import moe_kernel_quantize_input from vllm.utils.flashinfer import nvfp4_block_scale_interleave -def get_local_sizes(): +def get_local_sizes() -> list[int] | None: dp_metadata = get_forward_context().dp_metadata - assert dp_metadata is not None + if dp_metadata is None: # PCP with DP=1 + return None return dp_metadata.get_chunk_sizes_across_dp_rank()