correct nsys path (#131)

malay-nagda · web-flow · commit 8b713654181a · 2024-12-31T09:55:19.000-08:00
diff --git a/src/nemo_run/core/execution/docker.py b/src/nemo_run/core/execution/docker.py
@@ -209,7 +209,7 @@ def package(self, packager: Packager, job_name: str):
 
         if self.get_launcher().nsys_profile:
             remote_nsys_extraction_path = os.path.join(
-                self.job_dir, job_name, self.get_launcher().nsys_folder
+                self.job_dir, self.get_launcher().nsys_folder
             )
             ctx.run(f"mkdir -p {remote_nsys_extraction_path}")
         if local_pkg:
diff --git a/src/nemo_run/core/execution/skypilot.py b/src/nemo_run/core/execution/skypilot.py
@@ -302,7 +302,7 @@ def package(self, packager: Packager, job_name: str):
 
         if self.get_launcher().nsys_profile:
             remote_nsys_extraction_path = os.path.join(
-                self.job_dir, job_name, self.get_launcher().nsys_folder
+                self.job_dir, self.get_launcher().nsys_folder
             )
             ctx.run(f"mkdir -p {remote_nsys_extraction_path}")
         if local_pkg:
diff --git a/src/nemo_run/core/execution/slurm.py b/src/nemo_run/core/execution/slurm.py
@@ -595,7 +595,7 @@ def package(self, packager: Packager, job_name: str):
 
         if self.get_launcher().nsys_profile:
             remote_nsys_extraction_path = os.path.join(
-                self.job_dir, job_name, self.get_launcher().nsys_folder
+                self.job_dir, self.get_launcher().nsys_folder
             )
             ctx.run(f"mkdir -p {remote_nsys_extraction_path}")
             # Touch hidden init file

Original file line number	Diff line number	Diff line change
`@@ -209,7 +209,7 @@ def package(self, packager: Packager, job_name: str):`
`209`	`209`
`210`	`210`	`if self.get_launcher().nsys_profile:`
`211`	`211`	`remote_nsys_extraction_path = os.path.join(`
`212`		`- self.job_dir, job_name, self.get_launcher().nsys_folder`
	`212`	`+ self.job_dir, self.get_launcher().nsys_folder`
`213`	`213`	`)`
`214`	`214`	`ctx.run(f"mkdir -p {remote_nsys_extraction_path}")`
`215`	`215`	`if local_pkg:`
Original file line number	Diff line number	Diff line change
`@@ -302,7 +302,7 @@ def package(self, packager: Packager, job_name: str):`
`302`	`302`
`303`	`303`	`if self.get_launcher().nsys_profile:`
`304`	`304`	`remote_nsys_extraction_path = os.path.join(`
`305`		`- self.job_dir, job_name, self.get_launcher().nsys_folder`
	`305`	`+ self.job_dir, self.get_launcher().nsys_folder`
`306`	`306`	`)`
`307`	`307`	`ctx.run(f"mkdir -p {remote_nsys_extraction_path}")`
`308`	`308`	`if local_pkg:`
Original file line number	Diff line number	Diff line change
`@@ -595,7 +595,7 @@ def package(self, packager: Packager, job_name: str):`
`595`	`595`
`596`	`596`	`if self.get_launcher().nsys_profile:`
`597`	`597`	`remote_nsys_extraction_path = os.path.join(`
`598`		`- self.job_dir, job_name, self.get_launcher().nsys_folder`
	`598`	`+ self.job_dir, self.get_launcher().nsys_folder`
`599`	`599`	`)`
`600`	`600`	`ctx.run(f"mkdir -p {remote_nsys_extraction_path}")`
`601`	`601`	`# Touch hidden init file`