mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-04-08 23:27:09 +08:00
bf8b26cad1
Signed-off-by: Manoel Marques <manoel.marques@ibm.com> Signed-off-by: Manoel Marques <manoelmrqs@gmail.com> Co-authored-by: Harry Mellor <19981378+hmellor@users.noreply.github.com> Co-authored-by: Luka Govedič <ProExpertProg@users.noreply.github.com>
33 lines
819 B
Python
33 lines
819 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
"""
|
|
Provides a timeslice logging decorator
|
|
"""
|
|
|
|
import functools
|
|
import time
|
|
|
|
|
|
def logtime(logger, msg=None):
|
|
"""
|
|
Logs the execution time of the decorated function.
|
|
Always place it beneath other decorators.
|
|
"""
|
|
|
|
def _inner(func):
|
|
|
|
@functools.wraps(func)
|
|
def _wrapper(*args, **kwargs):
|
|
start = time.perf_counter()
|
|
result = func(*args, **kwargs)
|
|
elapsed = time.perf_counter() - start
|
|
|
|
prefix = f"Function '{func.__module__}.{func.__qualname__}'" \
|
|
if msg is None else msg
|
|
logger.debug("%s: Elapsed time %.7f secs", prefix, elapsed)
|
|
return result
|
|
|
|
return _wrapper
|
|
|
|
return _inner
|