We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent c8750ea commit c58ab6dCopy full SHA for c58ab6d
slime/utils/profile_utils.py
@@ -94,14 +94,12 @@ def create(args):
94
return c(args)
95
96
def __init__(self, args):
97
- # Use absolute path to ensure it works across Ray actors with different working directories
98
snapshot_dir = Path(args.memory_snapshot_dir).resolve()
99
snapshot_dir.mkdir(parents=True, exist_ok=True)
100
self._path_dump = (
101
snapshot_dir
102
/ f"memory_snapshot_time{time.time()}_rank{torch.distributed.get_rank()}_{args.memory_snapshot_path}"
103
)
104
- logger.info(f"Memory snapshot will be saved to: {self._path_dump} (absolute path)")
105
106
def start(self):
107
raise NotImplementedError
0 commit comments