cu-matrix-test.cc File Reference

#include <iostream>
#include <vector>
#include <cstdlib>
#include "base/kaldi-common.h"
#include "util/common-utils.h"
#include "cudamatrix/cu-matrix-lib.h"

Include dependency graph for cu-matrix-test.cc:

Go to the source code of this file.

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 >
static void	InitRand (VectorBase< Real > *v)
template<typename Real >
static void	InitRand (MatrixBase< Real > *M)
template<typename Real >
static void	RandZeroToOneMatrix (MatrixBase< Real > *mat)
template<typename Real >
static void	UnitTestCuMatrixTraceMatMat ()
template<typename Real >
static void	UnitTestCuCholesky ()
template<typename Real >
static void	UnitTestCuMatrixApplyLog ()
template<typename Real >
static void	UnitTestCuMatrixApplyExpSpecial ()
template<typename Real >
static void	UnitTestCuMatrixApplyExp ()
template<typename Real >
static void	UnitTestCuMatrixApplyExpLimited ()
template<typename Real >
static void	UnitTestCuMatrixSigmoid ()
template<typename Real >
static void	UnitTestCuMatrixScale ()
template<typename Real >
static void	UnitTestCuMatrixAdd ()
template<typename Real >
static void	UnitTestCuMatrixSoftHinge ()
template<typename Real >
static void	UnitTestCuMatrixGroupPnorm ()
template<typename Real >
static void	UnitTestCuMatrixGroupMax ()
template<typename Real >
static void	UnitTestCuMatrixSet ()
template<typename Real >
static void	UnitTestCuMatrixApplyPow ()
template<typename Real >
static void	UnitTestCuMatrixApplyPowAbs ()
template<typename Real >
static void	UnitTestCuMatrixCopyRowsFromVec ()
template<typename Real >
static void	UnitTestCuMatrixCopyColsFromVec ()
template<typename Real >
static void	UnitTestCuMatrixCopyRows ()
template<typename Real >
static void	UnitTestCuMatrixCopyToRows ()
template<typename Real >
static void	UnitTestCuMatrixAddRows ()
template<typename Real >
static void	UnitTestCuMatrixMulRows ()
template<typename Real >
static void	UnitTestCuMatrixAddToRows ()
template<typename Real >
void	UnitTestCuMatrixCopyCross ()
template<typename Real >
void	UnitTestCuMatrixCopyCross2 ()
template<typename Real >
static void	UnitTestCuMatrixSumColumnRanges ()
template<typename Real >
static void	UnitTestCuMatrixAddRowRanges ()
template<typename Real >
static void	UnitTestCuMatrixCopyCols ()
template<typename Real >
static void	UnitTextCuMatrixAddSmat ()
template<typename Real >
static void	UnitTextCuMatrixAddMatSmat ()
template<typename Real >
static void	UnitTextCuMatrixAddSmatMat ()
template<typename Real >
static void	UnitTestCuMatrixAddCols ()
template<typename Real >
static void	UnitTestCuMatrixApplyFloor ()
template<typename Real >
static void	UnitTestCuMatrixApplyCeiling ()
template<typename Real >
static void	UnitTestCuMatrixApplyHeaviside ()
template<typename Real >
static void	UnitTestCuMatrixHeaviside ()
template<typename Real >
static void	UnitTestCuMatrixMulElements ()
template<typename Real >
static void	UnitTestCuMatrixDivElements ()
template<typename Real >
static void	UnitTestCuMatrixMax ()
template<typename Real >
static void	UnitTestCuMatrixMin ()
template<typename Real >
static void	UnitTestCuMatrixMulColsVec ()
template<typename Real >
static void	UnitTestCuMatrixMulRowsVec ()
template<typename Real >
static void	UnitTestCuMatrixMulRowsGroupMat ()
template<typename Real >
static void	UnitTestCuMatrixDiffGroupPnorm ()
template<typename Real >
static void	UnitTestCuMatrixGroupMaxDeriv ()
template<typename Real >
static void	UnitTestCuMatrixAddDiagVecMat ()
template<typename Real >
static void	UnitTestCuMatrixAddMatDiagVec ()
template<typename Real >
static void	UnitTestCuMatrixAddMatMatElements ()
template<typename Real >
static void	UnitTestCuMatrixSetMatMatDivMat ()
template<typename Real >
static void	UnitTestCuMatrixDivRowsVec ()
template<typename Real >
static void	UnitTestCuMatrixAddMat ()
template<typename Real >
static void	UnitTestCuMatrixAddMatBlocks1 ()
template<typename Real >
static void	UnitTestCuMatrixAddMatBlocks1Trans ()
template<typename Real >
static void	UnitTestCuMatrixAddMatBlocks2 ()
template<typename Real >
static void	UnitTestCuMatrixReduceSum ()
template<typename Real >
static void	UnitTestCuMatrixReduceMax ()
template<typename Real >
static void	UnitTestCuMatrixReduceMin ()
template<typename Real >
static void	UnitTestCuMatrixAddVecToCols ()
template<typename Real >
static void	UnitTestCuMatrixAddVecToRows ()
template<typename Real >
static void	UnitTestCuMatrixSymAddMat2 ()
template<typename Real >
static void	UnitTestCuMatrixSymInvertPosDef ()
template<typename Real >
static void	UnitTestCuMatrixAddMatMat ()
template<typename Real >
static void	UnitTestCuMatrixAddVecVec ()
template<typename Real >
static void	UnitTestCuMatrixAddMatMatBatched ()
template<typename Real >
static void	UnitTestCuMatrixAddToDiag ()
template<typename Real >
static void	UnitTestCuMatrixAdd2 ()
template<typename Real >
static void	UnitTestCuMatrixCopyFromMat ()
template<typename Real >
static void	UnitTestCuMatrixCopyFromTp ()
template<typename Real >
static void	UnitTestCuMatrixAddMatTp ()
template<typename Real >
static void	UnitTestCuMatrixTranspose ()
template<typename Real >
static void	UnitTestCuMatrixAddTpMat ()
template<typename Real >
static void	UnitTestCuVectorAddVec ()
template<typename Real >
static void	UnitTestCuVectorAddRowSumMat ()
template<typename Real >
static void	UnitTestCuVectorAddRowSumMatLarge ()
template<typename Real >
static void	UnitTestCuVectorAddColSumMat ()
template<typename Real >
static void	UnitTestCuSubMatrix ()
template<typename Real >
static void	UnitTestCuVectorAddColSumMatLarge ()
template<typename Real >
static void	UnitTestCuVectorInvertElements ()
template<typename Real >
static void	UnitTestCuMatrixInvertElements ()
template<class Real >
static void	UnitTestCuMatrixIO ()
template<typename Real >
static void	UnitTestCuVectorAddTpVec ()
template<typename Real >
static void	UnitTestCuApproxEqual ()
template<typename Real >
static void	UnitTestCuVectorMulTp ()
template<typename Real , typename OtherReal >
static void	UnitTestCuCopy ()
template<typename Real >
static void	UnitTestCuSigmoid ()
template<typename Real >
static void	UnitTestCuDiffSigmoid ()
template<typename Real >
static void	UnitTestCuDiffSoftmax ()
template<typename Real >
static void	UnitTestCuDiffLogSoftmax ()
template<typename Real >
static void	UnitTestCuSoftmax ()
template<typename Real >
static void	UnitTestCuLogSoftmax ()
template<typename Real >
static void	UnitTestCuFindRowMaxId ()
template<typename Real >
static void	UnitTestCuDiffXent ()
template<typename Real >
void	UnitTestCheck ()
template<typename Real >
void	UnitTestSwapCu2Cu ()
template<typename Real >
void	UnitTestSwapCu2M ()
template<typename Real >
void	UnitTestCuTanh ()
template<typename Real >
static void	UnitTestCuDiffTanh ()
static int32	DoubleFactorial (int32 i)
template<typename Real >
static void	UnitTestCuMatrixSetRandn ()
template<typename Real >
static void	UnitTestCuMatrixSetRandUniform ()
template<typename Real >
static void	UnitTestCuMatrixCopyLowerToUpper ()
template<typename Real >
static void	UnitTestCuMatrixSetZeroAboveDiag ()
template<typename Real >
static void	UnitTestCuMatrixCopyUpperToLower ()
template<typename Real >
static void	UnitTestCuMatrixObjfDeriv ()
template<typename Real >
static void	UnitTestCuMatrixAddElements ()
template<typename Real >
static void	UnitTestCuMatrixAddToElements ()
template<typename Real >
static void	UnitTestCuMatrixLookup ()
template<typename Real >
static void	UnitTestCuMatrixEqualElementMask ()
template<typename Real >
void	CudaMatrixUnitTest ()
int	main ()

