From 43878386d0418fd7e6be88fae98ae94c89c12a8c Mon Sep 17 00:00:00 2001 From: Fr4nk1in Date: Wed, 26 Mar 2025 06:50:27 +0800 Subject: [PATCH] Fix wrong variable name when stopping memory profile --- 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 f1f0b896fdce..d51c64ad33b3 100644 --- a/python/sglang/srt/managers/scheduler.py +++ b/python/sglang/srt/managers/scheduler.py @@ -1878,7 +1878,7 @@ def stop_profile(self) -> None: if "MEM" in self.torch_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)