<p>This work presents an open, efficient (fast) CUDA convolution neural network inference implementation specialized in some layers of popular nets like ResNet, VGG, and GoogLeNet. The proposed algorithm implements convolution directly instead of preprocessing with image to columns. Algorithm parameters are selected to meet constraints on global and shared memory access bandwidth, register usage, shared memory usage, and instructions per clock. Parallel arithmetic operations and memory access are achieved with several parallel blocks per streaming processor. Results comparing with state-of-the-art implementations are presented.</p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Open CUDA convolution neural network inference implementation

  • Paulo Lopes

摘要

This work presents an open, efficient (fast) CUDA convolution neural network inference implementation specialized in some layers of popular nets like ResNet, VGG, and GoogLeNet. The proposed algorithm implements convolution directly instead of preprocessing with image to columns. Algorithm parameters are selected to meet constraints on global and shared memory access bandwidth, register usage, shared memory usage, and instructions per clock. Parallel arithmetic operations and memory access are achieved with several parallel blocks per streaming processor. Results comparing with state-of-the-art implementations are presented.