cu-matrixdim.h File Reference
#include <cstdint>
Include dependency graph for cu-matrixdim.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

struct  MatrixElement< Real >
 
struct  MatrixDim_
 Structure containing size of the matrix plus stride. More...
 
struct  CuBlockMatrixData_
 This structure is used in cu-block-matrix.h to store information about a block-diagonal matrix. More...
 
struct  Int32Pair
 

Macros

#define CU1DBLOCK   256
 
#define CU2DBLOCK   16
 

Typedefs

typedef uint32_t uint32_cuda
 
typedef int32_t int32_cuda
 
typedef int32_t MatrixIndexT_cuda
 
typedef struct MatrixDim_ MatrixDim
 Structure containing size of the matrix plus stride. More...
 
typedef struct CuBlockMatrixData_ CuBlockMatrixData
 This structure is used in cu-block-matrix.h to store information about a block-diagonal matrix. More...
 
typedef struct Int32Pair Int32Pair
 

Functions

bool operator< (const Int32Pair &a, const Int32Pair &b)
 

Macro Definition Documentation

◆ CU1DBLOCK

#define CU1DBLOCK   256

Definition at line 57 of file cu-matrixdim.h.

Referenced by CuVectorBase< float >::Add(), CuVectorBase< float >::AddColSumMat(), CuVectorBase< float >::AddDiagMatMat(), CuMatrixBase< float >::AddElements(), CuBlockMatrix< Real >::AddMatMat(), CuVectorBase< float >::AddRowSumMat(), CuMatrixBase< float >::AddSmat(), CuPackedMatrix< Real >::AddToDiag(), CuMatrixBase< float >::AddToDiag(), CuMatrixBase< float >::AddToElements(), CuVectorBase< float >::AddVecVec(), CuVectorBase< float >::ApplyExp(), CuVectorBase< float >::ApplyLog(), CuVectorBase< float >::ApplyLogSoftMax(), CuVectorBase< float >::ApplySoftMax(), kaldi::cu::BackpropLstmNonlinearity(), CuVectorBase< float >::Ceiling(), CuMatrix< float >::CompObjfAndDeriv(), kaldi::cu::ComputeLstmNonlinearity(), CuVectorBase< float >::CopyColFromMat(), CuMatrixBase< float >::CopyColsFromVec(), CuVectorBase< float >::CopyDiagFromPacked(), CuVectorBase< float >::CopyElements(), CuSpMatrix< Real >::CopyFromMat(), CuMatrixBase< float >::CopyFromMat(), CuSparseMatrix< Real >::CopyToMat(), CuMatrixBase< float >::DiffGroupPnorm(), CuMatrixBase< float >::DiffLogSoftmaxPerRow(), kaldi::cu::DiffNormalizePerRow(), CuMatrixBase< float >::DiffSoftmaxPerRow(), CuMatrixBase< float >::FindRowMaxId(), CuVectorBase< float >::Floor(), CuMatrixBase< float >::GroupMax(), CuMatrixBase< float >::GroupPnorm(), CuVectorBase< float >::InvertElements(), CuMatrixBase< float >::LogSoftMaxPerRow(), CuMatrixBase< float >::Lookup(), CuVectorBase< float >::Max(), CuMatrixBase< float >::Max(), CuVectorBase< float >::Min(), CuMatrixBase< float >::Min(), CuVectorBase< float >::MulElements(), kaldi::cu::NormalizePerRow(), CuVectorBase< float >::Pow(), CuVectorBase< float >::ReplaceValue(), CuVectorBase< float >::Scale(), CuPackedMatrix< Real >::ScaleDiag(), CuSparseMatrix< Real >::SelectRows(), CuArrayBase< Int32Pair >::Sequence(), CuVectorBase< float >::Set(), CuPackedMatrix< Real >::SetDiag(), CuMatrixBase< float >::SoftMaxPerRow(), CuVectorBase< float >::Sum(), CuMatrixBase< float >::Sum(), kaldi::TestCuVectorCopyFromVec(), CuMatrixBase< float >::Trace(), kaldi::TraceMatMat(), and kaldi::TraceMatSmat().

◆ CU2DBLOCK

Typedef Documentation

◆ CuBlockMatrixData

This structure is used in cu-block-matrix.h to store information about a block-diagonal matrix.

We declare it here so that it will be accessible

◆ int32_cuda

typedef int32_t int32_cuda

Definition at line 31 of file cu-matrixdim.h.

◆ Int32Pair

typedef struct Int32Pair Int32Pair

◆ MatrixDim

typedef struct MatrixDim_ MatrixDim

Structure containing size of the matrix plus stride.

This structure is an argument of most of the CUDA kernels.

◆ MatrixIndexT_cuda

typedef int32_t MatrixIndexT_cuda

Definition at line 32 of file cu-matrixdim.h.

◆ uint32_cuda

typedef uint32_t uint32_cuda

Definition at line 30 of file cu-matrixdim.h.

Function Documentation

◆ operator<()

bool operator< ( const Int32Pair a,
const Int32Pair b 
)
inline

Definition at line 83 of file cu-matrixdim.h.

References Int32Pair::first, and Int32Pair::second.

Referenced by NnetBatchComputer::PrintMinibatchStats().

83  {
84  if (a.first < b.first)
85  return true;
86  if (a.first > b.first)
87  return false;
88  return a.second < b.second;
89  }
int32_cuda second
Definition: cu-matrixdim.h:80
int32_cuda first
Definition: cu-matrixdim.h:79