fix plumbing

Signed-off-by: varun sundar rabindranath <vsundarr@redhat.com>

Author: varun sundar rabindranath

vllm/model_executor/layers/fused_moe/layer.py

@@ -437,9 +437,9 @@ def apply(
         custom_routing_function: Optional[Callable] = None,
         scoring_func: str = "softmax",
         e_score_correction_bias: Optional[torch.Tensor] = None,
+        apply_router_weight_on_input: bool = False,
         activation: str = "silu",
     ) -> torch.Tensor:
-
         assert activation == "silu"
         assert global_num_experts == layer.w13_weight.shape[0]
         assert expert_map is None
@@ -466,6 +466,7 @@ def apply(
             self.c_strides1,
             self.ab_strides2,
             self.c_strides2,
+            apply_router_weight_on_input=apply_router_weight_on_input,
         )