cu-sp-matrix-test.cc File Reference

#include <iostream>
#include <vector>
#include <cstdlib>
#include "base/kaldi-common.h"
#include "cudamatrix/cu-device.h"
#include "cudamatrix/cu-sp-matrix.h"
#include "cudamatrix/cu-vector.h"
#include "cudamatrix/cu-math.h"

Include dependency graph for cu-sp-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	UnitTestCuSpMatrixConstructor ()
template<typename Real >
static void	UnitTestCuSpMatrixApproxEqual ()
template<typename Real >
static void	UnitTestCuSpMatrixOperator ()
template<typename Real >
static void	UnitTestCuSpMatrixAddToDiag ()
template<typename Real >
static void	UnitTestCuSpMatrixCopyFromMat ()
template<typename Real >
static void	UnitTestCuSpMatrixInvert ()
template<typename Real >
static void	UnitTestCuSpMatrixAddVec2 ()
template<typename Real >
static void	UnitTestCuSpMatrixAddMat2 ()
template<typename Real >
static void	UnitTestCuSpMatrixAddSp ()
template<typename Real , typename OtherReal >
static void	UnitTestCuSpMatrixTraceSpSp ()
template<typename Real >
void	UnitTestCuSpMatrixSetUnit ()
template<class Real >
static void	UnitTestCuSpMatrixIO ()
template<typename Real , typename OtherReal >
static void	UnitTestCuSpMatrixAddSp ()
template<typename Real >
void	CudaSpMatrixUnitTest ()
template<typename Real , typename OtherReal >
void	CudaSpMatrixUnitTest ()
int	main ()

Function Documentation

main()

int main

(

)

Definition at line 364 of file cu-sp-matrix-test.cc.

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

           {
  using namespace kaldi;
  SetVerboseLevel(1);
  int32 loop = 0;
#if HAVE_CUDA == 1
  for (; loop < 2; loop++) {
    CuDevice::Instantiate().SetDebugStrideMode(true);
    if (loop == 0)
      CuDevice::Instantiate().SelectGpuId("no"); // -1 means no GPU
    else
      CuDevice::Instantiate().SelectGpuId("yes"); // -2 .. automatic selection
#endif

    kaldi::CudaSpMatrixUnitTest<float>();
    kaldi::CudaSpMatrixUnitTest<float, float>();
#if HAVE_CUDA == 1
    if (CuDevice::Instantiate().DoublePrecisionSupported()) {
      kaldi::CudaSpMatrixUnitTest<double>();
      kaldi::CudaSpMatrixUnitTest<float, double>();
      kaldi::CudaSpMatrixUnitTest<double, float>();
      kaldi::CudaSpMatrixUnitTest<double, double>();
    } else {
      KALDI_WARN << "Double precision not supported";
    }
#else
    kaldi::CudaSpMatrixUnitTest<float, double>();
    kaldi::CudaSpMatrixUnitTest<double, float>();
    kaldi::CudaSpMatrixUnitTest<double, 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
  }
  CuDevice::Instantiate().PrintProfile();
#endif
  return 0;
}