97
97
whl_tmp_dir=" ${MAC_PACKAGE_WORK_DIR} /dist"
98
98
mkdir -p " $whl_tmp_dir "
99
99
100
- if [[ -n " $CROSS_COMPILE_ARM64 " || $( uname -m) == " arm64" ]]; then
101
- mac_version=' macosx_11_0_arm64'
102
- libtorch_arch=' arm64'
103
- else
104
- mac_version=' macosx_10_9_x86_64'
105
- libtorch_arch=' x86_64'
106
- fi
100
+ mac_version=' macosx_11_0_arm64'
101
+ libtorch_arch=' arm64'
107
102
108
103
# Create a consistent wheel package name to rename the wheel to
109
104
wheel_filename_new=" ${TORCH_PACKAGE_NAME} -${build_version}${build_number_prefix} -cp${python_nodot} -none-${mac_version} .whl"
@@ -163,13 +158,9 @@ case $desired_python in
163
158
;;
164
159
3.8)
165
160
echo " Using 3.8 deps"
166
- if [[ " $( uname -m) " == " arm64" ]]; then
167
- SETUPTOOLS_PINNED_VERSION=" >=46.0.0"
168
- PYYAML_PINNED_VERSION=" >=5.3"
169
- NUMPY_PINNED_VERSION=" =1.19"
170
- else
171
- NUMPY_PINNED_VERSION=" =1.17"
172
- fi
161
+ SETUPTOOLS_PINNED_VERSION=" >=46.0.0"
162
+ PYYAML_PINNED_VERSION=" >=5.3"
163
+ NUMPY_PINNED_VERSION=" =1.19"
173
164
;;
174
165
* )
175
166
echo " Using default deps"
@@ -182,30 +173,19 @@ tmp_env_name="wheel_py$python_nodot"
182
173
conda create ${EXTRA_CONDA_INSTALL_FLAGS} -yn " $tmp_env_name " python=" $desired_python "
183
174
source activate " $tmp_env_name "
184
175
185
- retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -yq " numpy${NUMPY_PINNED_VERSION} " nomkl " setuptools${SETUPTOOLS_PINNED_VERSION} " " pyyaml${PYYAML_PINNED_VERSION} " typing_extensions requests
186
-
187
- if [[ " $( uname -m) " == " arm64" ]]; then
188
- retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -yq cmake ninja
189
- else
190
- retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -yq cmake ninja mkl-include==2022.2.1 mkl-static==2022.2.1 -c intel
191
- fi
176
+ retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -yq " numpy${NUMPY_PINNED_VERSION} " nomkl cmake ninja " setuptools${SETUPTOOLS_PINNED_VERSION} " " pyyaml${PYYAML_PINNED_VERSION} " typing_extensions requests
192
177
retry pip install -qr " ${pytorch_rootdir} /requirements.txt" || true
193
178
194
179
# For USE_DISTRIBUTED=1 on macOS, need libuv and pkg-config to find libuv.
195
180
export USE_DISTRIBUTED=1
196
181
retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -yq libuv pkg-config
197
182
198
- if [[ -n " $CROSS_COMPILE_ARM64 " || " $( uname -m) " == " arm64" ]]; then
199
- if [[ -n " $CROSS_COMPILE_ARM64 " ]]; then
200
- export CMAKE_OSX_ARCHITECTURES=arm64
201
- fi
202
- export USE_MKLDNN=OFF
203
- export USE_QNNPACK=OFF
204
- export BUILD_TEST=OFF
205
- elif [[ " $( uname -m) " == " x86_64" ]]; then
206
- retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -yq llvmdev=9
207
- export USE_LLVM=" ${CONDA_PREFIX} "
183
+ if [[ -n " $CROSS_COMPILE_ARM64 " ]]; then
184
+ export CMAKE_OSX_ARCHITECTURES=arm64
208
185
fi
186
+ export USE_MKLDNN=OFF
187
+ export USE_QNNPACK=OFF
188
+ export BUILD_TEST=OFF
209
189
210
190
pushd " $pytorch_rootdir "
211
191
echo " Calling setup.py bdist_wheel at $( date) "
289
269
zip -rq " $PYTORCH_FINAL_PACKAGE_DIR /libtorch-macos-${libtorch_arch} -$PYTORCH_BUILD_VERSION .zip" libtorch
290
270
cp " $PYTORCH_FINAL_PACKAGE_DIR /libtorch-macos-${libtorch_arch} -$PYTORCH_BUILD_VERSION .zip" \
291
271
" $PYTORCH_FINAL_PACKAGE_DIR /libtorch-macos-${libtorch_arch} -latest.zip"
292
- if [[ " ${libtorch_arch} " == " x86_64" ]]; then
293
- # For backward compatibility make unarched latest to point to x86_64
294
- cp " $PYTORCH_FINAL_PACKAGE_DIR /libtorch-macos-${libtorch_arch} -$PYTORCH_BUILD_VERSION .zip" \
295
- " $PYTORCH_FINAL_PACKAGE_DIR /libtorch-macos-latest.zip"
296
- fi
297
272
fi
0 commit comments