From 71cd3fbd6a03991977fd3faf82109bf27b701d2d Mon Sep 17 00:00:00 2001
From: Eugene Zhulenev <ezhulenev@google.com>
Date: Wed, 26 Sep 2018 11:08:47 -0700
Subject: Support multiple contraction kernel types in
 TensorContractionThreadPool

---
 cmake/FindMkldnn.cmake | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)
 create mode 100644 cmake/FindMkldnn.cmake

(limited to 'cmake')

diff --git a/cmake/FindMkldnn.cmake b/cmake/FindMkldnn.cmake
new file mode 100644
index 000000000..6acba9d8c
--- /dev/null
+++ b/cmake/FindMkldnn.cmake
@@ -0,0 +1,18 @@
+# Intel mkl-dnn support.
+# Link: https://github.com/intel/mkl-dnn
+if (MKLDNN)
+    set(MKLDNN_FIND_QUIETLY TRUE)
+    set(MKLDNN_INCLUDES ${MKLDNN}/include)
+    set(MKLDNN_LIBRARIES ${MKLDNN}/lib)
+endif (MKLDNN)
+find_path(MKLDNN
+        NAMES
+        mkldnn.h
+        PATHS
+        $ENV{MKLDNNDIR}/include
+        ${INCLUDE_INSTALL_DIR}
+        )
+include(FindPackageHandleStandardArgs)
+find_package_handle_standard_args(MKLDNN DEFAULT_MSG
+        MKLDNN)
+mark_as_advanced(MKLDNN)
\ No newline at end of file
-- 
cgit v1.2.3