cu-matrix.h File Reference

#include <sstream>
#include <vector>
#include "cudamatrix/cu-matrixdim.h"
#include "cudamatrix/cu-common.h"
#include "cudamatrix/cu-value.h"
#include "matrix/matrix-common.h"
#include "matrix/kaldi-matrix.h"
#include "cudamatrix/cu-array.h"
#include "cudamatrix/cu-math.h"
#include "cudamatrix/cu-rand.h"
#include "cudamatrix/cu-sparse-matrix.h"
#include "cudamatrix/cu-matrix-inl.h"

Include dependency graph for cu-matrix.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	CuMatrixBase< Real >
	Matrix for CUDA computing. More...
class	CuMatrix< Real >
	This class represents a matrix that's stored on the GPU if we have one, and in memory if not. More...
class	CuSubMatrix< Real >
	This class is used for a piece of a CuMatrix. More...

Namespaces
	kaldi
	This code computes Goodness of Pronunciation (GOP) and extracts phone-level pronunciation feature for mispronunciations detection tasks, the reference:

Functions
template<typename Real >
Real	TraceMatMat (const CuMatrixBase< Real > &A, const CuMatrixBase< Real > &B, MatrixTransposeType trans)
template<typename Real >
void	AddMatMatBatched (const Real alpha, std::vector< CuSubMatrix< Real > *> &C, const std::vector< CuSubMatrix< Real > *> &A, MatrixTransposeType transA, const std::vector< CuSubMatrix< Real > *> &B, MatrixTransposeType transB, const Real beta)
	Does multiple matrix multiplications, executing them in parallel using cuBLAS's gemmBatched if we are using a GPU. More...
template<typename Real >
bool	ApproxEqual (const CuMatrixBase< Real > &A, const CuMatrixBase< Real > &B, Real tol=0.01)
template<typename Real >
void	AssertEqual (const CuMatrixBase< Real > &A, const CuMatrixBase< Real > &B, float tol=0.01)
template<typename Real >
bool	SameDim (const CuMatrixBase< Real > &M, const CuMatrixBase< Real > &N)
template<typename Real >
bool	SameDimAndStride (const CuMatrixBase< Real > &M, const CuMatrixBase< Real > &N)
template<typename Real >
std::ostream &	operator<< (std::ostream &out, const CuMatrixBase< Real > &mat)
	Print the matrix to stream. More...