diff --git a/nemo_run/core/execution/launcher.py b/nemo_run/core/execution/launcher.py index 980ad87c..cad0ebd3 100644 --- a/nemo_run/core/execution/launcher.py +++ b/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