[Bugfix] Fix mc2 operator error in aclgraph + ep<16 scenario (#2609)

### What this PR does / why we need it?
1. quickfix mc2 operator error in aclgraph + ep<16 scenario to recover
CI, will be refactorred in the future
2. disable aclgraph when testing w8a8

### How was this patch tested?
CI passed with existing test.

- vLLM version: v0.10.1.1
- vLLM main:
95089607fa

Signed-off-by: MengqingCao <cmq0113@163.com>
This commit is contained in:
Mengqing Cao
2025-08-29 21:59:16 +08:00
committed by GitHub
parent ee6d141dd4
commit 91c35d765a
2 changed files with 4 additions and 2 deletions

View File

@@ -242,7 +242,8 @@ class AscendFusedMoE(FusedMoE):
moe_comm_method_name = forward_context.moe_comm_method_name
# TODO: Can we refactor this logic to model_runner?
if not self.moe_config.use_ep:
# TODO: Adjusted logic to differentiate between A2 and A3, we check ep_size here since mc2 only support ep_size >= 16 on A3 now
if self.moe_config.ep_size < 16:
moe_comm_method_name = "allgathercommimpl"
forward_context.moe_comm_method = getattr(self, moe_comm_method_name)