A Software-Hardware Co-design Platform for Convolutional Neural Network Accelerator with Tengine
摘要
With the development of convolutional neural network, its model scale and data volume are getting bigger and bigger, so the convolutional neural network gas pedal has been born. NVDLA as an open-source project from NVIDIA, supports a variety of classical models. However, due to the complexity of the ASIC design flow and the lack of support for the latest operators, its application scenarios are significantly limited. In this paper, we propose a method to map NVDLA onto FPGA, and employ the Tengine toolchain to replace the original software stack provided by the NVDLA. This approach optimizes the performance of the supported LeNet5 and ResNet18 models. Experimental results show that the Tengine toolchain increases inference speed by 151% compared to the NVDLA website’s software toolchain.