Unverified Commit d848800e authored by Cyrus Leung's avatar Cyrus Leung Committed by GitHub
Browse files

[Misc] Move `print_*_once` from utils to logger (#11298)


Signed-off-by: default avatarDarkLight1337 <tlleungac@connect.ust.hk>
Signed-off-by: default avatarMaxime Fournioux <55544262+mfournioux@users.noreply.github.com>
Co-authored-by: default avatarMaxime Fournioux <55544262+mfournioux@users.noreply.github.com>
parent 730e9592
...@@ -696,18 +696,6 @@ def create_kv_caches_with_random( ...@@ -696,18 +696,6 @@ def create_kv_caches_with_random(
return key_caches, value_caches return key_caches, value_caches
@lru_cache
def print_info_once(msg: str) -> None:
# Set the stacklevel to 2 to print the caller's line info
logger.info(msg, stacklevel=2)
@lru_cache
def print_warning_once(msg: str) -> None:
# Set the stacklevel to 2 to print the caller's line info
logger.warning(msg, stacklevel=2)
@lru_cache(maxsize=None) @lru_cache(maxsize=None)
def is_pin_memory_available() -> bool: def is_pin_memory_available() -> bool:
from vllm.platforms import current_platform from vllm.platforms import current_platform
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment