EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Fr4nk1in
2025-03-29 01:35:02 +08:00
committed by GitHub
parent 74e0ac1dbd
commit c483377ed7
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/sglang/srt/managers/scheduler.py
View File

@@ -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)