[BugFix][PCP] Handle missing DP metadata in one-sided EP (#54016)

Signed-off-by: LopezCastroRoberto <robertol.c510@gmail.com>
Co-authored-by: Lucas Wilkinson <lwilkins@redhat.com>
Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
This commit is contained in:
Roberto L. Castro
2026-09-16 10:24:59 -04:00
committed by GitHub
co-authored by Lucas Wilkinson mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
parent ca542ee154
commit e97ff80613
@@ -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()