mirror of
https://github.com/ml-explore/mlx.git
synced 2025-07-21 16:51:15 +08:00
20 lines
499 B
Bash
20 lines
499 B
Bash
|
#!/bin/bash
|
||
|
|
|
||
|
|
auditwheel repair dist/* \
|
||
|
|
--plat manylinux_2_35_x86_64 \
|
||
|
|
--exclude libmlx* \
|
||
|
|
-w wheel_tmp
|
||
|
|
|
||
|
|
mkdir wheelhouse
|
||
|
|
cd wheel_tmp
|
||
|
|
repaired_wheel=$(find . -name "*.whl" -print -quit)
|
||
|
|
unzip -q "${repaired_wheel}"
|
||
|
|
rm "${repaired_wheel}"
|
||
|
|
core_so=$(find mlx -name "core*.so" -print -quit)
|
||
|
|
rpath="\$ORIGIN/lib"
|
||
|
|
patchelf --force-rpath --set-rpath "$rpath" "$core_so"
|
||
|
|
python ../python/scripts/repair_record.py ${core_so}
|
||
|
|
|
||
|
|
# Re-zip the repaired wheel
|
||
|
|
zip -r -q "../wheelhouse/${repaired_wheel}" .
|