fix fine grained callable test

Signed-off-by: Jimmy Zhang <[email protected]>

Author: jiemingz

megatron/core/models/gpt/fine_grained_callables.py

@@ -431,7 +431,9 @@ def submodule_combine_forward(
         """
         residual = node.layer_state.residual
 
-        output = layer.mlp.combine(output, shared_expert_output)
+        output = layer.mlp.combine(output)
+        output = layer.mlp.postprocess(output, shared_expert_output)
+
         mlp_output_with_bias = (output, None)
 
         with layer.bias_dropout_add_exec_handler():