diff options
author | Gael Guennebaud <g.gael@free.fr> | 2017-06-08 15:06:27 +0200 |
---|---|---|
committer | Gael Guennebaud <g.gael@free.fr> | 2017-06-08 15:06:27 +0200 |
commit | 682b2ef17ee360ac11cebbe7286dc4edd9accfa3 (patch) | |
tree | 32711f0ce5bf53ecb87d0578fcd3ea61fa4b6c36 /Eigen | |
parent | 4bbc32046810f65bb0f77f6dbe538abad51de281 (diff) |
bug #1423: fix LSCG\'s Jacobi preconditioner for row-major matrices.
Diffstat (limited to 'Eigen')
-rw-r--r-- | Eigen/src/IterativeLinearSolvers/BasicPreconditioners.h | 27 |
1 files changed, 21 insertions, 6 deletions
diff --git a/Eigen/src/IterativeLinearSolvers/BasicPreconditioners.h b/Eigen/src/IterativeLinearSolvers/BasicPreconditioners.h index 358444aff..279c9173c 100644 --- a/Eigen/src/IterativeLinearSolvers/BasicPreconditioners.h +++ b/Eigen/src/IterativeLinearSolvers/BasicPreconditioners.h @@ -152,13 +152,28 @@ class LeastSquareDiagonalPreconditioner : public DiagonalPreconditioner<_Scalar> { // Compute the inverse squared-norm of each column of mat m_invdiag.resize(mat.cols()); - for(Index j=0; j<mat.outerSize(); ++j) + if(MatType::IsRowMajor) { - RealScalar sum = mat.innerVector(j).squaredNorm(); - if(sum>0) - m_invdiag(j) = RealScalar(1)/sum; - else - m_invdiag(j) = RealScalar(1); + m_invdiag.setZero(); + for(Index j=0; j<mat.outerSize(); ++j) + { + for(typename MatType::InnerIterator it(mat,j); it; ++it) + m_invdiag(it.index()) += it.value(); + } + for(Index j=0; j<mat.cols(); ++j) + if(m_invdiag(j)>0) + m_invdiag(j) = RealScalar(1)/m_invdiag(j); + } + else + { + for(Index j=0; j<mat.outerSize(); ++j) + { + RealScalar sum = mat.innerVector(j).squaredNorm(); + if(sum>0) + m_invdiag(j) = RealScalar(1)/sum; + else + m_invdiag(j) = RealScalar(1); + } } Base::m_isInitialized = true; return *this; |