杨维科, 贺光辉, 景乃锋. 基于Rocket-Chip开源处理器的CNN加速模块的设计及实现[J]. 微电子学与计算机, 2018, 35(4): 17-21.
引用本文: 杨维科, 贺光辉, 景乃锋. 基于Rocket-Chip开源处理器的CNN加速模块的设计及实现[J]. 微电子学与计算机, 2018, 35(4): 17-21.
YANG Wei-ke, HE Guang-hui, JING Nai-feng. Design and Implementation of CNN Acceleration Module Based on Rocket-Chip Open Source Processor[J]. Microelectronics & Computer, 2018, 35(4): 17-21.
Citation: YANG Wei-ke, HE Guang-hui, JING Nai-feng. Design and Implementation of CNN Acceleration Module Based on Rocket-Chip Open Source Processor[J]. Microelectronics & Computer, 2018, 35(4): 17-21.

基于Rocket-Chip开源处理器的CNN加速模块的设计及实现

Design and Implementation of CNN Acceleration Module Based on Rocket-Chip Open Source Processor

  • 摘要: 基于RISC-V开源指令集及Rocket-Chip开源处理器, 提出了一种基于Eyeriss结构的卷积神经网络加速模块, 并与处理器连接形成完整的系统.该加速器结构通过进行横向(卷积核权值), 纵向(输出计算结果)以及斜向(输入图像)的数据重用, 大大减少了卷积层的时间消耗.此外, 利用加州大学伯克利分校开发的Chisel3语言, 一方面生成该系统的C语言模拟器进行调试, 另一方面生成Verilog进行综合和布局布线.通过对LeNet-5手写数字识别网络及MNIST数据集进行测试, 准确度、速度等都达到了令人满意的结果.

     

    Abstract: Based on the RISC-V open source instruction set and Rocket-Chip open source processor, we propose a convolution neural network acceleration module based on Eyeriss structure and form a complete system with the processor connection. The accelerator structure greatly reduces the time consumption of the convolution layer by performing data reuse in the lateral (convolution core weight), longitudinal (output calculation result) and oblique (input image). In addition, the use of the UCB developed Chisel3 language, on the one hand to generate the system C language simulator for debugging, on the other hand generate Verilog for integrated and layout and routing. Through the LeNet-5 handwritten digital identification network and MNIST data set, we achieved satisfactory results on speed and accuracy.

     

/

返回文章
返回