MatrixUtils::EigenSolver< ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > > > Class Template Reference

Collaboration diagram for MatrixUtils::EigenSolver< ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > > >:

Public Types
using	Scalar = _Scalar
using	RealScalar = typename Eigen::NumTraits< _Scalar >::Real

Public Member Functions
	EigenSolver ()=delete
	EigenSolver (EigenSolver const &)=delete
	EigenSolver (EigenSolver &&)=delete
	EigenSolver (int dim)
	EigenSolver (MatrixOnGPU_t const &mat, Eigen::DecompositionOptions option=Eigen::ComputeEigenvectors)=delete
	EigenSolver (MatrixOnGPU_t &&mat, Eigen::DecompositionOptions option=Eigen::ComputeEigenvectors)
EigenSolver &	compute ()
EigenSolver &	setOpt (magma_vec_t option)
Eigen::VectorX< RealScalar > const &	eigenvalues ()
MatrixOnGPU_t const &	eigenvectors ()
MatrixOnGPU_t &	inMatrix ()

Private Types
using	MatrixOnGPU_t = ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > >

Private Attributes
Eigen::VectorX< RealScalar >	m_hEigVals
MatrixOnGPU_t	m_dEigVecs
bool	m_isDiagonalized = false
magma_int_t	m_info = -1
magma_vec_t	m_option = MagmaVec
magma_uplo_t	m_uplo = MagmaLower

Member Typedef Documentation

MatrixOnGPU_t

template<typename _Scalar , int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>

using MatrixUtils::EigenSolver< ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > > >::MatrixOnGPU_t = ObjectOnGPU< Eigen::Matrix<_Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols> >

private

RealScalar

template<typename _Scalar , int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>

using MatrixUtils::EigenSolver< ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > > >::RealScalar = typename Eigen::NumTraits<_Scalar>::Real

Scalar

template<typename _Scalar , int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>

using MatrixUtils::EigenSolver< ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > > >::Scalar = _Scalar

Constructor & Destructor Documentation

EigenSolver() [1/6]

template<typename _Scalar , int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>

MatrixUtils::EigenSolver< ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > > >::EigenSolver ( )

delete

EigenSolver() [2/6]

template<typename _Scalar , int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>

MatrixUtils::EigenSolver< ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > > >::EigenSolver ( EigenSolver< ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > > > const &  )

delete

EigenSolver() [3/6]

template<typename _Scalar , int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>

MatrixUtils::EigenSolver< ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > > >::EigenSolver ( EigenSolver< ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > > > &&  )

delete

EigenSolver() [4/6]

template<typename _Scalar , int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>

MatrixUtils::EigenSolver< ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > > >::EigenSolver ( int  dim )

inline

: m_hEigVals(dim), m_dEigVecs(dim, dim) {}

EigenSolver() [5/6]

template<typename _Scalar , int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>

MatrixUtils::EigenSolver< ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > > >::EigenSolver ( MatrixOnGPU_t const &  mat, Eigen::DecompositionOptions  option = Eigen::ComputeEigenvectors  )

delete

EigenSolver() [6/6]

template<typename _Scalar , int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>

MatrixUtils::EigenSolver< ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > > >::EigenSolver ( MatrixOnGPU_t &&  mat, Eigen::DecompositionOptions  option = Eigen::ComputeEigenvectors  )

inline

                : m_dEigVecs{std::move(mat)}, m_hEigVals(mat.rows()) {
                if(option == Eigen::EigenvaluesOnly) m_option = MagmaNoVec;
                this->compute();
            }

Member Function Documentation

compute()

template<typename _Scalar , int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>

EigenSolver & MatrixUtils::EigenSolver< ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > > >::compute ( )

inline

                                   {
                m_hEigVals.resize(m_dEigVecs.rows());
                debug_print("EigenSolver(GPU)" << __func__
                                               << ": m_dEigVecs.rows()=" << m_dEigVecs.rows()
                                               << ", m_dEigVecs.cols()=" << m_dEigVecs.cols()
                                               << ", m_dEigVecs.LD()=" << m_dEigVecs.LD());
                Eigen::MatrixX<Scalar>   wA(m_dEigVecs.LD(), m_dEigVecs.rows());
                std::vector<Scalar>      work(1);
                std::vector<RealScalar>  rwork(1);
                std::vector<magma_int_t> iwork(1);
 
                // Query about the sizes of workspaces.
                m_info = magma_heevd_gpu_wrapper(
                    m_option, m_uplo, m_dEigVecs.rows(), m_dEigVecs.data(), m_dEigVecs.LD(),
                    m_hEigVals.data(), wA.data(), wA.rows(), work.data(), -1, rwork.data(), -1,
                    iwork.data(), -1, &m_info);
                work.resize(static_cast<magma_int_t>(real(work[0])));
                rwork.resize(static_cast<magma_int_t>(rwork[0]));
                iwork.resize(iwork[0]);
 
                // Calculate eigenvalues and eigenvectors.
                m_info = magma_heevd_gpu_wrapper(
                    m_option, m_uplo, m_dEigVecs.rows(), m_dEigVecs.data(), m_dEigVecs.LD(),
                    m_hEigVals.data(), wA.data(), wA.rows(), work.data(), work.size(), rwork.data(),
                    rwork.size(), iwork.data(), iwork.size(), &m_info);
                m_isDiagonalized = true;
                return *this;
            }

eigenvalues()

template<typename _Scalar , int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>

Eigen::VectorX< RealScalar > const & MatrixUtils::EigenSolver< ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > > >::eigenvalues ( )

inline

{ return m_hEigVals; }

eigenvectors()

template<typename _Scalar , int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>

MatrixOnGPU_t const & MatrixUtils::EigenSolver< ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > > >::eigenvectors ( )

inline

{ return m_dEigVecs; }

inMatrix()

template<typename _Scalar , int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>

MatrixOnGPU_t & MatrixUtils::EigenSolver< ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > > >::inMatrix ( )

inline

{ return m_dEigVecs; }

setOpt()

template<typename _Scalar , int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>

EigenSolver & MatrixUtils::EigenSolver< ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > > >::setOpt ( magma_vec_t  option )

inline

                                                    {
                m_option = option;
                return *this;
            }

Member Data Documentation

m_dEigVecs

template<typename _Scalar , int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>

MatrixOnGPU_t MatrixUtils::EigenSolver< ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > > >::m_dEigVecs

private

m_hEigVals

template<typename _Scalar , int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>

Eigen::VectorX<RealScalar> MatrixUtils::EigenSolver< ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > > >::m_hEigVals

private

m_info

template<typename _Scalar , int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>

magma_int_t MatrixUtils::EigenSolver< ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > > >::m_info = -1

private

m_isDiagonalized

template<typename _Scalar , int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>

bool MatrixUtils::EigenSolver< ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > > >::m_isDiagonalized = false

private

m_option

template<typename _Scalar , int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>

magma_vec_t MatrixUtils::EigenSolver< ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > > >::m_option = MagmaVec

private

m_uplo

template<typename _Scalar , int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>

magma_uplo_t MatrixUtils::EigenSolver< ObjectOnGPU< Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > > >::m_uplo = MagmaLower

private

---

The documentation for this class was generated from the following file:

• /Users/shoki/GitHub/Locality/Headers/MatrixUtils/MatrixUtils.cuh