Align distributed_runner with openmpi 5.0 (#2320) (#2341)

AKloniecki · astachowiczhabana · Adam Stachowicz · web-flow · commit e2118d037b5c · 2025-11-06T16:17:58.000+01:00
Co-authored-by: Adam Stachowicz &lt;105052242+astachowiczhabana@users.noreply.github.com&gt;
Co-authored-by: Adam Stachowicz &lt;astachow@habana.ai&gt;
diff --git a/optimum/habana/distributed/distributed_runner.py b/optimum/habana/distributed/distributed_runner.py
@@ -161,7 +161,7 @@ def create_single_node_setup_mpirun(self):
 
         mpi_cmd = self.setup_config_env_mpirun()
         self._interpreter = (
-            f"mpirun -n {self._world_size} --bind-to core {mpi_cmd} --rank-by core --report-bindings"
+            f"mpirun -n {self._world_size} --bind-to core {mpi_cmd} --rank-by slot --report-bindings"
             f" --allow-run-as-root {sys.executable} "
         )
 

Original file line number	Diff line number	Diff line change
`@@ -161,7 +161,7 @@ def create_single_node_setup_mpirun(self):`
`161`	`161`
`162`	`162`	`mpi_cmd = self.setup_config_env_mpirun()`
`163`	`163`	`self._interpreter = (`
`164`		`- f"mpirun -n {self._world_size} --bind-to core {mpi_cmd} --rank-by core --report-bindings"`
	`164`	`+ f"mpirun -n {self._world_size} --bind-to core {mpi_cmd} --rank-by slot --report-bindings"`
`165`	`165`	`f" --allow-run-as-root {sys.executable} "`
`166`	`166`	`)`
`167`	`167`