[發明專利]卷積運算的處理方法、電子設備、移動終端及存儲介質

有效

• 申請號: 202110553804.4
• 申請日: 2021-05-20
• 公開（公告）號: CN113392957B
• 公開（公告）日: 2023-01-17
• 發明（設計）人: 莊晨;孟金濤;魏彥杰
• 申請（專利權）人: 中國科學院深圳先進技術研究院
• 主分類號: G06N3/04
• 分類號: G06N3/04;G06N3/08;G06F17/15
• 代理公司: 深圳市威世博知識產權代理事務所(普通合伙) 44280
• 代理人: 黎堅怡

1.一種卷積運算的處理方法，其特征在于，所述方法包括：

獲取待處理的卷積運算以及配置數據庫；

將所述卷積運算轉換為矩陣乘法，所述矩陣乘法對應一卷積尺寸，包括：對卷積執行Im2col算法，以將所述卷積運算轉換成所述卷積對應的矩陣乘法計算，所述矩陣乘法對應一卷積尺寸；

判斷所述配置數據庫中是否有所述卷積尺寸對應的配置參數；

若所述配置數據庫中有所述卷積尺寸對應的配置參數，則按照所述配置參數生成運算代碼并進行計算得到運算結果；

若確定所述配置數據庫中無所述卷積尺寸對應的配置參數，則根據所述卷積尺寸以及硬件參數定義一參數搜索空間，包括：根據所述硬件參數配置所述卷積尺寸對應的多組參數組合，得到所述配置參數；從多組所述參數組合中選取一組；基于選取的一組參數組合，定義對應的所述參數搜索空間；其中所述卷積尺寸對應的所述配置參數至少包括第一矩陣的行數、第一矩陣的列數、第一矩陣的緩存塊的行數、第一矩陣的緩存塊的列數、第二矩陣的列數、第二矩陣的緩存塊的列數、寄存器塊的行數、寄存器塊的列數、第一矩陣的預取值、第二矩陣的預取值以及搜索空間標簽；根據所述參數搜索空間中的配置參數生成多個運算代碼，并利用所述多個運算代碼對所述矩陣乘法進行計算，以得到多個運算結果，包括：

基于選取的參數組合，在所述參數搜索空間中，生成所述卷積對應的多個運算代碼；

利用所述多個運算代碼，對所述矩陣乘法進行計算，得到第一運算結果以及第I運算結果，其中I為大于1的正整數且小于或等于所述運算代碼的個數；

將所述多個運算結果中滿足預設條件的一個運算結果對應的運算代碼的配置參數，存儲至所述配置數據庫，包括：

判斷所述第一運算結果和/或所述第I運算結果是否滿足預設條件，所述預設條件至少包括所述矩陣乘法計算的時間段為多個運算結果中最短；

若所述第一運算結果和/或所述第I運算結果滿足預設條件，則將所述第一運算結果和/或所述第I運算結果對應的所述配置參數存儲至所述配置數據庫；

若所述第一運算結果和/或第I運算結果不滿足預設條件，則舍棄所述第I運算結果對應的所述配置參數，并將所述第一運算結果對應的所述配置參數存儲至所述配置數據庫。

2.根據權利要求1所述的處理方法，其特征在于，

所述第一矩陣的緩存塊的行數的取值范圍為[8，max(M，1024)]，所述M為所述第一矩陣的行數，所述第一矩陣的緩存塊的列數的取值范圍為[8，max(K，1024)]，所述K為所述第一矩陣的列數，所述第二矩陣的緩存塊的列數的取值范圍為[8，max(N，1024)]，所述N為所述第二矩陣的行數，所述寄存器塊的行數的取值范圍為4或8，所述寄存器塊的列數為8、12或16，第一矩陣的預取值以及第二矩陣的預取值至少包括0、32、64、128、256或512之一，所述搜索空間標簽取值至少包括0、1、2或3。

3.一種移動終端，其特征在于，包括：處理器和存儲器，所述存儲器中存儲有計算機程序，所述處理器用于執行所述計算機程序以實現如權利要求1或2所述的處理方法。

4.一種計算機可讀存儲介質，其特征在于，該計算機可讀存儲介質存儲有計算機程序，計算機程序能夠被處理器執行時實現如權利要求1或2所述的處理方法。