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

Add mlx.distributed_config script (#1902)

Browse Source
This commit is contained in:
Angelos Katharopoulos
2025-02-28 11:16:39 -08:00
committed by GitHub
parent 89d327075f
commit 607181644f
2 changed files with 402 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
setup.py
View File

@@ -194,7 +194,12 @@ if __name__ == "__main__":
"typing_extensions",
],
},
entry_points={"console_scripts": ["mlx.launch = mlx.distributed_run:main"]},
entry_points={
"console_scripts": [
"mlx.launch = mlx.distributed_run:main",
"mlx.distributed_config = mlx.distributed_run:distributed_config",
]
},
ext_modules=[CMakeExtension("mlx.core")],
cmdclass={"build_ext": CMakeBuild, "generate_stubs": GenerateStubs},
zip_safe=False,