diff --git a/python/sglang/srt/layers/attention/flashinfer_backend.py b/python/sglang/srt/layers/attention/flashinfer_backend.py index a91262823..d568170ce 100644 --- a/python/sglang/srt/layers/attention/flashinfer_backend.py +++ b/python/sglang/srt/layers/attention/flashinfer_backend.py @@ -16,8 +16,9 @@ from typing import TYPE_CHECKING, Callable, List, Optional, Union import torch if os.environ["SGLANG_ENABLE_TORCH_COMPILE"] == "1": - import torch._dynamo + import logging + torch._logging.set_logs(dynamo=logging.ERROR) torch._dynamo.config.suppress_errors = True from sglang.global_config import global_config diff --git a/python/sglang/srt/layers/attention/flashinfer_mla_backend.py b/python/sglang/srt/layers/attention/flashinfer_mla_backend.py index 485ed5f2c..dded9a970 100644 --- a/python/sglang/srt/layers/attention/flashinfer_mla_backend.py +++ b/python/sglang/srt/layers/attention/flashinfer_mla_backend.py @@ -18,8 +18,9 @@ import torch import triton if os.environ["SGLANG_ENABLE_TORCH_COMPILE"] == "1": - import torch._dynamo + import logging + torch._logging.set_logs(dynamo=logging.ERROR) torch._dynamo.config.suppress_errors = True from sglang.global_config import global_config