跳到内容

vllm_gaudi.extension.debug

init_debug_logger

init_debug_logger(area: str)

如果当前正在调试区域,则初始化日志记录函数

源代码位于 vllm_gaudi/extension/debug.py
def init_debug_logger(area: str):
    """ Initialize logging function if area is currently debugged """
    from vllm_gaudi.extension.runtime import get_config
    dbg_flags: list[str] = get_config().VLLM_DEBUG or []
    if area in dbg_flags:

        def enabled_dbg_logger(msg):
            logger().warning(f'[{area}] {msg}')

        return enabled_dbg_logger
    return None