Skip to content

Commit b09c435

Browse files
committed
Add logs dir to container mount for ray slurm
Signed-off-by: Hemil Desai <hemild@nvidia.com>
1 parent 143b17c commit b09c435

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

nemo_run/run/ray/slurm.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -214,6 +214,7 @@ def get_srun_flags(mounts: list[str], container_image: Optional[str]) -> str:
214214
new_mounts = mounts
215215

216216
new_mounts.append(f"{self.cluster_dir}:{self.cluster_dir}")
217+
new_mounts.append(f"{logs_dir}:{logs_dir}")
217218

218219
_srun_flags += ["--container-mounts", ",".join(new_mounts)]
219220
container_workdir = self.workdir or self.cluster_dir

0 commit comments

Comments
 (0)