File tree 2 files changed +6
-6
lines changed 2 files changed +6
-6
lines changed Original file line number Diff line number Diff line change @@ -142,7 +142,7 @@ if [[ "$cuda_ver" != 'cpu' ]]; then
142
142
fi
143
143
fi
144
144
145
- # Check that OpenBlas is not linked to on Macs
145
+ # Check that OpenBlas is not linked to on MacOS
146
146
if [[ " $( uname) " == ' Darwin' ]]; then
147
147
echo " Checking the OpenBLAS is not linked to"
148
148
all_dylibs=($( find " $( python -c " import site; print(site.getsitepackages()[0])" ) " /torch -name ' *.dylib' ) )
@@ -153,6 +153,9 @@ if [[ "$(uname)" == 'Darwin' ]]; then
153
153
exit 1
154
154
fi
155
155
done
156
+
157
+ echo " Checking that OpenMP is available"
158
+ python -c " import torch; exit(0 if torch.backends.openmp.is_available() else 1)"
156
159
fi
157
160
158
161
popd
Original file line number Diff line number Diff line change @@ -182,11 +182,8 @@ tmp_env_name="wheel_py$python_nodot"
182
182
conda create ${EXTRA_CONDA_INSTALL_FLAGS} -yn " $tmp_env_name " python=" $desired_python "
183
183
source activate " $tmp_env_name "
184
184
185
- if [[ " $desired_python " == " 3.11" ]]; then
186
- retry pip install -q " numpy${NUMPY_PINNED_VERSION} " " setuptools${SETUPTOOLS_PINNED_VERSION} " " pyyaml${PYYAML_PINNED_VERSION} " typing_extensions requests
187
- else
188
- retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -yq " numpy${NUMPY_PINNED_VERSION} " nomkl " setuptools${SETUPTOOLS_PINNED_VERSION} " " pyyaml${PYYAML_PINNED_VERSION} " typing_extensions requests
189
- fi
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
+
190
187
if [[ " $( uname -m) " == " arm64" ]]; then
191
188
retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -yq cmake ninja
192
189
else
You can’t perform that action at this time.
0 commit comments