Set rpath with cmake for CUDA build (#2932)

2025-12-19 12:53:38 +09:00
parent ab4dce4e18
commit b2e2b19bf7
4 changed files with 29 additions and 33 deletions
@@ -21,4 +21,11 @@ runs:
        pip install auditwheel build patchelf setuptools
        python setup.py clean --all
        MLX_BUILD_STAGE=2 python -m build -w
-        bash python/scripts/repair_cuda.sh ${{ inputs.arch }}
+
+        auditwheel repair dist/* \
+          --plat manylinux_2_35_${{ inputs.arch }} \
+          --exclude libcublas* \
+          --exclude libcuda* \
+          --exclude libcudnn* \
+          --exclude libnccl* \
+          --exclude libnvrtc*