From ec7c7def4038e3d787ed2452742d513fb3c83eaf Mon Sep 17 00:00:00 2001 From: Awni Hannun Date: Mon, 3 Feb 2025 12:02:15 -0800 Subject: [PATCH] no line buffer for mpi jobs (#1825) --- python/mlx/distributed_run.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/mlx/distributed_run.py b/python/mlx/distributed_run.py index 2cf74a038..828e22efb 100644 --- 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 []),