[實(shí)用新型]一種卷積神經(jīng)網(wǎng)絡(luò)推理加速器

有效

• 申請(qǐng)?zhí)枺?/td>: 202020683437.0
• 申請(qǐng)日: 2020-04-28
• 公開（公告）號(hào): CN211554991U
• 公開（公告）日: 2020-09-22
• 發(fā)明（設(shè)計(jì)）人: 李麗;黃延;傅玉祥;宋文清;何書專
• 申請(qǐng)（專利權(quán)）人: 南京寧麒智能計(jì)算芯片研究院有限公司
• 主分類號(hào): G06N3/04
• 分類號(hào): G06N3/04;G06N3/063;G06N5/04;G06K9/00
• 代理公司: 江蘇瑞途律師事務(wù)所 32346
• 代理人: 金龍

1.一種卷積神經(jīng)網(wǎng)絡(luò)推理加速器，其特征在于，包括主控制模塊、地址產(chǎn)生模塊，SRAM存儲(chǔ)模塊、數(shù)據(jù)輸入模塊、計(jì)算引擎模塊及結(jié)果輸出模塊，其中：

主控制模塊，接收開始信號(hào)，向地址產(chǎn)生模塊、數(shù)據(jù)輸入模塊、計(jì)算引擎模塊和結(jié)果輸出模塊分發(fā)數(shù)據(jù)；

地址產(chǎn)生模塊，接收主控制模塊和結(jié)果輸出模塊的數(shù)據(jù)，向SRAM存儲(chǔ)模塊發(fā)送控制信號(hào)；

SRAM存儲(chǔ)模塊，接收地址產(chǎn)生模塊的控制信號(hào)，向數(shù)據(jù)輸入模塊發(fā)送存儲(chǔ)數(shù)據(jù)；

數(shù)據(jù)輸入模塊，接收主控制模塊和SRAM存儲(chǔ)模塊的數(shù)據(jù)，將數(shù)據(jù)發(fā)送至計(jì)算引擎模塊；

計(jì)算引擎模塊，接收主控制模塊和數(shù)據(jù)輸入模塊的數(shù)據(jù)，將結(jié)果數(shù)據(jù)發(fā)送至結(jié)果輸出模塊；

結(jié)果輸出模塊，接收計(jì)算引擎模塊的結(jié)果數(shù)據(jù)，將結(jié)果數(shù)據(jù)發(fā)送至地址產(chǎn)生模塊。

2.根據(jù)權(quán)利要求1所述的一種卷積神經(jīng)網(wǎng)絡(luò)推理加速器，其特征在于：所述計(jì)算引擎模塊包括若干個(gè)并行計(jì)算單元。

3.根據(jù)權(quán)利要求2所述的一種卷積神經(jīng)網(wǎng)絡(luò)推理加速器，其特征在于：所述并行計(jì)算單元包括若干個(gè)乘累加單元和激活函數(shù)計(jì)算單元，乘累加單元通過數(shù)據(jù)輸入總線接收數(shù)據(jù)輸入模塊的數(shù)據(jù)，通過結(jié)果輸入總線將數(shù)據(jù)發(fā)送至激活函數(shù)計(jì)算單元。

4.根據(jù)權(quán)利要求3所述的一種卷積神經(jīng)網(wǎng)絡(luò)推理加速器，其特征在于：所述乘累加單元包括定點(diǎn)乘法器和定點(diǎn)加法器，定點(diǎn)乘法器與定點(diǎn)加法器相連接。

5.根據(jù)權(quán)利要求3所述的一種卷積神經(jīng)網(wǎng)絡(luò)推理加速器，其特征在于：所述激活函數(shù)單元包括查找表存儲(chǔ)器、定點(diǎn)乘法器、定點(diǎn)加法器以及兩個(gè)寄存器，查找表存儲(chǔ)器和第一寄存器接收數(shù)據(jù)，查找表存儲(chǔ)器將數(shù)據(jù)發(fā)送至定點(diǎn)乘法器和第二寄存器，定點(diǎn)乘法器與第二寄存器將數(shù)據(jù)發(fā)送至定點(diǎn)加法器。

6.根據(jù)權(quán)利要求1所述的一種卷積神經(jīng)網(wǎng)絡(luò)推理加速器，其特征在于：所述主控制模塊向地址產(chǎn)生模塊發(fā)送的數(shù)據(jù)包括開始信號(hào)和補(bǔ)零信號(hào)。

7.根據(jù)權(quán)利要求6所述的一種卷積神經(jīng)網(wǎng)絡(luò)推理加速器，其特征在于：地址產(chǎn)生模塊接收補(bǔ)零信號(hào)，將生成的補(bǔ)零地址發(fā)送至結(jié)果輸出模塊，結(jié)果輸出模塊將結(jié)果數(shù)據(jù)與補(bǔ)零地址發(fā)送至SRAM存儲(chǔ)模塊。

8.根據(jù)權(quán)利要求1所述的一種卷積神經(jīng)網(wǎng)絡(luò)推理加速器，其特征在于：計(jì)算引擎模塊接收到數(shù)據(jù)輸入模塊的數(shù)據(jù)包括輸入數(shù)據(jù)有效信號(hào)、模塊關(guān)斷信號(hào)和定點(diǎn)位置信號(hào)。