From 3779eb8c81449b924a23457fc77e45a0e6171178 Mon Sep 17 00:00:00 2001 From: "rongfu.leng" Date: Tue, 22 Jul 2025 14:41:14 +0800 Subject: [PATCH] [Feature][eplb] add verify ep or tp or dp (#21102) Signed-off-by: rongfu.leng --- vllm/config.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/vllm/config.py b/vllm/config.py index 3e6aa2a93e6a7..d649eb75033f9 100644 --- a/vllm/config.py +++ b/vllm/config.py @@ -2108,6 +2108,15 @@ class ParallelConfig: raise ValueError( "num_redundant_experts must be non-negative, but got " f"{self.num_redundant_experts}.") + if not self.enable_expert_parallel: + raise ValueError( + "enable_expert_parallel must be True to use EPLB.") + if self.tensor_parallel_size * self.data_parallel_size <= 1: + raise ValueError( + "EPLB requires tensor_parallel_size or data_parallel_size " + f"to be greater than 1, but got " + f"TP={self.tensor_parallel_size},DP={self.data_parallel_size}." + ) else: if self.num_redundant_experts != 0: raise ValueError(