[BugFix] fix outplace_fused_experts missing is_gated (#13864)
This commit is contained in:
@@ -325,6 +325,7 @@ def fused_experts(
|
|||||||
b1,
|
b1,
|
||||||
b2,
|
b2,
|
||||||
moe_runner_config.activation,
|
moe_runner_config.activation,
|
||||||
|
moe_runner_config.is_gated,
|
||||||
moe_runner_config.apply_router_weight_on_input,
|
moe_runner_config.apply_router_weight_on_input,
|
||||||
use_fp8_w8a8,
|
use_fp8_w8a8,
|
||||||
use_int8_w8a8,
|
use_int8_w8a8,
|
||||||
|
|||||||
Reference in New Issue
Block a user