Harry MellorandClaude Opus 5
8c1557a79c
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com> Signed-off-by: Harry Mellor <hej.mellor@gmail.com> Co-authored-by: Claude Opus 5 (1M context) <noreply@anthropic.com>
32 lines
858 B
Python
32 lines
858 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
|