Skip to content

Commit d86285a

Browse files
authored
[Core][Logging] Add last frame information for better debugging (#4278)
1 parent d87f39e commit d86285a

File tree

1 file changed

+16
-2
lines changed

1 file changed

+16
-2
lines changed

vllm/logger.py

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -83,13 +83,27 @@ def _trace_calls(log_path, root_dir, frame, event, arg=None):
8383
return
8484
# Log every function call or return
8585
try:
86+
last_frame = frame.f_back
87+
if last_frame is not None:
88+
last_filename = last_frame.f_code.co_filename
89+
last_lineno = last_frame.f_lineno
90+
last_func_name = last_frame.f_code.co_name
91+
else:
92+
# initial frame
93+
last_filename = ""
94+
last_lineno = 0
95+
last_func_name = ""
8696
with open(log_path, 'a') as f:
8797
if event == 'call':
8898
f.write(f"{datetime.datetime.now()} Call to"
89-
f" {func_name} in {filename}:{lineno}\n")
99+
f" {func_name} in {filename}:{lineno}"
100+
f" from {last_func_name} in {last_filename}:"
101+
f"{last_lineno}\n")
90102
else:
91103
f.write(f"{datetime.datetime.now()} Return from"
92-
f" {func_name} in {filename}:{lineno}\n")
104+
f" {func_name} in {filename}:{lineno}"
105+
f" to {last_func_name} in {last_filename}:"
106+
f"{last_lineno}\n")
93107
except NameError:
94108
# modules are deleted during shutdown
95109
pass

0 commit comments

Comments
 (0)