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

2025-03-28 10:35:02 -07:00
parent 74e0ac1dbd
commit c483377ed7
1 changed files with 1 additions and 1 deletions
@@ -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)