From a7a5e3991e3e0841f4866cca6f533600d229cffc Mon Sep 17 00:00:00 2001 From: shwetaoj Date: Wed, 2 Dec 2020 12:35:49 -0800 Subject: [PATCH 1/2] Added cmake option DNNL_ENABLE_MAX_CPU_ISA --- third_party/mkl_dnn/mkldnn_v1.BUILD | 1 + 1 file changed, 1 insertion(+) diff --git a/third_party/mkl_dnn/mkldnn_v1.BUILD b/third_party/mkl_dnn/mkldnn_v1.BUILD index 96d48506ba1..7ecea9645dd 100644 --- a/third_party/mkl_dnn/mkldnn_v1.BUILD +++ b/third_party/mkl_dnn/mkldnn_v1.BUILD @@ -83,6 +83,7 @@ cc_library( }) + [ "-UUSE_MKL", "-UUSE_CBLAS", + "-DNNL_ENABLE_MAX_CPU_ISA", ] + tf_openmp_copts(), includes = [ "include", From 62a33f10d097cbdf69d3af8a9e33c62e5f324a0d Mon Sep 17 00:00:00 2001 From: shwetaoj Date: Thu, 3 Dec 2020 09:21:27 -0800 Subject: [PATCH 2/2] fix --- third_party/mkl_dnn/mkldnn_v1.BUILD | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/third_party/mkl_dnn/mkldnn_v1.BUILD b/third_party/mkl_dnn/mkldnn_v1.BUILD index 7ecea9645dd..561f417afb3 100644 --- a/third_party/mkl_dnn/mkldnn_v1.BUILD +++ b/third_party/mkl_dnn/mkldnn_v1.BUILD @@ -83,7 +83,7 @@ cc_library( }) + [ "-UUSE_MKL", "-UUSE_CBLAS", - "-DNNL_ENABLE_MAX_CPU_ISA", + "-DDNNL_ENABLE_MAX_CPU_ISA", ] + tf_openmp_copts(), includes = [ "include",