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

5
unsupported/Eigen/CXX11/Tensor
View File

@@ -75,6 +75,10 @@ typedef unsigned __int64 uint64_t;
#include "libxsmm.h"
#endif
#if defined(EIGEN_USE_MKLDNN)
#include "mkldnn.h"
#endif
#ifdef EIGEN_USE_THREADS
#include "ThreadPool"
#endif
@@ -121,6 +125,7 @@ typedef unsigned __int64 uint64_t;
#include "src/Tensor/TensorArgMax.h"
#include "src/Tensor/TensorConcatenation.h"
#include "src/Tensor/TensorContractionMapper.h"
#include "src/Tensor/TensorContractionMkldnn.h"
#include "src/Tensor/TensorContractionBlocking.h"
#include "src/Tensor/TensorContraction.h"
#include "src/Tensor/TensorContractionThreadPool.h"