enable mkl in eigen for qr_op (#11116)

17e82cc2 · Ziming Dong · Rasmus Munk Larsen · ed7d08ea · 17e82cc2
隐藏空白更改
内联并排

Showing with 4 addition and 0 deletion

tensorflow/core/kernels/qr_op_impl.h tensorflow/core/kernels/qr_op_impl.h +4 -0

未找到文件。
--- 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"