[發明專利]資源調度方法、硬件加速器及電子設備

在審

• 申請號: 202010812916.2
• 申請日: 2020-08-13
• 公開（公告）號: CN111949405A
• 公開（公告）日: 2020-11-17
• 發明（設計）人: 劉君
• 申請（專利權）人: OPPO廣東移動通信有限公司
• 主分類號: G06F9/50
• 分類號: G06F9/50;G06F9/48;G06N3/063
• 代理公司: 北京派特恩知識產權代理有限公司 11270
• 代理人: 侯艷華;張穎玲

本申請公開了一種資源調度方法、硬件加速器及電子設備。資源調度方法應用于硬件加速器，硬件加速器的硬件架構包括至少一個憶阻器片；至少一個憶阻器片中的每個憶阻器片包括至少一個憶阻器陣列；資源調度方法包括：確定設定計算模型的模型信息；設定計算模型包括至少一個計算層；基于所述模型信息和所述硬件架構，確定所述硬件加速器在執行至少一個計算層中每個計算層的運算時對應的配置信息；所述配置信息至少包括所述至少一個憶阻器片中每個憶阻器片的工作模式；所述工作模式表征每一種硬件資源在憶阻器片中對應配置的憶阻器陣列的數量；在執行所述至少一個計算層中每個計算層的運算時，根據對應的配置信息對所述硬件加速器進行硬件資源調度。

技術領域

本申請涉及通信領域，尤其涉及資源調度方法、硬件加速器及電子設備。

背景技術

相關技術中，通常采用基于憶阻器(ReRAM，Resistance Random Access Memory)的加速器來執行神經網絡的計算任務。由于不同的神經網絡的計算量和存儲量不同，在調度加速器中的硬件資源來執行相關的計算任務的過程中，存在加速器的硬件資源和實際的資源需求不匹配的問題，從而導致硬件資源浪費或硬件資源無法滿足計算任務的相關需求。

發明內容

有鑒于此，本申請實施例期望提供一種資源調度方法、硬件加速器及電子設備，以解決相關技術中基于憶阻器的加速器在執行神經網絡的計算任務時，加速器的硬件資源和計算任務的實際的資源需求不匹配的技術問題。

為達到上述目的，本申請的技術方案是這樣實現的：

本申請實施例提供一種資源調度方法，應用于硬件加速器；所述硬件加速器的硬件架構包括至少一個憶阻器片；所述至少一個憶阻器片中的每個憶阻器片包括至少一個憶阻器陣列；

所述資源調度方法包括：

確定設定計算模型的模型信息；所述設定計算模型包括至少一個計算層；

基于所述模型信息和所述硬件架構，確定所述硬件加速器在執行所述至少一個計算層中每個計算層的運算時對應的配置信息；所述配置信息至少包括所述至少一個憶阻器片中每個憶阻器片的工作模式；所述工作模式表征每一種硬件資源在憶阻器片中對應配置的憶阻器陣列的數量；所述硬件資源包括以下至少之一：計算資源、存儲資源和緩存資源；

在執行所述至少一個計算層中每個計算層的運算時，根據對應的配置信息對所述硬件加速器進行硬件資源調度。

上述方案中，所述模型信息包括所述至少一個計算層中每個計算層的拓撲結構和權重參數；

所述確定所述硬件加速器在執行所述至少一個計算層中每個計算層的運算時對應的配置信息時，所述資源調度方法包括：

基于計算層相關的拓撲結構和權重參數，確定出計算層對應的至少一類特征參數；所述至少一類特征參數中的每類特征參數對應表征計算層對對應類別的硬件資源的需求；

根據所述至少一類特征參數中的每類特征參數和所述硬件架構，確定出計算層對應的配置信息；其中，所述配置信息至少包括：

第一數量；所述第一數量表征憶阻器片中配置為計算資源的憶阻器陣列的數量；

第二數量；所述第二數量表征憶阻器片中配置為存儲資源的憶阻器陣列的數量；

第三數量；所述第三數量表征憶阻器片中配置為緩存資源的憶阻器陣列的數量。

上述方案中，所述設定計算模型為神經網絡模型；所述根據所述至少一類特征參數中的每類特征參數和所述硬件架構，確定出計算層對應的配置信息時，所述資源調度方法包括：

對應于所述神經網絡模型的卷積層或全連接層，基于濾波器相關信息和所述憶阻器片中憶阻器陣列的尺寸信息，確定出所述第一數量；所述濾波器相關信息包括濾波器的數量和濾波器的尺寸信息。