Skip to content

Commit 8914d5a

Browse files
authored
[Quickfix] Add the missing apply_router_weight_on_input in FusedMoE init (#2348)
### What this PR does / why we need it? Add the missing `apply_router_weight_on_input` in FusedMoE init Quick fix on #2268 (comment) ### Does this PR introduce _any_ user-facing change? N/A ### How was this patch tested? CI passed with existing test. - vLLM version: v0.10.0 - vLLM main: vllm-project/vllm@6807af8 Signed-off-by: MengqingCao <[email protected]>
1 parent 0f7492d commit 8914d5a

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

vllm_ascend/ops/fused_moe.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1258,6 +1258,7 @@ def __init__(
12581258
scoring_func=scoring_func,
12591259
e_score_correction_bias=e_score_correction_bias,
12601260
activation=activation,
1261+
apply_router_weight_on_input=apply_router_weight_on_input,
12611262
)
12621263
AscendFusedMoE.moe_counter += 1
12631264
self.moe_instance_id = AscendFusedMoE.moe_counter

0 commit comments

Comments
 (0)