run _select_moe_comm_method only in MoE model

Signed-off-by: realliujiaxu <realliujiaxu@163.com>

Author: realliujiaxu

vllm_ascend/ops/moe/moe_comm_method.py

@@ -39,7 +39,7 @@
 
 def get_moe_comm_method(
         moe_comm_type: Optional[MoECommType]) -> Optional[MoECommMethod]:
-    return _MoECommMethods.get(moe_comm_type)
+    return _MoECommMethods.get(moe_comm_type, None)
 
 
 def setup_moe_comm_method(moe_config):

vllm_ascend/worker/model_runner_v1.py

@@ -1824,7 +1824,7 @@ def _pool(
         )
 
     def _select_moe_comm_method(self, num_tokens: int,
-                                with_prefill: bool) -> MoECommType:
+                                with_prefill: bool) -> Optional[MoECommType]:
         """1. If expert parallel is not enabled, we use all-gather since MC2 and all-to-all
         are designed for expert parallelism.
         2. If expert parallel is enabled, we need to consider the soc version and the
@@ -1847,6 +1847,9 @@ def _select_moe_comm_method(self, num_tokens: int,
         Returns:
             MoECommType: The selected MoE communication method.
         """
+        if not is_moe_model(self.vllm_config):
+            return None
+
         soc_version = get_ascend_soc_version()
         quant_type = getattr(self.vllm_config.model_config.hf_config,
                              'moe_quantize', None)