Skip to content

Commit ef93e89

Browse files
authored
Update MKL to 2022.1 (#1245)
As previous one occasionally crashes on AMD CPUs May be addresses pytorch/pytorch#89817 Please note, that in order to get maximum perf on AMD CPUs one needs to compile and LD_PRELOAD following library: ``` int mkl_serv_intel_cpu_true() { return 1; } ```
1 parent 3a407b2 commit ef93e89

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

common/install_mkl.sh

+2-2
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@
33
set -ex
44

55
# MKL
6-
MKL_VERSION=2020.0
7-
MKL_BUILD=166
6+
MKL_VERSION=2022.2.1
7+
MKL_BUILD=16993
88
mkdir -p /opt/intel/lib
99
pushd /tmp
1010
curl -fsSL https://anaconda.org/intel/mkl-static/${MKL_VERSION}/download/linux-64/mkl-static-${MKL_VERSION}-intel_${MKL_BUILD}.tar.bz2 | tar xjv

wheel/build_wheel.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,7 @@ conda create ${EXTRA_CONDA_INSTALL_FLAGS} -yn "$tmp_env_name" python="$desired_p
168168
source activate "$tmp_env_name"
169169

170170
retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -yq cmake "numpy${NUMPY_PINNED_VERSION}" nomkl "setuptools${SETUPTOOLS_PINNED_VERSION}" "pyyaml${PYYAML_PINNED_VERSION}" cffi typing_extensions ninja requests
171-
retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -yq mkl-include==2020.1 mkl-static==2020.1 -c intel
171+
retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -yq mkl-include==2022.2.1 mkl-static==2022.2.1 -c intel
172172
retry pip install -qr "${pytorch_rootdir}/requirements.txt" || true
173173

174174
# For USE_DISTRIBUTED=1 on macOS, need libuv and pkg-config to find libuv.

0 commit comments

Comments
 (0)