From a59d7762ae752ee5944e294fb693c799d81d1039 Mon Sep 17 00:00:00 2001 From: "ag.ramesh" Date: Sat, 18 Apr 2020 11:21:05 -0700 Subject: [PATCH] Upgrading DNNL(previously known as MKL DNN) to v1.4 --- tensorflow/workspace.bzl | 8 ++++---- third_party/mkl_dnn/mkldnn_v1.BUILD | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/tensorflow/workspace.bzl b/tensorflow/workspace.bzl index 0977fb23660..010dca07db5 100755 --- a/tensorflow/workspace.bzl +++ b/tensorflow/workspace.bzl @@ -203,11 +203,11 @@ def tf_repositories(path_prefix = "", tf_repo_name = ""): tf_http_archive( name = "mkl_dnn_v1", build_file = clean_dep("//third_party/mkl_dnn:mkldnn_v1.BUILD"), - sha256 = "a71ec1f27c30b8a176605e8a78444f1f12301a3c313b70ff93290926c140509c", - strip_prefix = "mkl-dnn-1.2.2", + sha256 = "54737bcb4dc1961d32ee75da3ecc529fa48198f8b2ca863a079e19a9c4adb70f", + strip_prefix = "oneDNN-1.4", urls = [ - "https://storage.googleapis.com/mirror.tensorflow.org/github.com/intel/mkl-dnn/archive/v1.2.2.tar.gz", - "https://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/oneapi-src/oneDNN/archive/v1.4.tar.gz", ], ) diff --git a/third_party/mkl_dnn/mkldnn_v1.BUILD b/third_party/mkl_dnn/mkldnn_v1.BUILD index b916738cb20..243ec00a60f 100644 --- a/third_party/mkl_dnn/mkldnn_v1.BUILD +++ b/third_party/mkl_dnn/mkldnn_v1.BUILD @@ -43,7 +43,7 @@ template_rule( out = "include/dnnl_version.h", substitutions = { "@DNNL_VERSION_MAJOR@": "1", - "@DNNL_VERSION_MINOR@": "2", + "@DNNL_VERSION_MINOR@": "4", "@DNNL_VERSION_PATCH@": "0", "@DNNL_VERSION_HASH@": "N/A", },