matrix_utils.hpp File Reference

#include "data.hpp"
#include "part_descriptor.hpp"
#include <raft/core/handle.hpp>
#include <raft/random/rng.cuh>

Include dependency graph for matrix_utils.hpp:

Go to the source code of this file.

Namespaces
	MLCommon
	MLCommon::Matrix
	MLCommon::Matrix::opg

Functions
void	MLCommon::Matrix::opg::gatherPart (const raft::handle_t &h, float *gatheredPart, std::vector< Matrix::Data< float > * > &parts, Matrix::PartDescriptor &desc, int partIndex, int rootRank, int myRank, cudaStream_t stream)
void	MLCommon::Matrix::opg::allGatherPart (const raft::handle_t &h, float *gatheredPart, std::vector< Matrix::Data< float > * > &parts, Matrix::PartDescriptor &desc, int partIndex, int myRank, cudaStream_t stream)
void	MLCommon::Matrix::opg::gather (const raft::handle_t &h, float *gatheredMatrix, std::vector< Matrix::Data< float > * > &parts, Matrix::PartDescriptor &desc, int rootRank, int myRank, cudaStream_t stream)
void	MLCommon::Matrix::opg::allGather (const raft::handle_t &h, float *gatheredMatrix, std::vector< Matrix::Data< float > * > &parts, Matrix::PartDescriptor &desc, int myRank, cudaStream_t stream)
void	MLCommon::Matrix::opg::allocate (const raft::handle_t &h, std::vector< Matrix::Data< float > * > &parts, Matrix::PartDescriptor &desc, int myRank, cudaStream_t stream)
void	MLCommon::Matrix::opg::deallocate (const raft::handle_t &h, std::vector< Matrix::Data< float > * > &parts, Matrix::PartDescriptor &desc, int myRank, cudaStream_t stream)
void	MLCommon::Matrix::opg::randomize (const raft::handle_t &h, raft::random::Rng &r, std::vector< Matrix::Data< float > * > &parts, Matrix::PartDescriptor &desc, int myRank, cudaStream_t stream, float low=-1.0f, float high=1.0f)
void	MLCommon::Matrix::opg::reset (const raft::handle_t &h, std::vector< Matrix::Data< float > * > &parts, Matrix::PartDescriptor &desc, int myRank, cudaStream_t stream)
void	MLCommon::Matrix::opg::printRaw2D (float *buffer, int rows, int cols, bool isColMajor, cudaStream_t stream)
void	MLCommon::Matrix::opg::print (const raft::handle_t &h, std::vector< Matrix::Data< float > * > &parts, Matrix::PartDescriptor &desc, const char *matrixName, int myRank, cudaStream_t stream)
void	MLCommon::Matrix::opg::gatherPart (const raft::handle_t &h, double *gatheredPart, std::vector< Matrix::Data< double > * > &parts, Matrix::PartDescriptor &desc, int partIndex, int rootRank, int myRank, cudaStream_t stream)
void	MLCommon::Matrix::opg::allGatherPart (const raft::handle_t &h, double *gatheredPart, std::vector< Matrix::Data< double > * > &parts, Matrix::PartDescriptor &desc, int partIndex, int myRank, cudaStream_t stream)
void	MLCommon::Matrix::opg::gather (const raft::handle_t &h, double *gatheredMatrix, std::vector< Matrix::Data< double > * > &parts, Matrix::PartDescriptor &desc, int rootRank, int myRank, cudaStream_t stream)
void	MLCommon::Matrix::opg::allGather (const raft::handle_t &h, double *gatheredMatrix, std::vector< Matrix::Data< double > * > &parts, Matrix::PartDescriptor &desc, int myRank, cudaStream_t stream)
void	MLCommon::Matrix::opg::allocate (const raft::handle_t &h, std::vector< Matrix::Data< double > * > &parts, Matrix::PartDescriptor &desc, int myRank, cudaStream_t stream)
void	MLCommon::Matrix::opg::deallocate (const raft::handle_t &h, std::vector< Matrix::Data< double > * > &parts, Matrix::PartDescriptor &desc, int myRank, cudaStream_t stream)
void	MLCommon::Matrix::opg::randomize (const raft::handle_t &h, raft::random::Rng &r, std::vector< Matrix::Data< double > * > &parts, Matrix::PartDescriptor &desc, int myRank, cudaStream_t stream, double low=-1.0, double high=1.0)
void	MLCommon::Matrix::opg::reset (const raft::handle_t &h, std::vector< Matrix::Data< double > * > &parts, Matrix::PartDescriptor &desc, int myRank, cudaStream_t stream)
void	MLCommon::Matrix::opg::printRaw2D (double *buffer, int rows, int cols, bool isColMajor, cudaStream_t stream)
void	MLCommon::Matrix::opg::print (const raft::handle_t &h, std::vector< Matrix::Data< double > * > &parts, Matrix::PartDescriptor &desc, const char *matrixName, int myRank, cudaStream_t stream)