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

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