add nsys filename param (#205)

* add nsys filename param

Signed-off-by: Malay Nagda <[email protected]>

* ruff fmt

Signed-off-by: Malay Nagda <[email protected]>

---------

Signed-off-by: Malay Nagda <[email protected]>
Signed-off-by: Hemil Desai <[email protected]>
Co-authored-by: Hemil Desai <[email protected]>

Author: malay-nagda

nemo_run/core/execution/launcher.py

@@ -13,6 +13,7 @@
 class Launcher(ConfigurableMixin):
     nsys_profile: bool = False
     nsys_folder: str = "nsys_profile"
+    nsys_filename: str = "profile_%p"
     nsys_trace: list[str] = field(default_factory=lambda: ["nvtx", "cuda"])
     nsys_extra_args: list[str] = field(
         default_factory=lambda: [
@@ -35,7 +36,7 @@ def get_nsys_prefix(self, profile_dir: str) -> Optional[list[str]]:
                 "-t",
                 ",".join(self.nsys_trace),
                 "-o",
-                f"{profile_out_path}/profile_%p",
+                f"{profile_out_path}/{self.nsys_filename}",
             ] + self.nsys_extra_args
             return args