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

fix uv install and add dev release

Browse Source
This commit is contained in:
Awni Hannun
2025-07-23 07:48:12 -07:00
parent d107d8d495
commit e2b35e06a4
2 changed files with 92 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
setup.py
View File

@@ -269,11 +269,9 @@ if __name__ == "__main__":
# - Package name is back-end specific, e.g mlx-metal
if build_stage != 2:
if build_stage == 1:
if build_macos:
install_requires += [f"mlx-metal=={version}"]
else:
extras["cuda"] = [f"mlx-cuda=={version}"]
extras["cpu"] = [f"mlx-cpu=={version}"]
install_requires += [f'mlx-metal=={version}; platform_system == "Darwin"']
extras["cuda"] = [f'mlx-cuda=={version}; platform_system == "Linux"']
extras["cpu"] = [f'mlx-cpu=={version}; platform_system == "Linux"']
_setup(
name="mlx",