/CUDA_Test

CUDA's usage

Primary LanguageC++

CUDA_Test

The main role of the project:

  • CUDA 8.0'usage(each test code gives the implementation of C ++ and CUDA, respectively, and gives the calculation time for each method)
  • TensorRT's usage
  • Simd's usage Simd GitHub

CUDA 8.0 test code:

  • simple
    • vector add: C = A + B
    • matrix multiplication: C = A * B
    • dot product
    • Julia
    • ripple
    • green ball
    • ray tracking
    • heat conduction
    • calculate histogram
    • streams' usage
  • layer(approximate)
    • channel normalize(mean/standard deviation)
    • reverse
    • prior_vbox
  • image process
    • bgr to gray
    • bgr to bgr565
    • gray image histogram equalization(only C++ implementation)
    • gray image edge detection: Laplacian(only C++ implementation)

TensorRT 2.1.2 test code:

  • MNIST
  • MNIST API(use api produce network)
  • GoogleNet
  • CharRNN
  • Plugin(add a custom layer)
  • MNIST Infer(serialize TensorRT model)

The project support platform:

  • windows10 64 bits: It can be directly build with VS2013 in windows10 64bits(Except TensorRT).
  • Linux:
    • CUDA supports cmake build(file position: prj/linux_cuda_cmake)
    • TensorRT support cmake build(file position: prj/linux_tensorrt_cmake)

Screenshot:

Blog: fengbingchun