Skip to content

Commit

Permalink
job dir for vboost stdout (#11703)
Browse files Browse the repository at this point in the history
Signed-off-by: Malay Nagda <[email protected]>
  • Loading branch information
malay-nagda authored Dec 23, 2024
1 parent 17e7101 commit 446b03c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion nemo/lightning/run/plugins.py
Original file line number Diff line number Diff line change
Expand Up @@ -361,7 +361,7 @@ def setup(self, task: run.Partial | run.Script, executor: run.Executor):

# Improve perf by steering power to tensor cores, may not work on all systems
if self.enable_vboost and isinstance(executor, run.SlurmExecutor):
vboost_cmd = self.get_vboost_srun_cmd(executor.nodes, executor.job_dir)
vboost_cmd = self.get_vboost_srun_cmd(executor.nodes, executor.tunnel.job_dir)
executor.setup_lines = (
executor.setup_lines + vboost_cmd
if (executor.setup_lines and len(executor.setup_lines) > 0)
Expand Down

0 comments on commit 446b03c

Please sign in to comment.