Update build_wheel.sh

This commit is contained in:
Nikita Shulga 2025-01-28 15:14:41 -08:00 committed by GitHub
parent eea7d395e5
commit 049f042e52
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -190,9 +190,8 @@ retry pip install -r "${pytorch_rootdir}/requirements.txt" || true
retry pip uninstall -y cmake retry pip uninstall -y cmake
retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -y llvm-openmp=14.0.6 cmake retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -y llvm-openmp=14.0.6 cmake
# For USE_DISTRIBUTED=1 on macOS, need libuv and pkg-config to find libuv. # For USE_DISTRIBUTED=1 on macOS, need libuv, which is build as part of tensorpipe submodule
export USE_DISTRIBUTED=1 export USE_DISTRIBUTED=1
retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -y libuv pkg-config
if [[ -n "$CROSS_COMPILE_ARM64" ]]; then if [[ -n "$CROSS_COMPILE_ARM64" ]]; then
export CMAKE_OSX_ARCHITECTURES=arm64 export CMAKE_OSX_ARCHITECTURES=arm64