diff options
author | 2016-12-15 09:34:54 -0800 | |
---|---|---|
committer | 2016-12-15 09:34:54 -0800 | |
commit | 95195bd0ef742f2876a1694ebb4c018f858af062 (patch) | |
tree | 97f861630898e1aa38dd6f08f351640688b3c17f /third_party/eigen3/unsupported/Eigen/CXX11/FixedPoint | |
parent | dbe5e17e2ed307e86e1a6e79e558ec3e335d46fc (diff) |
Added support for AVX512 to fixed point instructions. (#6323)
Diffstat (limited to 'third_party/eigen3/unsupported/Eigen/CXX11/FixedPoint')
-rw-r--r-- | third_party/eigen3/unsupported/Eigen/CXX11/FixedPoint | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/third_party/eigen3/unsupported/Eigen/CXX11/FixedPoint b/third_party/eigen3/unsupported/Eigen/CXX11/FixedPoint index 9d6b9c3f01..8e55a1f3e8 100644 --- a/third_party/eigen3/unsupported/Eigen/CXX11/FixedPoint +++ b/third_party/eigen3/unsupported/Eigen/CXX11/FixedPoint @@ -31,8 +31,15 @@ #include "src/FixedPoint/FixedPointTypes.h" // Use optimized implementations whenever available -#ifdef EIGEN_VECTORIZE_AVX2 +#ifdef EIGEN_VECTORIZE_AVX512 +#include "src/Tensor/TensorContractionThreadPool.h" +#include "src/FixedPoint/PacketMathAVX512.h" +#include "src/FixedPoint/TypeCastingAVX512.h" + +#elif defined EIGEN_VECTORIZE_AVX2 #define EIGEN_USE_OPTIMIZED_INT8_UINT8_MAT_MAT_PRODUCT +#define EIGEN_USE_OPTIMIZED_INT16_INT16_MAT_MAT_PRODUCT +#include "src/Tensor/TensorContractionThreadPool.h" #include "src/FixedPoint/PacketMathAVX2.h" #include "src/FixedPoint/MatMatProductAVX2.h" #include "src/FixedPoint/TypeCastingAVX2.h" |