[發明專利]神經網絡處理裝置及其執行向量乘加指令的方法

有效

• 申請號: 201810914600.7
• 申請日: 2018-08-08
• 公開（公告）號: CN109165732B
• 公開（公告）日: 2022-05-31
• 發明（設計）人: 王秉睿;周聖元;張堯
• 申請（專利權）人: 上海寒武紀信息科技有限公司
• 主分類號: G06N3/063
• 分類號: G06N3/063;G06N3/04
• 代理公司: 廣州三環專利商標代理有限公司 44202
• 代理人: 郝傳鑫;熊永強

1.一種處理裝置，用于根據向量乘加指令執行運算，其特征在于，所述處理裝置包括：存儲器、運算單元、控制單元和寄存器單元；

所述寄存器單元，用于存儲向量乘加指令；所述向量乘加指令包括操作碼和操作域，所述操作碼用于指示進行向量乘加運算；所述操作域包括參與向量乘加運算的數據的首地址和參與向量乘加運算的數據中部分數據或者全部數據的索引的地址；

所述存儲器，用于存儲參與向量乘加運算的數據；

所述控制單元，用于從所述寄存器單元中獲取所述向量乘加指令，解析所述向量乘加指令，以得到所述操作碼和操作域；根據所述操作域從所述存儲器中獲取所述參與向量乘加運算的數據和所述參與向量乘加運算的數據中部分數據或者全部數據的索引；并將所述參與向量乘加運算的數據傳輸至所述運算單元；

所述運算單元，用于對所述參與向量乘加運算的數據進行向量乘加運算；

其中，當所述向量乘加指令為第一向量乘加指令時，所述操作碼用于指示進行第一向量乘加運算，所述操作域包括：稀疏向量X6的索引表示方式、稀疏向量X6中元素的首地址、稀疏向量X6的索引的首地址、向量Y6中元素的首地址和第四標量的首地址；

所述控制單元具體用于：根據所述稀疏向量X6中元素的首地址從所述存儲器中獲取所述稀疏向量X6；根據所述向量Y6中元素的首地址從所述存儲器中獲取所述向量Y6；根據所述稀疏向量X6的索引的首地址從所述存儲器中分別獲取所述稀疏向量X6的索引；根據所述第四標量的首地址從所述存儲器中獲取所述第四標量；根據所述稀疏向量X6的索引表示方式和所述稀疏向量X6的索引對所述向量Y6中的元素進行篩選，以得到稀疏向量Y6；并將所述稀疏向量X6、稀疏向量Y6和第四標量傳輸至所述運算單元；

所述運算單元具體用于：對所述稀疏向量X6、稀疏向量Y6和第四標量進行第一向量乘加運算；

或者，

當所述向量乘加指令為第二向量乘加指令時，所述操作碼用于指示進行第二向量乘加運算，所述操作域包括：稀疏向量sX6的索引表示方式、稀疏向量sX6中元素的首地址、稀疏向量sX6的索引的首地址、稀疏向量sY6中元素的首地址、稀疏向量sY6的索引的首地址和第四標量的首地址；

所述控制單元具體用于：根據所述稀疏向量sX6中元素的首地址和所述稀疏向量sY6中元素的首地址分別從所述存儲器中獲取所述稀疏向量sX6和所述稀疏向量sY6；根據所述稀疏向量sX6的索引的首地址和所述稀疏向量sY6的索引的首地址分別從所述存儲器中分別獲取所述稀疏向量sX6的索引和稀疏向量sY6的索引；根據所述第四標量的首地址從所述存儲器中獲取所述第四標量；根據所述稀疏向量sX6的索引表示方式、所述稀疏向量sX6的索引和所述稀疏向量sY6的索引得到第一中間索引；根據所述第一中間索引對所述稀疏向量sX6和所述稀疏向量sY6中的元素進行篩選，以得到篩選后的稀疏向量sX6和篩選后的稀疏向量sY6；并將所述篩選后的稀疏向量sX6和篩選后的稀疏向量sY6和第四標量傳輸至所述運算單元；所述運算單元對所述篩選后的稀疏向量sX6和篩選后的稀疏向量sY6和第四標量進行第二向量乘加運算。

2.根據權利要求1所述的處理裝置，其特征在于，所述操作域還包括：稀疏向量X6中元素的個數、向量Y6中元素的個數和向量Y6中元素之間的地址間隔；

所述控制單元根據所述稀疏向量X6中元素的首地址從所述存儲器中獲取所述稀疏向量X6，包括：

根據所述稀疏向量X6中元素的首地址和所述稀疏向量X6中元素的個數從所述存儲器中獲取所述稀疏向量X6；

所述控制單元根據所述向量Y6中元素的首地址從所述存儲器中獲取所述向量Y6，包括：

根據所述向量Y6中元素的首地址、所述向量Y6中元素的個數和所述向量Y6中元素之間的地址間隔從所述存儲器中獲取所述向量Y6。