Skip to content

vllm_gaudi.extension.debug

init_debug_logger 

init_debug_logger(area: str)

Initialize logging function if area is currently debugged

Source code in vllm_gaudi/extension/debug.py 
11
12
13
14
15
16
17
18
19
20
21
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