Fix logging for inplace setting in the flashInfer-trtllm backend (#25522)
This commit is contained in:
@@ -70,6 +70,7 @@ from sglang.srt.utils import (
|
|||||||
get_bool_env_var,
|
get_bool_env_var,
|
||||||
is_cpu,
|
is_cpu,
|
||||||
is_hip,
|
is_hip,
|
||||||
|
print_info_once,
|
||||||
round_up,
|
round_up,
|
||||||
)
|
)
|
||||||
from sglang.srt.utils.custom_op import register_custom_op
|
from sglang.srt.utils.custom_op import register_custom_op
|
||||||
@@ -313,7 +314,7 @@ class FusedMoE(torch.nn.Module):
|
|||||||
or get_moe_runner_backend().is_flashinfer_trtllm()
|
or get_moe_runner_backend().is_flashinfer_trtllm()
|
||||||
):
|
):
|
||||||
if self.moe_runner_config.inplace:
|
if self.moe_runner_config.inplace:
|
||||||
logging.info(
|
print_info_once(
|
||||||
"Setting inplace to False for FlashInfer TRTLLM MoE backend."
|
"Setting inplace to False for FlashInfer TRTLLM MoE backend."
|
||||||
)
|
)
|
||||||
self.moe_runner_config.inplace = False
|
self.moe_runner_config.inplace = False
|
||||||
|
|||||||
Reference in New Issue
Block a user