"tests/compile/distributed/__init__.py" did not exist on "e575df33b1963212f79b2efddd45d53225fa5c45"
Commit 29ee4475 authored by zhuwenwen's avatar zhuwenwen
Browse files

update fused_experts_impl_w4a8

parent 0975d9e8
......@@ -1585,7 +1585,9 @@ def fused_experts_impl(
a1_scale=a1_scale,
a2_scale=a2_scale,
block_shape=block_shape,
use_nn_moe= False
use_nn_moe= False,
shared_output=shared_output,
routed_scaling_factor=routed_scaling_factor,
)
#
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment