mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-07 01:22:59 +08:00
12 lines
296 B
C++
12 lines
296 B
C++
#include <Eigen/Cholesky>
|
|
#include "gemm_common.h"
|
|
|
|
EIGEN_DONT_INLINE
|
|
void llt(const Mat &A, const Mat &B, Mat &C) {
|
|
C = A;
|
|
C.diagonal().array() += 1000;
|
|
Eigen::internal::llt_inplace<Mat::Scalar, Lower>::blocked(C);
|
|
}
|
|
|
|
int main(int argc, char **argv) { return main_gemm(argc, argv, llt); }
|