Skip to content

Commit 51e5806

Browse files
[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 #3512. ### Does this PR introduce _any_ user-facing change? No ### How was this patch tested? e2e & ut Signed-off-by: Pr0Wh1teGivee <[email protected]>
1 parent cd652ac commit 51e5806

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

vllm_ascend/ops/moe/token_dispatcher.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -178,6 +178,10 @@ def token_dispatch(self,
178178
apply_router_weight_on_input: bool = False,
179179
with_quant: bool = False,
180180
dynamic_eplb: bool = False):
181+
# Apply log2phy if needed
182+
if log2phy is not None:
183+
topk_ids = log2phy[topk_ids]
184+
181185
self.with_quant = with_quant
182186
self.expert_map = expert_map
183187
self.topk_ids = topk_ids

0 commit comments

Comments
 (0)