Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
IlyasMoutawwakil committed Nov 26, 2024
1 parent 30a4472 commit f4ee5b4
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions optimum_benchmark/launchers/torchrun/launcher.py
Original file line number Diff line number Diff line change
Expand Up @@ -105,7 +105,7 @@ def launch(self, worker: Callable[..., BenchmarkReport], worker_args: List[Any])
raise RuntimeError(f"Received an unexpected response from isolated process: {output}")

self.logger.info("\t+ Aggregating reports from all rank processes")
report = BenchmarkReport.aggregate_across_processes(reports)
report = BenchmarkReport.aggregate(reports)
return report


Expand Down Expand Up @@ -159,8 +159,6 @@ def entrypoint(worker: Callable[..., BenchmarkReport], worker_args: List[Any], l
else:
setup_logging(level="ERROR", to_file=log_to_file, prefix=f"RANK-PROCESS-{rank}")



if torch.cuda.is_available():
logger.info(f"\t+ Setting torch.distributed cuda device to {rank}")
device = torch.device("cuda", rank)
Expand Down

0 comments on commit f4ee5b4

Please sign in to comment.