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/Eigen/CXX11/Tensor | |
parent | 0a3356f4ece30cd486b616eb1da9128aa4f245be (diff) |
Support multiple contraction kernel types in TensorContractionThreadPool
Diffstat (limited to 'unsupported/Eigen/CXX11/Tensor')
-rw-r--r-- | unsupported/Eigen/CXX11/Tensor | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/unsupported/Eigen/CXX11/Tensor b/unsupported/Eigen/CXX11/Tensor index 14a2d0f0d..420afe650 100644 --- a/unsupported/Eigen/CXX11/Tensor +++ b/unsupported/Eigen/CXX11/Tensor @@ -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" |