Skip to content

GitLab

Unverified Commit c483377e authored by Fr4nk1in's avatar Fr4nk1in Committed by GitHub
Browse files

Fix wrong variable name when stopping memory profile (#4772)

parent 74e0ac1d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
python/sglang/srt/managers/scheduler.py
View file @ c483377e
...@@ -1887,7 +1887,7 @@ class Scheduler( ...@@ -1887,7 +1887,7 @@ class Scheduler(
if "MEM" in self.profiler_activities: if "MEM" in self.profiler_activities:
memory_profile_path = os.path.join( memory_profile_path = os.path.join(
self.torch_profiler_trace_dir, self.torch_profiler_output_dir,
str(time.time()) + f"-TP-{self.tp_rank}-memory" + ".pickle", str(time.time()) + f"-TP-{self.tp_rank}-memory" + ".pickle",
) )
torch.cuda.memory._dump_snapshot(memory_profile_path) torch.cuda.memory._dump_snapshot(memory_profile_path)
......
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