zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-25 18:11:15 +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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show Stats 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 []),