cu-vector-speed-test.cc File Reference

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

Include dependency graph for cu-vector-speed-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 >
std::string	NameOf ()
template<typename Real >
void	TestCuVectorSoftmax (int32 dim)
template<typename Real >
void	TestCuVectorSum (int32 dim)
template<typename Real , typename OtherReal >
void	TestCuVectorCopyFromVec (int32 dim)
template<typename Real >
void	TestCuVectorVecVecOne (int32 dim)
template<typename Real >
void	TestCuVectorAddDiagMatMat (int32 dim, MatrixTransposeType transN, MatrixTransposeType transO)
template<typename Real >
void	TestCuVectorAddDiagMat2OnVariousShapes (int32 dim, MatrixTransposeType trans)
template<typename Real >
void	TestCuVectorAddDiagMat2 (int32 dim, MatrixTransposeType trans)
template<typename Real >
void	TestCuVectorAddRowSumMat (int32 dim, MatrixTransposeType trans)
template<typename Real >
void	TestCuVectorAddColSumMat (int32 dim, MatrixTransposeType trans)
template<typename Real >
void	TestCuVectorApplyFloor (int32 dim)
template<typename Real >
void	TestCuVectorApplyFloorNoCount (int32 dim)
template<typename Real >
void	TestCuVectorApplyCeiling (int32 dim)
template<typename Real >
void	TestCuVectorApplyCeilingNoCount (int32 dim)
template<typename Real >
void	TestCuVectorAddDiagMatMatShape (int32 num_rows, int32 num_cols, MatrixTransposeType transM, MatrixTransposeType transN)
template<typename Real >
void	CudaVectorSpeedTest ()
int	main ()

Function Documentation

main()

int main

(

)

Definition at line 471 of file cu-vector-speed-test.cc.

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

           {
  kaldi::SetVerboseLevel(1);
  //Select the GPU
#if HAVE_CUDA == 1
  CuDevice::Instantiate().SelectGpuId("yes"); //-2 .. automatic selection
#endif

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