From 59a938ebe59094dd6a4e338ca3907ee666cbc63c Mon Sep 17 00:00:00 2001 From: Malay Nagda Date: Tue, 8 Apr 2025 19:38:22 +0530 Subject: [PATCH 1/2] add nsys filename param Signed-off-by: Malay Nagda --- nemo_run/core/execution/launcher.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/nemo_run/core/execution/launcher.py b/nemo_run/core/execution/launcher.py index f40d52cf..1e3c8dbd 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"]) def get_nsys_prefix(self, profile_dir: str) -> Optional[list[str]]: @@ -26,7 +27,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}", "--force-overwrite", "true", "--capture-range=cudaProfilerApi", From cba5f270dbf94fe471892610b4b75d35b577e305 Mon Sep 17 00:00:00 2001 From: Malay Nagda Date: Tue, 8 Apr 2025 21:38:44 +0530 Subject: [PATCH 2/2] ruff fmt Signed-off-by: Malay Nagda --- nemo_run/package_info.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nemo_run/package_info.py b/nemo_run/package_info.py index 6316ccf3..0682451f 100644 --- a/nemo_run/package_info.py +++ b/nemo_run/package_info.py @@ -13,7 +13,7 @@ # limitations under the License. from packaging.version import Version -__version__ = '0.5.0rc0.dev0' +__version__ = "0.5.0rc0.dev0" MAJOR = Version(__version__).major MINOR = Version(__version__).minor