GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-04-30 15:54:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

Disable AVX512 GEMM kernels by default.

Browse Source
This commit is contained in:
Antonio Sánchez 2022-07-20 21:22:48 +00:00
parent a678a3e052
commit 2cf4d18c9c
3 changed files with 13 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
Eigen/src/Core/arch/AVX512/GemmKernel.h
View File

@ -7,8 +7,8 @@
// Public License v. 2.0. If a copy of the MPL was not distributed // Public License v. 2.0. If a copy of the MPL was not distributed
// with this file, You can obtain one at http://mozilla.org/MPL/2.0/. // with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
#ifndef GEMM_KERNEL_H #ifndef EIGEN_CORE_ARCH_AVX512_GEMM_KERNEL_H
#define GEMM_KERNEL_H #define EIGEN_CORE_ARCH_AVX512_GEMM_KERNEL_H
#if EIGEN_COMP_MSVC #if EIGEN_COMP_MSVC
#include <intrin.h> #include <intrin.h>
@ -21,7 +21,8 @@
#include "../../InternalHeaderCheck.h" #include "../../InternalHeaderCheck.h"
#if !defined(EIGEN_USE_AVX512_GEMM_KERNELS) #if !defined(EIGEN_USE_AVX512_GEMM_KERNELS)
#define EIGEN_USE_AVX512_GEMM_KERNELS 1 // Disable new AVX512 kernels by default.
#define EIGEN_USE_AVX512_GEMM_KERNELS 0
#endif #endif
#define SECOND_FETCH (32) #define SECOND_FETCH (32)
@ -1229,4 +1230,6 @@ EIGEN_ALWAYS_INLINE void gebp_kernel<Scalar, Scalar, Index, DataMapper, mr, 8, C
} // namespace internal } // namespace internal
} // namespace Eigen } // namespace Eigen
#endif // GEMM_KERNEL_H #undef SECOND_FETCH
#endif // EIGEN_CORE_ARCH_AVX512_GEMM_KERNEL_H

6
Eigen/src/Core/arch/AVX512/TrsmKernel.h
View File

@ -7,8 +7,8 @@
// Public License v. 2.0. If a copy of the MPL was not distributed // Public License v. 2.0. If a copy of the MPL was not distributed
// with this file, You can obtain one at http://mozilla.org/MPL/2.0/. // with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
#ifndef EIGEN_TRSM_KERNEL_IMPL_H #ifndef EIGEN_CORE_ARCH_AVX512_TRSM_KERNEL_H
#define EIGEN_TRSM_KERNEL_IMPL_H #define EIGEN_CORE_ARCH_AVX512_TRSM_KERNEL_H
#include "../../InternalHeaderCheck.h" #include "../../InternalHeaderCheck.h"
@ -1188,4 +1188,4 @@ EIGEN_DONT_INLINE void trsmKernelL<double, Index, Mode, false, TriStorageOrder,
#endif // EIGEN_USE_AVX512_TRSM_KERNELS #endif // EIGEN_USE_AVX512_TRSM_KERNELS
} // namespace internal } // namespace internal
} // namespace Eigen } // namespace Eigen
#endif // EIGEN_TRSM_KERNEL_IMPL_H #endif // EIGEN_CORE_ARCH_AVX512_TRSM_KERNEL_H

6
Eigen/src/Core/arch/AVX512/TrsmUnrolls.inc
View File

@ -7,8 +7,8 @@
// Public License v. 2.0. If a copy of the MPL was not distributed // Public License v. 2.0. If a copy of the MPL was not distributed
// with this file, You can obtain one at http://mozilla.org/MPL/2.0/. // with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
#ifndef EIGEN_UNROLLS_IMPL_H #ifndef EIGEN_CORE_ARCH_AVX512_TRSM_UNROLLS_H
#define EIGEN_UNROLLS_IMPL_H #define EIGEN_CORE_ARCH_AVX512_TRSM_UNROLLS_H
template <bool isARowMajor = true> template <bool isARowMajor = true>
static EIGEN_ALWAYS_INLINE int64_t idA(int64_t i, int64_t j, int64_t LDA) { static EIGEN_ALWAYS_INLINE int64_t idA(int64_t i, int64_t j, int64_t LDA) {
@ -1209,4 +1209,4 @@ class gemm {
}; };
} // namespace unrolls } // namespace unrolls
#endif // EIGEN_UNROLLS_IMPL_H #endif // EIGEN_CORE_ARCH_AVX512_TRSM_UNROLLS_H