Fix tridiagonalization_inplace_selector::run() when called from CUDA

This commit is contained in:
ahmed 2024-04-19 21:06:59 +00:00 committed by Rasmus Munk Larsen
parent 1550c99541
commit ee9d57347b

View File

@ -445,8 +445,8 @@ struct tridiagonalization_inplace_selector<MatrixType, 3, false> {
typedef typename MatrixType::RealScalar RealScalar;
template <typename DiagonalType, typename SubDiagonalType, typename CoeffVectorType, typename WorkSpaceType>
static void run(MatrixType& mat, DiagonalType& diag, SubDiagonalType& subdiag, CoeffVectorType&, WorkSpaceType&,
bool extractQ) {
static EIGEN_DEVICE_FUNC void run(MatrixType& mat, DiagonalType& diag, SubDiagonalType& subdiag, CoeffVectorType&,
WorkSpaceType&, bool extractQ) {
using std::sqrt;
const RealScalar tol = (std::numeric_limits<RealScalar>::min)();
diag[0] = mat(0, 0);