experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Upgrading DNNL(previously known as MKL DNN) to v1.4

Browse Source
This commit is contained in:
ag.ramesh 2020-04-18 11:21:05 -07:00
parent a76f10521a
commit a59d7762ae
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/workspace.bzl
View File

@ -203,11 +203,11 @@ def tf_repositories(path_prefix = "", tf_repo_name = ""):
tf_http_archive( tf_http_archive(
name = "mkl_dnn_v1", name = "mkl_dnn_v1",
build_file = clean_dep("//third_party/mkl_dnn:mkldnn_v1.BUILD"), build_file = clean_dep("//third_party/mkl_dnn:mkldnn_v1.BUILD"),
sha256 = "a71ec1f27c30b8a176605e8a78444f1f12301a3c313b70ff93290926c140509c", sha256 = "54737bcb4dc1961d32ee75da3ecc529fa48198f8b2ca863a079e19a9c4adb70f",
strip_prefix = "mkl-dnn-1.2.2", strip_prefix = "oneDNN-1.4",
urls = [ urls = [
"https://storage.googleapis.com/mirror.tensorflow.org/github.com/intel/mkl-dnn/archive/v1.2.2.tar.gz", "https://storage.googleapis.com/mirror.tensorflow.org/github.com/oneapi-src/oneDNN/archive/v1.4.tar.gz",
"https://github.com/intel/mkl-dnn/archive/v1.2.2.tar.gz", "https://github.com/oneapi-src/oneDNN/archive/v1.4.tar.gz",
], ],
) )

2
third_party/mkl_dnn/mkldnn_v1.BUILD vendored
View File

@ -43,7 +43,7 @@ template_rule(
out = "include/dnnl_version.h", out = "include/dnnl_version.h",
substitutions = { substitutions = {
"@DNNL_VERSION_MAJOR@": "1", "@DNNL_VERSION_MAJOR@": "1",
"@DNNL_VERSION_MINOR@": "2", "@DNNL_VERSION_MINOR@": "4",
"@DNNL_VERSION_PATCH@": "0", "@DNNL_VERSION_PATCH@": "0",
"@DNNL_VERSION_HASH@": "N/A", "@DNNL_VERSION_HASH@": "N/A",
}, },