From ba318852a1f76f6fba47cc3ed937235a4f6265a6 Mon Sep 17 00:00:00 2001
From: Nikita Shulga <nikita.shulga@gmail.com>
Date: Sat, 17 Dec 2022 18:27:18 -0800
Subject: [PATCH] Update MKL to 2022.1

As previous one occasionally crashes on AMD CPUs
---
 common/install_mkl.sh | 4 ++--
 wheel/build_wheel.sh  | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/common/install_mkl.sh b/common/install_mkl.sh
index 5ebdd94b1..5889dc1f0 100644
--- a/common/install_mkl.sh
+++ b/common/install_mkl.sh
@@ -3,8 +3,8 @@
 set -ex
 
 # MKL
-MKL_VERSION=2020.0
-MKL_BUILD=166
+MKL_VERSION=2022.2.1
+MKL_BUILD=16993
 mkdir -p /opt/intel/lib
 pushd /tmp
 curl -fsSL https://anaconda.org/intel/mkl-static/${MKL_VERSION}/download/linux-64/mkl-static-${MKL_VERSION}-intel_${MKL_BUILD}.tar.bz2 | tar xjv
diff --git a/wheel/build_wheel.sh b/wheel/build_wheel.sh
index 08b47335a..b8b663ac0 100755
--- a/wheel/build_wheel.sh
+++ b/wheel/build_wheel.sh
@@ -168,7 +168,7 @@ conda create ${EXTRA_CONDA_INSTALL_FLAGS} -yn "$tmp_env_name" python="$desired_p
 source activate "$tmp_env_name"
 
 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
-retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -yq mkl-include==2020.1 mkl-static==2020.1 -c intel
+retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -yq mkl-include==2022.2.1 mkl-static==2022.2.1 -c intel
 retry pip install -qr "${pytorch_rootdir}/requirements.txt" || true
 
 # For USE_DISTRIBUTED=1 on macOS, need libuv and pkg-config to find libuv.