From c483377ed7438f61178dc03f8d8c14195c413154 Mon Sep 17 00:00:00 2001 From: Fr4nk1in Date: Sat, 29 Mar 2025 01:35:02 +0800 Subject: [PATCH] Fix wrong variable name when stopping memory profile (#4772) --- python/sglang/srt/managers/scheduler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/sglang/srt/managers/scheduler.py b/python/sglang/srt/managers/scheduler.py index f696862da..63acbe2aa 100644 --- a/python/sglang/srt/managers/scheduler.py +++ b/python/sglang/srt/managers/scheduler.py @@ -1887,7 +1887,7 @@ class Scheduler( if "MEM" in self.profiler_activities: 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", ) torch.cuda.memory._dump_snapshot(memory_profile_path)