[Diff]: support SGLANG_TORCH_PROFILER_DIR environment variable for profiler log directory (#18454)
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
co-authored by
gemini-code-assist[bot]
parent
0af9dcc407
commit
bc79a64d3a
@@ -27,12 +27,17 @@ class SGLDiffusionProfiler:
|
|||||||
full_profile: bool = False,
|
full_profile: bool = False,
|
||||||
num_steps: int | None = None,
|
num_steps: int | None = None,
|
||||||
num_inference_steps: int | None = None,
|
num_inference_steps: int | None = None,
|
||||||
log_dir: str = "./logs",
|
log_dir: str | None = None,
|
||||||
):
|
):
|
||||||
self.request_id = request_id or "profile_trace"
|
self.request_id = request_id or "profile_trace"
|
||||||
self.rank = rank
|
self.rank = rank
|
||||||
self.full_profile = full_profile
|
self.full_profile = full_profile
|
||||||
self.log_dir = log_dir
|
|
||||||
|
self.log_dir = (
|
||||||
|
log_dir
|
||||||
|
if log_dir is not None
|
||||||
|
else os.getenv("SGLANG_TORCH_PROFILER_DIR", "./logs")
|
||||||
|
)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
os.makedirs(self.log_dir, exist_ok=True)
|
os.makedirs(self.log_dir, exist_ok=True)
|
||||||
|
|||||||
Reference in New Issue
Block a user