From 48811bc0b8e398d2d94c3d8c75fd84dbbf998e31 Mon Sep 17 00:00:00 2001 From: Nengjun Ma Date: Thu, 8 Jan 2026 09:26:49 +0800 Subject: [PATCH] Optimize the print info format when deprecated code is used in vllm-ascend (#5696) ### What this PR does / why we need it? Optimize the warning print information format when detects depredated code is used in vllm-ascend. ### Does this PR introduce _any_ user-facing change? NA - vLLM version: v0.13.0 - vLLM main: https://github.com/vllm-project/vllm/commit/2f4e6548efec402b913ffddc8726230d9311948d --------- Signed-off-by: leo-pony --- vllm_ascend/platform.py | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) diff --git a/vllm_ascend/platform.py b/vllm_ascend/platform.py index 9bdf62aa..06f8be7b 100644 --- a/vllm_ascend/platform.py +++ b/vllm_ascend/platform.py @@ -47,6 +47,36 @@ else: _CUSTOM_OP_REGISTERED = False +def config_deprecated_logging(): + """Configure deprecated logging format, when used deprecated codes + in vllm-ascend. + """ + import logging + import warnings + + # Customize warning format to be one line + def one_line_formatwarning(message, category, filename, lineno, line=None): + return f"{filename}:{lineno}: {category.__name__}: {message}" + + warnings.formatwarning = one_line_formatwarning + + logging.captureWarnings(True) + warnings.simplefilter("once", DeprecationWarning) + + vllm_logger = logging.getLogger("vllm") + warnings_logger = logging.getLogger("py.warnings") + + # Propagate vllm logger handlers to warnings logger, to keep the same + # format with vllm + if vllm_logger.handlers: + warnings_logger.handlers = [] + + for handler in vllm_logger.handlers: + warnings_logger.addHandler(handler) + + warnings_logger.propagate = False + + class NPUPlatform(Platform): _enum = PlatformEnum.OOT @@ -112,6 +142,8 @@ class NPUPlatform(Platform): from vllm_ascend.quantization.quant_config import \ AscendQuantConfig # noqa: F401 + config_deprecated_logging() + @classmethod def get_device_capability(cls, device_id: int = 0): return None