diff options
author | Eugene Zhulenev <ezhulenev@google.com> | 2018-09-26 11:08:47 -0700 |
---|---|---|
committer | Eugene Zhulenev <ezhulenev@google.com> | 2018-09-26 11:08:47 -0700 |
commit | 71cd3fbd6a03991977fd3faf82109bf27b701d2d (patch) | |
tree | bb017d89f9901d9d6d550eb20f9c5af3d0dd2aae /unsupported/test/CMakeLists.txt | |
parent | 0a3356f4ece30cd486b616eb1da9128aa4f245be (diff) |
Support multiple contraction kernel types in TensorContractionThreadPool
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) |