install linux with mlx[cuda] and mlx[cpu]

This commit is contained in:
Awni Hannun
2025-07-10 17:18:14 -07:00
parent 8347575ba1
commit 9381163788
3 changed files with 76 additions and 31 deletions

View File

@@ -336,10 +336,11 @@ jobs:
pip install typing_extensions
python setup.py generate_stubs
<< parameters.extra_env >> python -m build --wheel
<< parameters.extra_env >> MLX_BUILD_COMMON=1 python -m build --wheel
auditwheel show dist/*
auditwheel repair dist/* --plat manylinux_2_31_x86_64
- run:
name: Upload package
name: Upload packages
command: |
source env/bin/activate
twine upload wheelhouse/*