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

cleanup circle, fix cuda repair

Browse Source
This commit is contained in:
Awni Hannun
2025-07-11 08:17:58 -07:00
parent c623cc7683
commit 8f93ca9e52
2 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
.circleci/config.yml
View File

@@ -336,9 +336,10 @@ jobs:
pip install typing_extensions pip install typing_extensions
python setup.py generate_stubs python setup.py generate_stubs
<< parameters.extra_env >> python -m build --wheel << parameters.extra_env >> python -m build --wheel
<< parameters.extra_env >> MLX_BUILD_COMMON=1 python -m build --wheel
auditwheel show dist/* auditwheel show dist/*
auditwheel repair dist/* --plat manylinux_2_31_x86_64 auditwheel repair dist/* --plat manylinux_2_31_x86_64
<< parameters.extra_env >> MLX_BUILD_COMMON=1 \
python -m build --wheel --outdir wheelhouse
- run: - run:
name: Upload packages name: Upload packages
command: | command: |

10
python/scripts/repair_cuda.sh
View File

@@ -3,15 +3,19 @@
auditwheel repair dist/* \ auditwheel repair dist/* \
--plat manylinux_2_35_x86_64 \ --plat manylinux_2_35_x86_64 \
--exclude libcublas* \ --exclude libcublas* \
--exclude libnvrtc* --exclude libnvrtc* \
-w wheel_tmp
cd wheelhouse
mkdir wheelhouse
cd wheel_tmp
repaired_wheel=$(find . -name "*.whl" -print -quit) repaired_wheel=$(find . -name "*.whl" -print -quit)
unzip -q "${repaired_wheel}" unzip -q "${repaired_wheel}"
rm "${repaired_wheel}"
core_so=$(find mlx -name "core*.so" -print -quit) core_so=$(find mlx -name "core*.so" -print -quit)
rpath=$(patchelf --print-rpath "${core_so}") rpath=$(patchelf --print-rpath "${core_so}")
rpath=$rpath:\$ORIGIN/../nvidia/cublas/lib:\$ORIGIN/../nvidia/cuda_nvrtc/lib rpath=$rpath:\$ORIGIN/../nvidia/cublas/lib:\$ORIGIN/../nvidia/cuda_nvrtc/lib
patchelf --force-rpath --set-rpath "$rpath" "$core_so" patchelf --force-rpath --set-rpath "$rpath" "$core_so"
# Re-zip the repaired wheel # Re-zip the repaired wheel
zip -r -q "${repaired_wheel}" . zip -r -q "../wheelhouse/${repaired_wheel}" .