GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-05-02 00:34:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

AVX512 TRSM kernels use alloca if EIGEN_NO_MALLOC requested

Browse Source
This commit is contained in:
b-shi 2022-06-17 18:05:26 +00:00 committed by Rasmus Munk Larsen
parent 4d1c16eab8
commit 37673ca1bc
4 changed files with 2087 additions and 2030 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

File diff suppressed because it is too large Load Diff

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

File diff suppressed because it is too large Load Diff

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

File diff suppressed because it is too large Load Diff

6
Eigen/src/Core/products/TriangularSolverMatrix.h
View File

@ -171,7 +171,7 @@ EIGEN_DONT_INLINE void triangular_solve_matrix<Scalar,Index,OnTheLeft,Mode,Conju
std::ptrdiff_t l1, l2, l3;
manage_caching_sizes(GetAction, &l1, &l2, &l3);
#if defined(EIGEN_ENABLE_AVX512_NOCOPY_TRSML_CUTOFFS)
#if (EIGEN_ENABLE_AVX512_NOCOPY_TRSM_L_CUTOFFS)
EIGEN_IF_CONSTEXPR( (OtherInnerStride == 1 &&
(std::is_same<Scalar,float>::value ||
std::is_same<Scalar,double>::value)) ) {
@ -246,7 +246,7 @@ EIGEN_DONT_INLINE void triangular_solve_matrix<Scalar,Index,OnTheLeft,Mode,Conju
// tr solve
{
Index i = IsLower ? k2+k1 : k2-k1;
#if defined(EIGEN_USE_AVX512_TRSM_L_KERNELS)
#if EIGEN_USE_AVX512_TRSM_L_KERNELS
EIGEN_IF_CONSTEXPR( (OtherInnerStride == 1 &&
(std::is_same<Scalar,float>::value ||
std::is_same<Scalar,double>::value)) ) {
@ -318,7 +318,7 @@ EIGEN_DONT_INLINE void triangular_solve_matrix<Scalar,Index,OnTheRight,Mode,Conj
{
Index rows = otherSize;
#if defined(EIGEN_ENABLE_AVX512_NOCOPY_TRSM_R_CUTOFFS)
#if EIGEN_ENABLE_AVX512_NOCOPY_TRSM_R_CUTOFFS
EIGEN_IF_CONSTEXPR( (OtherInnerStride == 1 &&
(std::is_same<Scalar,float>::value ||
std::is_same<Scalar,double>::value)) ) {