Function Documentation

main()

int main

(

)

Definition at line 3061 of file cu-matrix-test.cc.

References rnnlm::i, KALDI_LOG, KALDI_WARN, and kaldi::SetVerboseLevel().

           {
  SetVerboseLevel(1);
  int32 loop = 0;
  bool test_threads = true;
  // num_threads only matters if test_threads == true.   Don't make it
  // to large, because it will affect CPU usage if you are using CPU.
  int32 num_threads = 4;


#if HAVE_CUDA == 1
  for (loop = 0; loop < 2; loop++) {
    CuDevice::Instantiate().SetDebugStrideMode(true);
    if (test_threads)
      CuDevice::Instantiate().AllowMultithreading();
    if (loop == 0)
      CuDevice::Instantiate().SelectGpuId("no");
    else
      CuDevice::Instantiate().SelectGpuId("yes");
#endif

    if (test_threads) {
      KALDI_LOG << "Doing matrix unit test with "
                << num_threads << " threads.";
      std::vector<std::thread*> threads;
      for (int32 i = 0;  i < num_threads - 1; i++)
        threads.push_back(new std::thread(kaldi::CudaMatrixUnitTest<float>));
      // the last thread running is the main thread.
      kaldi::CudaMatrixUnitTest<float>();
      for (size_t i = 0; i < threads.size(); i++) {
        threads[i]->join();
        delete threads[i];
      }
    } else {
      kaldi::CudaMatrixUnitTest<float>();
    }

#if HAVE_CUDA == 1
    if (CuDevice::Instantiate().DoublePrecisionSupported()) {
      kaldi::CudaMatrixUnitTest<double>();
    } else {
      KALDI_WARN << "Double precision not supported";
    }
#else
    kaldi::CudaMatrixUnitTest<double>();
#endif

    if (loop == 0)
      KALDI_LOG << "Tests without GPU use succeeded.";
    else
      KALDI_LOG << "Tests with GPU use (if available) succeeded.";
#if HAVE_CUDA == 1
  } // No for loop if 'HAVE_CUDA != 1',
  CuDevice::Instantiate().PrintProfile();
#endif
  return 0;
}