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 Packages Projects Releases Wiki Activity

temp for testing

Browse Source
This commit is contained in:
Awni Hannun
2025-07-10 19:36:36 -07:00
parent 9381163788
commit c623cc7683
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
setup.py
View File

@@ -218,9 +218,10 @@ if __name__ == "__main__":
]
}
test = "-awni-test"
if not is_release or build_macos:
_setup(
name="mlx",
name="mlx" + test,
include_package_data=True,
packages=packages,
extras_require=extras,
@@ -229,18 +230,18 @@ if __name__ == "__main__":
cmdclass={"build_ext": CMakeBuild, "generate_stubs": GenerateStubs},
)
elif build_common:
extras["cpu"] = [f"mlx-cpu=={version}"]
extras["cuda"] = [f"mlx-cuda=={version}"]
extras["cpu"] = [f"mlx-cpu{test}=={version}"]
extras["cuda"] = [f"mlx-cuda{test}=={version}"]
_setup(
name="mlx",
include_package_data=False,
name="mlx" + test,
packages=["mlx"],
extras_require=extras,
entry_points=entry_points,
exclude_package_data=package_data,
)
else:
_setup(
name="mlx-cuda" if build_cuda else "mlx-cpu",
name="mlx-cuda" if build_cuda else "mlx-cpu" + test,
include_package_data=True,
packages=packages,
extras_require=extras,