Unverified Commit c483377e authored by Fr4nk1in's avatar Fr4nk1in Committed by GitHub
Browse files

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

parent 74e0ac1d
......@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment