zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-24 14:58:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

no line buffer for mpi jobs (#1825)

Browse Source
This commit is contained in:
Awni Hannun
2025-02-03 12:02:15 -08:00
committed by GitHub
parent 2d8e667400
commit ec7c7def40
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/mlx/distributed_run.py
View File

@@ -253,6 +253,8 @@ def launch_mpi(parser, hosts, args, command):
cmd = [ cmd = [
mpirun, mpirun,
"--output",
":raw", # do not line buffer output
"--hostfile", "--hostfile",
f.name, f.name,
*(["-cwd", args.cwd] if args.cwd else []), *(["-cwd", args.cwd] if args.cwd else []),