CUDA_Test
The main role of the project: CUDA 8.0/TensorRT's usage(each test code gives the implementation of C ++ and CUDA, respectively, and gives the calculation time for each method):
- 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
- simple
- 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.
- Linux:
- CUDA supports cmake build(file position: prj/linux_cuda_cmake)
- TensorRT support cmake build(file position: prj/linux_tensorrt_cmake)
Blog: fengbingchun