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

Properly set the number of registers for AVX512

Browse Source
This commit is contained in:
Gael Guennebaud
2018-12-11 15:33:17 +01:00
parent 7166496f70
commit 0a7e7af6fd
3 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
Eigen/src/Core/arch/AVX/PacketMath.h
View File

@@ -18,8 +18,8 @@ namespace internal {
#define EIGEN_CACHEFRIENDLY_PRODUCT_THRESHOLD 8 #define EIGEN_CACHEFRIENDLY_PRODUCT_THRESHOLD 8
#endif #endif
#ifndef EIGEN_ARCH_DEFAULT_NUMBER_OF_REGISTERS #if !defined(EIGEN_VECTORIZE_AVX512) && !defined(EIGEN_ARCH_DEFAULT_NUMBER_OF_REGISTERS)
#define EIGEN_ARCH_DEFAULT_NUMBER_OF_REGISTERS (2*sizeof(void*)) #define EIGEN_ARCH_DEFAULT_NUMBER_OF_REGISTERS 16
#endif #endif
#ifdef EIGEN_VECTORIZE_FMA #ifdef EIGEN_VECTORIZE_FMA

2
Eigen/src/Core/arch/AVX512/PacketMath.h
View File

@@ -19,7 +19,7 @@ namespace internal {
#endif #endif
#ifndef EIGEN_ARCH_DEFAULT_NUMBER_OF_REGISTERS #ifndef EIGEN_ARCH_DEFAULT_NUMBER_OF_REGISTERS
#define EIGEN_ARCH_DEFAULT_NUMBER_OF_REGISTERS (2*sizeof(void*)) #define EIGEN_ARCH_DEFAULT_NUMBER_OF_REGISTERS 32
#endif #endif
#ifdef EIGEN_VECTORIZE_FMA #ifdef EIGEN_VECTORIZE_FMA

4
Eigen/src/Core/arch/SSE/PacketMath.h
View File

@@ -18,7 +18,9 @@ namespace internal {
#define EIGEN_CACHEFRIENDLY_PRODUCT_THRESHOLD 8 #define EIGEN_CACHEFRIENDLY_PRODUCT_THRESHOLD 8
#endif #endif
#ifndef EIGEN_ARCH_DEFAULT_NUMBER_OF_REGISTERS #if !defined(EIGEN_VECTORIZE_AVX) && !defined(EIGEN_ARCH_DEFAULT_NUMBER_OF_REGISTERS)
// 32 bits => 8 registers
// 64 bits => 16 registers
#define EIGEN_ARCH_DEFAULT_NUMBER_OF_REGISTERS (2*sizeof(void*)) #define EIGEN_ARCH_DEFAULT_NUMBER_OF_REGISTERS (2*sizeof(void*))
#endif #endif