From 17e82cc274ca05f1aa65d94fbcfa359d469e1b21 Mon Sep 17 00:00:00 2001
From: Ziming Dong <dzm1016397507@gmail.com>
Date: Fri, 11 Aug 2017 01:43:06 +0800
Subject: [PATCH] enable mkl in eigen for qr_op (#11116)

---
 tensorflow/core/kernels/qr_op_impl.h | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tensorflow/core/kernels/qr_op_impl.h b/tensorflow/core/kernels/qr_op_impl.h
index 029ef834808..ab664fa6d33 100644
--- a/tensorflow/core/kernels/qr_op_impl.h
+++ b/tensorflow/core/kernels/qr_op_impl.h
@@ -20,6 +20,10 @@ limitations under the License.
 // improve compilation times.
 #include <algorithm>
 
+#ifdef INTEL_MKL
+#define EIGEN_USE_MKL_ALL
+#endif // INTEL_MKL
+
 #include "third_party/eigen3/Eigen/QR"
 #include "tensorflow/core/framework/kernel_def_builder.h"
 #include "tensorflow/core/framework/op_kernel.h"
-- 
GitLab