mirror of
https://gitlab.com/libeigen/eigen.git
synced 2025-07-20 11:54:27 +08:00
fix #74: sparse triangular solver for lower/row-major matrices
This commit is contained in:
parent
72fc81dd9d
commit
b581cb870c
@ -43,8 +43,11 @@ struct ei_solve_triangular_selector<Lhs,Rhs,LowerTriangular,RowMajor|IsSparse>
|
|||||||
{
|
{
|
||||||
lastVal = it.value();
|
lastVal = it.value();
|
||||||
lastIndex = it.index();
|
lastIndex = it.index();
|
||||||
|
if(lastIndex == i)
|
||||||
|
break;
|
||||||
tmp -= lastVal * other.coeff(lastIndex,col);
|
tmp -= lastVal * other.coeff(lastIndex,col);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (Lhs::Flags & UnitDiagBit)
|
if (Lhs::Flags & UnitDiagBit)
|
||||||
other.coeffRef(i,col) = tmp;
|
other.coeffRef(i,col) = tmp;
|
||||||
else
|
else
|
||||||
|
@ -68,12 +68,20 @@ template<typename Scalar> void sparse_solvers(int rows, int cols)
|
|||||||
VERIFY_IS_APPROX(refMat2.template marked<LowerTriangular>().solveTriangular(vec2),
|
VERIFY_IS_APPROX(refMat2.template marked<LowerTriangular>().solveTriangular(vec2),
|
||||||
m2.template marked<LowerTriangular>().solveTriangular(vec3));
|
m2.template marked<LowerTriangular>().solveTriangular(vec3));
|
||||||
|
|
||||||
|
// lower - transpose
|
||||||
|
initSparse<Scalar>(density, refMat2, m2, ForceNonZeroDiag|MakeLowerTriangular, &zeroCoords, &nonzeroCoords);
|
||||||
|
VERIFY_IS_APPROX(refMat2.template marked<LowerTriangular>().transpose().solveTriangular(vec2),
|
||||||
|
m2.template marked<LowerTriangular>().transpose().solveTriangular(vec3));
|
||||||
|
|
||||||
// upper
|
// upper
|
||||||
initSparse<Scalar>(density, refMat2, m2, ForceNonZeroDiag|MakeUpperTriangular, &zeroCoords, &nonzeroCoords);
|
initSparse<Scalar>(density, refMat2, m2, ForceNonZeroDiag|MakeUpperTriangular, &zeroCoords, &nonzeroCoords);
|
||||||
VERIFY_IS_APPROX(refMat2.template marked<UpperTriangular>().solveTriangular(vec2),
|
VERIFY_IS_APPROX(refMat2.template marked<UpperTriangular>().solveTriangular(vec2),
|
||||||
m2.template marked<UpperTriangular>().solveTriangular(vec3));
|
m2.template marked<UpperTriangular>().solveTriangular(vec3));
|
||||||
|
|
||||||
// TODO test row major
|
// upper - transpose
|
||||||
|
initSparse<Scalar>(density, refMat2, m2, ForceNonZeroDiag|MakeUpperTriangular, &zeroCoords, &nonzeroCoords);
|
||||||
|
VERIFY_IS_APPROX(refMat2.template marked<UpperTriangular>().transpose().solveTriangular(vec2),
|
||||||
|
m2.template marked<UpperTriangular>().transpose().solveTriangular(vec3));
|
||||||
}
|
}
|
||||||
|
|
||||||
// test LLT
|
// test LLT
|
||||||
|
Loading…
x
Reference in New Issue
Block a user