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

Fr4nk1inCs · jimoosciuc · commit b3f10f28c7e3 · 2025-04-17T11:58:24.000+08:00
diff --git a/python/sglang/srt/managers/scheduler.py b/python/sglang/srt/managers/scheduler.py
@@ -2127,7 +2127,7 @@ def stop_profile(self) -> None:
 
         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)

Original file line number	Diff line number	Diff line change
`@@ -2127,7 +2127,7 @@ def stop_profile(self) -> None:`
`2127`	`2127`
`2128`	`2128`	`if "MEM" in self.profiler_activities:`
`2129`	`2129`	`memory_profile_path = os.path.join(`
`2130`		`- self.torch_profiler_trace_dir,`
	`2130`	`+ self.torch_profiler_output_dir,`
`2131`	`2131`	`str(time.time()) + f"-TP-{self.tp_rank}-memory" + ".pickle",`
`2132`	`2132`	`)`
`2133`	`2133`	`torch.cuda.memory._dump_snapshot(memory_profile_path)`