diff --git a/src/cloudai/workloads/megatron_bridge/report_generation_strategy.py b/src/cloudai/workloads/megatron_bridge/report_generation_strategy.py index 9aa3c2d34..06610f5e0 100644 --- a/src/cloudai/workloads/megatron_bridge/report_generation_strategy.py +++ b/src/cloudai/workloads/megatron_bridge/report_generation_strategy.py @@ -1,5 +1,5 @@ # SPDX-FileCopyrightText: NVIDIA CORPORATION & AFFILIATES -# Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 # # Licensed under the Apache License, Version 2.0 (the "License"); @@ -31,12 +31,12 @@ class MegatronBridgeReportGenerationStrategy(ReportGenerationStrategy): metrics: ClassVar[list[str]] = ["default", "step-time", "tflops-per-gpu"] def get_log_file(self) -> Path | None: - log = self.test_run.output_path / "megatron_bridge_launcher.log" + log = self.test_run.output_path / "cloudai_megatron_bridge_launcher.log" return log if log.is_file() else None @property def results_file(self) -> Path: - return self.get_log_file() or (self.test_run.output_path / "megatron_bridge_launcher.log") + return self.get_log_file() or (self.test_run.output_path / "cloudai_megatron_bridge_launcher.log") def can_handle_directory(self) -> bool: return self.get_log_file() is not None @@ -75,8 +75,8 @@ def generate_report(self) -> None: log_file, step_times_s, gpu_tflops = self._get_extracted_data() if not log_file: logging.error( - "No Megatron-Bridge launcher log file found: %s", - self.test_run.output_path / "megatron_bridge_launcher.log", + "No Megatron-Bridge launcher log file found in: %s", + self.test_run.output_path, ) return @@ -130,8 +130,8 @@ def get_metric(self, metric: str) -> float: log_file, step_times_s, gpu_tflops = self._get_extracted_data() if not log_file: logging.error( - "No Megatron-Bridge launcher log file found: %s", - self.test_run.output_path / "megatron_bridge_launcher.log", + "No Megatron-Bridge launcher log file found in: %s", + self.test_run.output_path, ) return METRIC_ERROR if not step_times_s: diff --git a/tests/report_generation_strategy/test_megatron_bridge_report_generation_strategy.py b/tests/report_generation_strategy/test_megatron_bridge_report_generation_strategy.py index 78c3a3055..a59e3a1fb 100644 --- a/tests/report_generation_strategy/test_megatron_bridge_report_generation_strategy.py +++ b/tests/report_generation_strategy/test_megatron_bridge_report_generation_strategy.py @@ -1,5 +1,5 @@ # SPDX-FileCopyrightText: NVIDIA CORPORATION & AFFILIATES -# Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 # # Licensed under the Apache License, Version 2.0 (the "License"); @@ -38,7 +38,7 @@ def mb_tr(tmp_path: Path) -> TestRun: "", ] ) - (tr.output_path / "megatron_bridge_launcher.log").write_text(log_content) + (tr.output_path / "cloudai_megatron_bridge_launcher.log").write_text(log_content) return tr