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...