Commit c42ccb33 authored by gaurides's avatar gaurides Committed by Sang Ik Lee

Update MKLDNN to v0.21 (#3641)

* Update MKLDNN to v0.21

* Changed for MKLML asset for WIN32
parent 7ea34fc2
......@@ -18,11 +18,12 @@ include(ExternalProject)
# Includes blas 3.8.0 in mkldnn
set(NGRAPH_MKLDNN_SHORT_VERSION 0)
set(NGRAPH_MKLDNN_FULL_VERSION 0.20)
set(NGRAPH_MKLDNN_ASSET_VERSION "v0.20")
set(NGRAPH_MKLDNN_VERSION "v0.20.2")
set(NGRAPH_MKLDNN_FULL_VERSION 0.21)
set(NGRAPH_MKLDNN_ASSET_VERSION "v0.21")
set(NGRAPH_MKLDNN_VERSION "v0.21")
set(NGRAPH_MKLDNN_MKLML_VERSION "2019.0.5.20190502")
set(NGRAPH_MKLDNN_GIT_TAG "v0.20.2")
set(NGRAPH_MKLDNN_MKLML_WIN32_VERSION "2020.0.20190813")
set(NGRAPH_MKLDNN_GIT_TAG "v0.21")
#------------------------------------------------------------------------------
# Fetch and install MKL-DNN
......@@ -91,6 +92,7 @@ endif()
set(MKLURLROOT "https://github.com/intel/mkl-dnn/releases/download/${NGRAPH_MKLDNN_ASSET_VERSION}/")
set(MKLVERSION ${NGRAPH_MKLDNN_MKLML_VERSION})
set(MKLWIN32VERSION ${NGRAPH_MKLDNN_MKLML_WIN32_VERSION})
if (LINUX)
set(MKLPACKAGE "mklml_lnx_${MKLVERSION}.tgz")
set(MKL_SHA1_HASH 6ab490f0b358124338d04ee9383c3cbc536969d8)
......@@ -98,8 +100,8 @@ elseif (APPLE)
set(MKLPACKAGE "mklml_mac_${MKLVERSION}.tgz")
set(MKL_SHA1_HASH a1c42af04f990b0e515a1c31946424b2e68fccc9)
elseif (WIN32)
set(MKLPACKAGE "mklml_win_${MKLVERSION}.zip")
set(MKL_SHA1_HASH 9d6ff4d5a486689338158093e96c43ee442b65f0)
set(MKLPACKAGE "mklml_win_${MKLWIN32VERSION}.zip")
set(MKL_SHA1_HASH cc117093e658d50a8e4e3d1cf192c300b6bac0fc)
endif()
set(MKL_LIBS ${MKLML_LIB} ${OMP_LIB})
set(MKLURL ${MKLURLROOT}${MKLPACKAGE})
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment