[Mics] add pcp basic support to MoE model (#31003)

2025-12-31 12:01:29 +08:00
parent ecd49ce7e6
commit 84f6cd741b
1 changed files with 8 additions and 0 deletions
--- a/vllm/model_executor/layers/fused_moe/config.py
+++ b/vllm/model_executor/layers/fused_moe/config.py
@@ -1027,6 +1027,10 @@ class FusedMoEConfig:
    def dp_size(self):
        return self.moe_parallel_config.dp_size

+    @property
+    def pcp_size(self):
+        return self.moe_parallel_config.pcp_size
+
    @property
    def ep_size(self):
        return self.moe_parallel_config.ep_size
@@ -1039,6 +1043,10 @@ class FusedMoEConfig:
    def dp_rank(self):
        return self.moe_parallel_config.dp_rank

+    @property
+    def pcp_rank(self):
+        return self.moe_parallel_config.pcp_rank
+
    @property
    def ep_rank(self):
        return self.moe_parallel_config.ep_rank