xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-07-10 02:37:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Core][Logging] Add last frame information for better debugging (#4278)

Browse Source
This commit is contained in:
youkaichao 2024-04-23 09:45:52 -07:00 committed by GitHub
parent d87f39e9a9
commit d86285a4a4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
vllm/logger.py
View File

@ -83,13 +83,27 @@ def _trace_calls(log_path, root_dir, frame, event, arg=None):
return return
# Log every function call or return # Log every function call or return
try: try:
last_frame = frame.f_back
if last_frame is not None:
last_filename = last_frame.f_code.co_filename
last_lineno = last_frame.f_lineno
last_func_name = last_frame.f_code.co_name
else:
# initial frame
last_filename = ""
last_lineno = 0
last_func_name = ""
with open(log_path, 'a') as f: with open(log_path, 'a') as f:
if event == 'call': if event == 'call':
f.write(f"{datetime.datetime.now()} Call to" f.write(f"{datetime.datetime.now()} Call to"
f" {func_name} in {filename}:{lineno}\n") f" {func_name} in {filename}:{lineno}"
f" from {last_func_name} in {last_filename}:"
f"{last_lineno}\n")
else: else:
f.write(f"{datetime.datetime.now()} Return from" f.write(f"{datetime.datetime.now()} Return from"
f" {func_name} in {filename}:{lineno}\n") f" {func_name} in {filename}:{lineno}"
f" to {last_func_name} in {last_filename}:"
f"{last_lineno}\n")
except NameError: except NameError:
# modules are deleted during shutdown # modules are deleted during shutdown
pass pass