[0.11.0-dev][Bugfix][EPLB] Quick fix for missing log2phy conversion (#4150)

### What this PR does / why we need it?
Quick fix for missing log2phy conversion in MC2 token_dispatcher, which
has been already fixed in main branch
https://github.com/vllm-project/vllm-ascend/pull/3512.

### Does this PR introduce _any_ user-facing change?
No

### How was this patch tested?
e2e & ut

Signed-off-by: Pr0Wh1teGivee <calvin_zhu0210@outlook.com>
This commit is contained in:
weichen
2025-11-13 14:32:40 +08:00
committed by GitHub
parent cd652acb65
commit 51e5806d76

View File

@@ -178,6 +178,10 @@ class TokenDispatcherWithMC2(MoETokenDispatcher):
apply_router_weight_on_input: bool = False,
with_quant: bool = False,
dynamic_eplb: bool = False):
# Apply log2phy if needed
if log2phy is not None:
topk_ids = log2phy[topk_ids]
self.with_quant = with_quant
self.expert_map = expert_map
self.topk_ids = topk_ids