no line buffer for mpi jobs (#1825)

2025-09-20 03:48:15 +08:00 · 2025-02-03 12:02:15 -08:00
parent 2d8e667400
commit ec7c7def40
1 changed files with 2 additions and 0 deletions
--- a/python/mlx/distributed_run.py
+++ b/python/mlx/distributed_run.py
@@ -253,6 +253,8 @@ def launch_mpi(parser, hosts, args, command):

        cmd = [
            mpirun,
+            "--output",
+            ":raw",  # do not line buffer output
            "--hostfile",
            f.name,
            *(["-cwd", args.cwd] if args.cwd else []),