Support multiple contraction kernel types in TensorContractionThreadPool

2026-04-10 11:34:33 +08:00 · 2018-09-26 11:08:47 -07:00
parent 0a3356f4ec
commit 71cd3fbd6a
6 changed files with 484 additions and 20 deletions
--- a/cmake/FindMkldnn.cmake
+++ 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)