diff options
Diffstat (limited to 'unsupported/test/CMakeLists.txt')
| -rw-r--r-- | unsupported/test/CMakeLists.txt | 12 |
1 files changed, 12 insertions, 0 deletions
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) |