Skip to content

N
ngraph
Commit c42ccb33 authored by gaurides's avatar gaurides Committed by Sang Ik Lee
Browse files
Options

Update MKLDNN to v0.21 (#3641) 

* Update MKLDNN to v0.21

* Changed for MKLML asset for WIN32
parent 7ea34fc2
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 6 deletions
cmake/external_mkldnn.cmake
View file @ c42ccb33
...@@ -18,11 +18,12 @@ include(ExternalProject) ...@@ -18,11 +18,12 @@ include(ExternalProject)
# Includes blas 3.8.0 in mkldnn # Includes blas 3.8.0 in mkldnn
set(NGRAPH_MKLDNN_SHORT_VERSION 0) set(NGRAPH_MKLDNN_SHORT_VERSION 0)
set(NGRAPH_MKLDNN_FULL_VERSION 0.20) set(NGRAPH_MKLDNN_FULL_VERSION 0.21)
set(NGRAPH_MKLDNN_ASSET_VERSION "v0.20") set(NGRAPH_MKLDNN_ASSET_VERSION "v0.21")
set(NGRAPH_MKLDNN_VERSION "v0.20.2") set(NGRAPH_MKLDNN_VERSION "v0.21")
set(NGRAPH_MKLDNN_MKLML_VERSION "2019.0.5.20190502") 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 # Fetch and install MKL-DNN
...@@ -91,6 +92,7 @@ endif() ...@@ -91,6 +92,7 @@ endif()
set(MKLURLROOT "https://github.com/intel/mkl-dnn/releases/download/${NGRAPH_MKLDNN_ASSET_VERSION}/") set(MKLURLROOT "https://github.com/intel/mkl-dnn/releases/download/${NGRAPH_MKLDNN_ASSET_VERSION}/")
set(MKLVERSION ${NGRAPH_MKLDNN_MKLML_VERSION}) set(MKLVERSION ${NGRAPH_MKLDNN_MKLML_VERSION})
set(MKLWIN32VERSION ${NGRAPH_MKLDNN_MKLML_WIN32_VERSION})
if (LINUX) if (LINUX)
set(MKLPACKAGE "mklml_lnx_${MKLVERSION}.tgz") set(MKLPACKAGE "mklml_lnx_${MKLVERSION}.tgz")
set(MKL_SHA1_HASH 6ab490f0b358124338d04ee9383c3cbc536969d8) set(MKL_SHA1_HASH 6ab490f0b358124338d04ee9383c3cbc536969d8)
...@@ -98,8 +100,8 @@ elseif (APPLE) ...@@ -98,8 +100,8 @@ elseif (APPLE)
set(MKLPACKAGE "mklml_mac_${MKLVERSION}.tgz") set(MKLPACKAGE "mklml_mac_${MKLVERSION}.tgz")
set(MKL_SHA1_HASH a1c42af04f990b0e515a1c31946424b2e68fccc9) set(MKL_SHA1_HASH a1c42af04f990b0e515a1c31946424b2e68fccc9)
elseif (WIN32) elseif (WIN32)
set(MKLPACKAGE "mklml_win_${MKLVERSION}.zip") set(MKLPACKAGE "mklml_win_${MKLWIN32VERSION}.zip")
set(MKL_SHA1_HASH 9d6ff4d5a486689338158093e96c43ee442b65f0) set(MKL_SHA1_HASH cc117093e658d50a8e4e3d1cf192c300b6bac0fc)
endif() endif()
set(MKL_LIBS ${MKLML_LIB} ${OMP_LIB}) set(MKL_LIBS ${MKLML_LIB} ${OMP_LIB})
set(MKLURL ${MKLURLROOT}${MKLPACKAGE}) 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