Skip to content

GitLab

Unverified Commit 34962746 authored by Ning Xie's avatar Ning Xie Committed by GitHub
Browse files

[Misc] Convert VLLM_TORCH_PROFILER_DIR path to absolute (#23191) 


Signed-off-by: default avatarAndy Xie <andy.xning@gmail.com>
parent 8a193031
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
vllm/envs.py
View file @ 34962746
...@@ -673,7 +673,8 @@ environment_variables: dict[str, Callable[[], Any]] = { ...@@ -673,7 +673,8 @@ environment_variables: dict[str, Callable[[], Any]] = {
# Note that it must be an absolute path. # Note that it must be an absolute path.
"VLLM_TORCH_PROFILER_DIR": "VLLM_TORCH_PROFILER_DIR":
lambda: (None if os.getenv("VLLM_TORCH_PROFILER_DIR", None) is None else os lambda: (None if os.getenv("VLLM_TORCH_PROFILER_DIR", None) is None else os
.path.expanduser(os.getenv("VLLM_TORCH_PROFILER_DIR", "."))), .path.abspath(os.path.expanduser(os.getenv(
"VLLM_TORCH_PROFILER_DIR", ".")))),
# Enable torch profiler to record shapes if set # Enable torch profiler to record shapes if set
# VLLM_TORCH_PROFILER_RECORD_SHAPES=1. If not set, torch profiler will # VLLM_TORCH_PROFILER_RECORD_SHAPES=1. If not set, torch profiler will
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment