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)