From 446b03c742e593c5038c74bc07c7ddf484d737d3 Mon Sep 17 00:00:00 2001 From: malay-nagda <164242706+malay-nagda@users.noreply.github.com> Date: Mon, 23 Dec 2024 23:07:32 +0530 Subject: [PATCH] job dir for vboost stdout (#11703) Signed-off-by: Malay Nagda --- nemo/lightning/run/plugins.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nemo/lightning/run/plugins.py b/nemo/lightning/run/plugins.py index 7118d6de0831..b12f04f3bf60 100644 --- a/nemo/lightning/run/plugins.py +++ b/nemo/lightning/run/plugins.py @@ -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)