cu-packed-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-packed-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	AssertEqual (const CuPackedMatrix< Real > &A, const CuPackedMatrix< Real > &B, float tol=0.001)
template<typename Real >
static void	AssertEqual (const PackedMatrix< Real > &A, const PackedMatrix< Real > &B, float tol=0.001)
template<typename Real >
static void	AssertDiagEqual (const PackedMatrix< Real > &A, const CuPackedMatrix< Real > &B, float value, float tol=0.001)
template<typename Real >
static void	AssertDiagEqual (const PackedMatrix< Real > &A, const PackedMatrix< Real > &B, float value, float tol=0.001)
template<typename Real >
static void	AssertEqual (const PackedMatrix< Real > &A, const CuPackedMatrix< Real > &B, float tol=0.001)
template<typename Real >
static bool	ApproxEqual (const PackedMatrix< Real > &A, const PackedMatrix< Real > &B, Real tol=0.001)
template<typename Real >
static void	UnitTestCuPackedMatrixConstructor ()
template<typename Real >
static void	UnitTestCuPackedMatrixCopy ()
template<typename Real >
static void	UnitTestCuPackedMatrixTrace ()
template<typename Real >
static void	UnitTestCuPackedMatrixScale ()
template<typename Real >
static void	UnitTestCuPackedMatrixScaleDiag ()
template<typename Real >
static void	UnitTestCuPackedMatrixAddToDiag ()
template<typename Real >
static void	UnitTestCuPackedMatrixSetUnit ()
template<typename Real >
void	CudaPackedMatrixUnitTest ()
int	main ()

Function Documentation

main()

int main

(

)

Definition at line 243 of file cu-packed-matrix-test.cc.

References KALDI_LOG, and KALDI_WARN.

           {
  using namespace kaldi;
#if HAVE_CUDA == 1
  CuDevice::Instantiate().SetDebugStrideMode(true);
  // Select the GPU
  CuDevice::Instantiate().SelectGpuId("yes");
#endif
  kaldi::CudaPackedMatrixUnitTest<float>();
#if HAVE_CUDA == 1
  if (CuDevice::Instantiate().DoublePrecisionSupported()) {
    kaldi::CudaPackedMatrixUnitTest<double>();
  } else {
    KALDI_WARN << "Double precision not supported";
  }
#else
  kaldi::CudaPackedMatrixUnitTest<double>();
#endif
  
  KALDI_LOG << "Tests succeeded";
#if HAVE_CUDA == 1
  CuDevice::Instantiate().PrintProfile();
#endif
  return 0;
}