xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-12 02:25:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Misc] small fixes to function tracing file path (#9543)

Browse Source 
Signed-off-by: Shawn Du <shawnd200@outlook.com>
Signed-off-by: youkaichao <youkaichao@gmail.com>
Co-authored-by: youkaichao <youkaichao@gmail.com>
This commit is contained in:
Shawn Du 2024-11-11 07:21:06 +08:00 committed by GitHub
parent bfb7d61a7c
commit 20cf2f553c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
docs/source/index.rst
View File

@ -161,11 +161,11 @@ Documentation
design/multimodal/multimodal_index design/multimodal/multimodal_index
design/huggingface_integration design/huggingface_integration
.. Contributing: contributing to the vLLM project .. For Developers: contributing to the vLLM project
.. toctree:: .. toctree::
:maxdepth: 2 :maxdepth: 2
:caption: Contributing :caption: For Developers
contributing/overview contributing/overview
contributing/profiling/profiling_index contributing/profiling/profiling_index

5
vllm/logger.py
View File

@ -117,13 +117,14 @@ def _trace_calls(log_path, root_dir, frame, event, arg=None):
last_lineno = 0 last_lineno = 0
last_func_name = "" last_func_name = ""
with open(log_path, 'a') as f: with open(log_path, 'a') as f:
ts = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S.%f")
if event == 'call': if event == 'call':
f.write(f"{datetime.datetime.now()} Call to" f.write(f"{ts} Call to"
f" {func_name} in {filename}:{lineno}" f" {func_name} in {filename}:{lineno}"
f" from {last_func_name} in {last_filename}:" f" from {last_func_name} in {last_filename}:"
f"{last_lineno}\n") f"{last_lineno}\n")
else: else:
f.write(f"{datetime.datetime.now()} Return from" f.write(f"{ts} Return from"
f" {func_name} in {filename}:{lineno}" f" {func_name} in {filename}:{lineno}"
f" to {last_func_name} in {last_filename}:" f" to {last_func_name} in {last_filename}:"
f"{last_lineno}\n") f"{last_lineno}\n")

3
vllm/utils.py
View File

@ -4,6 +4,7 @@ import contextlib
import datetime import datetime
import enum import enum
import gc import gc
import getpass
import inspect import inspect
import ipaddress import ipaddress
import os import os
@ -967,6 +968,8 @@ def enable_trace_function_call_for_thread() -> None:
if envs.VLLM_TRACE_FUNCTION: if envs.VLLM_TRACE_FUNCTION:
tmp_dir = tempfile.gettempdir() tmp_dir = tempfile.gettempdir()
# add username to tmp_dir to avoid permission issues
tmp_dir = os.path.join(tmp_dir, getpass.getuser())
filename = (f"VLLM_TRACE_FUNCTION_for_process_{os.getpid()}" filename = (f"VLLM_TRACE_FUNCTION_for_process_{os.getpid()}"
f"_thread_{threading.get_ident()}_" f"_thread_{threading.get_ident()}_"
f"at_{datetime.datetime.now()}.log").replace(" ", "_") f"at_{datetime.datetime.now()}.log").replace(" ", "_")