diff --git a/tensorflow/workspace.bzl b/tensorflow/workspace.bzl index ec62fde2965..a8f4806c130 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", },