no line buffer for mpi jobs (#1825)

2025-06-24 17:31:16 +08:00 · 2025-02-03 12:02:15 -08:00 · 2025-02-03 12:02:15 -08:00 · ec7c7def40
commit ec7c7def40
parent 2d8e667400
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 []),