Update logging for inplace setting in MoE layer (#25499)

2026-05-16 17:12:11 -07:00
parent 0c017db916
commit 229cadec04
1 changed files with 4 additions and 3 deletions
@@ -312,9 +312,10 @@ class FusedMoE(torch.nn.Module):
            get_moe_runner_backend().is_flashinfer_trtllm_routed()
            or get_moe_runner_backend().is_flashinfer_trtllm()
        ):
-            logging.warning(
-                "Setting inplace to False for FlashInfer TRTLLM MoE backend."
-            )
+            if self.moe_runner_config.inplace:
+                logging.info(
+                    "Setting inplace to False for FlashInfer TRTLLM MoE backend."
+                )
            self.moe_runner_config.inplace = False

        self.should_fuse_routed_scaling_factor_in_topk = (