cu-test.cc File Reference

#include <iostream>
#include <vector>
#include <cstdlib>
#include <ctime>
#include "base/kaldi-common.h"
#include "cudamatrix/cu-device.h"
#include "cudamatrix/cu-sp-matrix.h"
#include "cudamatrix/cu-tp-matrix.h"
#include "cudamatrix/cu-packed-matrix.h"
#include "cudamatrix/cu-vector.h"
#include <numeric>
#include <time.h>

Include dependency graph for cu-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 (SpMatrix< Real > *M)
template<typename Real >
static void	InitRand (VectorBase< Real > *v)
template<typename Real >
static void	UnitTestSetZeroAboveDiag ()
template<typename Real >
static void	UnitTestCholesky ()
template<typename Real >
static void	UnitTestTrace ()
template<typename Real >
static void	UnitInvert ()
template<typename Real >
static void	UnitTestInvert ()
template<typename Real >
static void	UnitTestConstructor ()
template<typename Real >
static void	UnitTestCopySp ()
template<typename Real >
static void	UnitTestCopyFromMat ()
template<typename Real >
static void	UnitTestMatrix ()
template<typename Real >
static void	UnitTestMulTp ()
template<typename Real >
static void	UnitTestVector ()
template<typename Real >
static void	CuMatrixUnitTest ()
int	main ()

Function Documentation

main()

int main

(

)

Definition at line 576 of file cu-test.cc.

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

           {
  using namespace kaldi;
  SetVerboseLevel(1);
#if HAVE_CUDA == 1
  for (int32 loop = 0; loop < 2; loop++) {
    if (loop == 0)
      CuDevice::Instantiate().SelectGpuId("no");
    else
      CuDevice::Instantiate().SelectGpuId("yes");
#endif
    kaldi::CuMatrixUnitTest<float>();

#if HAVE_CUDA == 1
    if (!kaldi::CuDevice::Instantiate().DoublePrecisionSupported()) {
      KALDI_WARN << "Double precision not supported, not testing that code";
    } else
#endif
    {
      kaldi::CuMatrixUnitTest<double>();
    }
#if HAVE_CUDA == 1
  }
  kaldi::CuDevice::Instantiate().PrintProfile();
#endif

  KALDI_LOG << "Tests succeeded.";
  return 0;
}