Update logging for inplace setting in MoE layer (#25499)
This commit is contained in:
@@ -312,9 +312,10 @@ class FusedMoE(torch.nn.Module):
|
|||||||
get_moe_runner_backend().is_flashinfer_trtllm_routed()
|
get_moe_runner_backend().is_flashinfer_trtllm_routed()
|
||||||
or get_moe_runner_backend().is_flashinfer_trtllm()
|
or get_moe_runner_backend().is_flashinfer_trtllm()
|
||||||
):
|
):
|
||||||
logging.warning(
|
if self.moe_runner_config.inplace:
|
||||||
"Setting inplace to False for FlashInfer TRTLLM MoE backend."
|
logging.info(
|
||||||
)
|
"Setting inplace to False for FlashInfer TRTLLM MoE backend."
|
||||||
|
)
|
||||||
self.moe_runner_config.inplace = False
|
self.moe_runner_config.inplace = False
|
||||||
|
|
||||||
self.should_fuse_routed_scaling_factor_in_topk = (
|
self.should_fuse_routed_scaling_factor_in_topk = (
|
||||||
|
|||||||
Reference in New Issue
Block a user