Upgrade MKL DNN

This commit is contained in:
AG Ramesh 2019-08-22 12:44:51 -07:00 committed by Penporn Koanantakool
parent 69b1feac62
commit 6c6c81fa37
2 changed files with 5 additions and 5 deletions

View File

@ -135,11 +135,11 @@ def tf_repositories(path_prefix = "", tf_repo_name = ""):
tf_http_archive(
name = "mkl_dnn",
build_file = clean_dep("//third_party/mkl_dnn:mkldnn.BUILD"),
sha256 = "26f720ed912843ba293e8a1e0822fe5318e93c529d80c87af1cf555d68e642d0",
strip_prefix = "mkl-dnn-0.20.1",
sha256 = "a198a9bd3c584607e6a467f780beca92c8411cd656fcc8ec6fa5abe73d4af823",
strip_prefix = "mkl-dnn-0.20.3",
urls = [
"https://storage.googleapis.com/mirror.tensorflow.org/github.com/intel/mkl-dnn/archive/v0.20.1.tar.gz",
"https://github.com/intel/mkl-dnn/archive/v0.20.1.tar.gz",
"https://storage.googleapis.com/mirror.tensorflow.org/github.com/intel/mkl-dnn/archive/v0.20.3.tar.gz",
"https://github.com/intel/mkl-dnn/archive/v0.20.3.tar.gz",
],
)

View File

@ -45,7 +45,7 @@ template_rule(
substitutions = {
"@MKLDNN_VERSION_MAJOR@": "0",
"@MKLDNN_VERSION_MINOR@": "20",
"@MKLDNN_VERSION_PATCH@": "0",
"@MKLDNN_VERSION_PATCH@": "3",
"@MKLDNN_VERSION_HASH@": "N/A",
},
)