diff options
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" |