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

---
 unsupported/test/CMakeLists.txt | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'unsupported/test/CMakeLists.txt')

diff --git a/unsupported/test/CMakeLists.txt b/unsupported/test/CMakeLists.txt
index 875842272..25ae1a8d2 100644
--- a/unsupported/test/CMakeLists.txt
+++ b/unsupported/test/CMakeLists.txt
@@ -23,6 +23,17 @@ else(XSMM_FOUND)
   ei_add_property(EIGEN_MISSING_BACKENDS  "Xsmm, ")
 endif(XSMM_FOUND)
 
+find_package(Mkldnn)
+if(MKLDNN_FOUND)
+  add_definitions("-DEIGEN_USE_MKLDNN")
+  include_directories(${MKLDNN_INCLUDES})
+  link_directories(${MKLDNN_LIBRARIES})
+  set(EXTERNAL_LIBS ${EXTERNAL_LIBS} mkldnn)
+  ei_add_property(EIGEN_TESTED_BACKENDS  "Mkldd, ")
+else(MKLDNN_FOUND)
+  ei_add_property(EIGEN_MISSING_BACKENDS  "Mkldnn, ")
+endif(MKLDNN_FOUND)
+
 find_package(GoogleHash)
 if(GOOGLEHASH_FOUND)
   add_definitions("-DEIGEN_GOOGLEHASH_SUPPORT")
@@ -190,6 +201,7 @@ if(EIGEN_TEST_CXX11)
   ei_add_test(cxx11_tensor_index_list)
   ei_add_test(cxx11_tensor_mixed_indices)
   ei_add_test(cxx11_tensor_contraction)
+  ei_add_test(cxx11_tensor_contraction_mkldnn)
   ei_add_test(cxx11_tensor_convolution)
   ei_add_test(cxx11_tensor_expr)
   ei_add_test(cxx11_tensor_fixed_size)
-- 
cgit v1.2.3