eig.hpp File Reference

#include "../matrix/data.hpp"
#include "../matrix/part_descriptor.hpp"
#include <raft/core/comms.hpp>
#include <raft/core/handle.hpp>

Include dependency graph for eig.hpp:

Go to the source code of this file.

Namespaces
	MLCommon
	MLCommon::LinAlg
	MLCommon::LinAlg::opg

Functions
void	MLCommon::LinAlg::opg::eigDC (const raft::handle_t &h, float *eigenValues, float *eigenVectors, std::vector< Matrix::Data< float > * > &inParts, Matrix::PartDescriptor &desc, int myRank, cudaStream_t stream)
	Multi-GPU version of Eigen decomposition. This function works for symmetric matrices only. Whole input matrix is gathered at rank 0 and Eigen decomposition is carried out sequentially. More...
void	MLCommon::LinAlg::opg::eigDC (const raft::handle_t &h, double *eigenValues, double *eigenVectors, std::vector< Matrix::Data< double > * > &inParts, Matrix::PartDescriptor &desc, int myRank, cudaStream_t stream)
void	MLCommon::LinAlg::opg::eigJacobi (const raft::handle_t &h, float *eigenValues, float *eigenVectors, std::vector< Matrix::Data< float > * > &inParts, Matrix::PartDescriptor &desc, int myRank, cudaStream_t stream)
void	MLCommon::LinAlg::opg::eigJacobi (const raft::handle_t &h, double *eigenValues, double *eigenVectors, std::vector< Matrix::Data< double > * > &inParts, Matrix::PartDescriptor &desc, int myRank, cudaStream_t stream)