zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Disable echo for interactive distributed

Browse Source
This commit is contained in:
Angelos Katharopoulos
2025-12-11 17:47:42 -08:00
parent d3a754c8aa
commit 753c6a4d0f
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/mlx/_distributed_utils/launch.py
View File

@@ -107,6 +107,9 @@ class RemoteProcess(CommandProcess):
def make_launch_script(rank, cwd, files, env, command): def make_launch_script(rank, cwd, files, env, command):
script = "" script = ""
# Disable echo
script = "stty -echo; "
# Write the PID to a file so we can kill the process if needed # Write the PID to a file so we can kill the process if needed
script += "pidfile=$(mktemp); " script += "pidfile=$(mktemp); "
script += "echo $$ > $pidfile; " script += "echo $$ > $pidfile; "