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

make EIGEN_BLAS macro names consistent and undef at end of file

Browse Source 
libeigen/eigen!2119

Closes #2954
This commit is contained in:
Blake
2026-01-29 16:50:27 +00:00
committed by Antonio Sánchez
parent 3d6f5fe8fe
commit 26c242ab58
8 changed files with 29 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Eigen/src/Core/products/GeneralMatrixMatrixTriangular_BLAS.h
View File

@@ -141,6 +141,8 @@ EIGEN_BLAS_RANKUPDATE_R(float, float, ssyrk_)
// EIGEN_BLAS_RANKUPDATE_C(dcomplex, double, double, zherk_)
// EIGEN_BLAS_RANKUPDATE_C(scomplex, float, float, cherk_)
#undef EIGEN_BLAS_RANKUPDATE_SPECIALIZE
#undef EIGEN_BLAS_RANKUPDATE_R
} // end namespace internal
} // end namespace Eigen

19
Eigen/src/Core/products/GeneralMatrixMatrix_BLAS.h
View File

@@ -49,7 +49,7 @@ namespace internal {
// gemm specialization
#define GEMM_SPECIALIZATION(EIGTYPE, EIGPREFIX, BLASTYPE, BLASFUNC) \
#define EIGEN_BLAS_GEMM_SPECIALIZATION(EIGTYPE, EIGPREFIX, BLASTYPE, BLASFUNC) \
template <typename Index, int LhsStorageOrder, bool ConjugateLhs, int RhsStorageOrder, bool ConjugateRhs> \
struct general_matrix_matrix_product<Index, EIGTYPE, LhsStorageOrder, ConjugateLhs, EIGTYPE, RhsStorageOrder, \
ConjugateRhs, ColMajor, 1> { \
@@ -105,15 +105,15 @@ namespace internal {
};
#ifdef EIGEN_USE_MKL
GEMM_SPECIALIZATION(double, d, double, dgemm)
GEMM_SPECIALIZATION(float, f, float, sgemm)
GEMM_SPECIALIZATION(dcomplex, cd, MKL_Complex16, zgemm)
GEMM_SPECIALIZATION(scomplex, cf, MKL_Complex8, cgemm)
EIGEN_BLAS_GEMM_SPECIALIZATION(double, d, double, dgemm)
EIGEN_BLAS_GEMM_SPECIALIZATION(float, f, float, sgemm)
EIGEN_BLAS_GEMM_SPECIALIZATION(dcomplex, cd, MKL_Complex16, zgemm)
EIGEN_BLAS_GEMM_SPECIALIZATION(scomplex, cf, MKL_Complex8, cgemm)
#else
GEMM_SPECIALIZATION(double, d, double, dgemm_)
GEMM_SPECIALIZATION(float, f, float, sgemm_)
GEMM_SPECIALIZATION(dcomplex, cd, double, zgemm_)
GEMM_SPECIALIZATION(scomplex, cf, float, cgemm_)
EIGEN_BLAS_GEMM_SPECIALIZATION(double, d, double, dgemm_)
EIGEN_BLAS_GEMM_SPECIALIZATION(float, f, float, sgemm_)
EIGEN_BLAS_GEMM_SPECIALIZATION(dcomplex, cd, double, zgemm_)
EIGEN_BLAS_GEMM_SPECIALIZATION(scomplex, cf, float, cgemm_)
#endif
// If OpenBLAS with BUILD_BFLOAT16=1 support is available,
@@ -198,6 +198,7 @@ struct general_matrix_matrix_product<Index, Eigen::bfloat16, LhsStorageOrder, Co
#endif // EIGEN_USE_OPENBLAS_SBGEMM
#undef EIGEN_BLAS_GEMM_SPECIALIZATION
} // namespace internal
} // end namespace Eigen

2
Eigen/src/Core/products/GeneralMatrixVector_BLAS.h
View File

@@ -132,6 +132,8 @@ EIGEN_BLAS_GEMV_SPECIALIZATION(dcomplex, double, zgemv_)
EIGEN_BLAS_GEMV_SPECIALIZATION(scomplex, float, cgemv_)
#endif
#undef EIGEN_BLAS_GEMV_SPECIALIZE
#undef EIGEN_BLAS_GEMV_SPECIALIZATION
} // namespace internal
} // end namespace Eigen

4
Eigen/src/Core/products/SelfadjointMatrixMatrix_BLAS.h
View File

@@ -270,6 +270,10 @@ EIGEN_BLAS_SYMM_R(float, float, f, ssymm_)
EIGEN_BLAS_HEMM_R(dcomplex, double, cd, zhemm_)
EIGEN_BLAS_HEMM_R(scomplex, float, cf, chemm_)
#endif
#undef EIGEN_BLAS_SYMM_L
#undef EIGEN_BLAS_SYMM_R
#undef EIGEN_BLAS_HEMM_R
} // end namespace internal
} // end namespace Eigen

2
Eigen/src/Core/products/SelfadjointMatrixVector_BLAS.h
View File

@@ -108,6 +108,8 @@ EIGEN_BLAS_SYMV_SPECIALIZATION(dcomplex, double, zhemv_)
EIGEN_BLAS_SYMV_SPECIALIZATION(scomplex, float, chemv_)
#endif
#undef EIGEN_BLAS_SYMV_SPECIALIZATION
#undef EIGEN_BLAS_SYMV_SPECIALIZE
} // end namespace internal
} // end namespace Eigen

4
Eigen/src/Core/products/TriangularMatrixMatrix_BLAS.h
View File

@@ -318,6 +318,10 @@ EIGEN_BLAS_TRMM_R(dcomplex, double, cd, ztrmm_)
EIGEN_BLAS_TRMM_R(float, float, f, strmm_)
EIGEN_BLAS_TRMM_R(scomplex, float, cf, ctrmm_)
#endif
#undef EIGEN_BLAS_TRMM_SPECIALIZE
#undef EIGEN_BLAS_TRMM_L
#undef EIGEN_BLAS_TRMM_R
} // end namespace internal
} // end namespace Eigen

3
Eigen/src/Core/products/TriangularMatrixVector_BLAS.h
View File

@@ -268,6 +268,9 @@ EIGEN_BLAS_TRMV_RM(float, float, f, s, _)
EIGEN_BLAS_TRMV_RM(scomplex, float, cf, c, _)
#endif
#undef EIGEN_BLAS_TRMV_RM
#undef EIGEN_BLAS_TRMV_SPECIALIZE
#undef EIGEN_BLAS_TRMV_CM
} // namespace internal
} // end namespace Eigen

2
Eigen/src/Core/products/TriangularSolverMatrix_BLAS.h
View File

@@ -159,6 +159,8 @@ EIGEN_BLAS_TRSM_R(float, float, strsm_)
EIGEN_BLAS_TRSM_R(scomplex, float, ctrsm_)
#endif
#undef EIGEN_BLAS_TRSM_R
#undef EIGEN_BLAS_TRSM_L
} // end namespace internal
} // end namespace Eigen