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

Fr4nk1inCs · web-flow · commit c483377ed743 · 2025-03-28T10:35:02.000-07:00
diff --git a/python/sglang/srt/managers/scheduler.py b/python/sglang/srt/managers/scheduler.py
@@ -1887,7 +1887,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
`@@ -1887,7 +1887,7 @@ def stop_profile(self) -> None:`
`1887`	`1887`
`1888`	`1888`	`if "MEM" in self.profiler_activities:`
`1889`	`1889`	`memory_profile_path = os.path.join(`
`1890`		`- self.torch_profiler_trace_dir,`
	`1890`	`+ self.torch_profiler_output_dir,`
`1891`	`1891`	`str(time.time()) + f"-TP-{self.tp_rank}-memory" + ".pickle",`
`1892`	`1892`	`)`
`1893`	`1893`	`torch.cuda.memory._dump_snapshot(memory_profile_path)`