Skip to content

Commit 0a1c2aa

Browse files
committed
fixing nsys prefix
Signed-off-by: Zoey Zhang <[email protected]>
1 parent 51aa582 commit 0a1c2aa

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

nemo_run/core/execution/dgxcloud.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -361,6 +361,11 @@ def assign(
361361
)
362362
self.experiment_id = exp_id
363363

364+
def get_launcher_prefix(self) -> Optional[list[str]]:
365+
launcher = self.get_launcher()
366+
if launcher.nsys_profile:
367+
return launcher.get_nsys_prefix(profile_dir="/nemo_run")
368+
364369
def package_configs(self, *cfgs: tuple[str, str]) -> list[str]:
365370
filenames = []
366371
basepath = os.path.join(self.job_dir, "configs")

0 commit comments

Comments
 (0)