devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Support multiple contraction kernel types in TensorContractionThreadPool

Browse Source
This commit is contained in:
Eugene Zhulenev
2018-09-26 11:08:47 -07:00
parent 0a3356f4ec
commit 71cd3fbd6a
6 changed files with 484 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
unsupported/test/CMakeLists.txt
View File

@@ -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)