[發明專利]內存分配方法、內存分配裝置、電子設備和可讀存儲介質

在審

• 申請號: 202111108224.0
• 申請日: 2021-09-22
• 公開（公告）號: CN113791910A
• 公開（公告）日: 2021-12-14
• 發明（設計）人: 馮浩恩
• 申請（專利權）人: 維沃移動通信有限公司
• 主分類號: G06F9/50
• 分類號: G06F9/50;G06N3/04
• 代理公司: 北京友聯知識產權代理事務所(普通合伙) 11343
• 代理人: 尚志峰;汪海屏

1.一種內存分配方法，其特征在于，包括：

獲取神經網絡的全局信息；

基于所述全局信息獲取所述神經網絡的內存塊復用策略；

基于所述內存塊復用策略確定所述神經網絡的內存分配策略。

2.根據權利要求1所述的內存分配方法，其特征在于，所述全局信息包括所述神經網絡的節點的所需內存塊；

所述基于所述全局信息獲取所述神經網絡的內存塊復用策略，包括：

按照所述所需內存塊由大到小的順序，將所述神經網絡的節點排序；

根據所述節點的排序順序，為所述神經網絡的節點分配內存塊，生成至少一個所述內存塊復用策略；

所述基于所述內存塊復用策略確定所述神經網絡的內存分配策略，包括：

將第一目標復用策略作為所述神經網絡的內存分配策略，其中，所述第一目標復用策略為內存塊的總和小于或等于第一預設閾值的內存塊復用策略。

3.根據權利要求2所述的內存分配方法，其特征在于，所述根據所述節點的排序順序，為所述神經網絡的節點分配內存塊，生成至少一個所述內存塊復用策略，包括：

根據所述節點的排序順序，將所述神經網絡的節點劃分為N個節點組；

為每個所述節點組的節點分配內存塊，生成至少一個所述內存塊復用策略；

其中，N為大于或等于4的正整數。

4.根據權利要求3所述的內存分配方法，其特征在于，所述為每個所述節點組的節點分配內存塊，生成至少一個所述內存塊復用策略，包括：

確定第1個節點組中的第i個節點的至少一個第一復用內存塊，并為所述第i個節點分配所述第一復用內存塊，生成至少一個第一內存塊復用策略；

確定所述第i個節點的第一非復用內存塊，并為所述第i個節點分配所述第一非復用內存塊，生成至少一個第二內存塊復用策略；

將第二目標復用策略作為所述內存塊復用策略；

其中，i大于或等于1且小于或等于O，O為所述第1個節點組的節點的總數量，所述第二目標復用策略為內存塊的總和小于或等于第二預設閾值的所述第一內存塊復用策略或所述第二內存塊復用策略。

5.根據權利要求3所述的內存分配方法，其特征在于，所述為每個所述節點組的節點分配內存塊，生成至少一個所述內存塊復用策略，包括：

基于第M-1個節點組確定的所述內存塊復用策略，確定第M個節點組中的第j個節點的至少一個第二復用內存塊，并為所述第j個節點分配所述第二復用內存塊，生成至少一個第三內存塊復用策略；

確定所述第j個節點的第二非復用內存塊，并為所述第i個節點分配所述第二非復用內存塊，生成至少一個第四內存塊復用策略；

將第三目標復用策略作為所述內存塊復用策略；

其中，M大于1且小于N，j大于或等于1且小于或等于P，P為所述第M個節點組的節點的總數量，所述第三目標復用策略為內存塊的總和小于或等于第三預設閾值的所述第三內存塊復用策略或所述第四內存塊復用策略。

6.根據權利要求3至5中任一項所述的內存分配方法，其特征在于，所述全局信息還包括所述神經網絡的節點的拓撲結構信息；

所述為每個所述節點組的節點分配內存塊，生成至少一個所述內存塊復用策略，包括：

基于第N-1個節點組確定的所述內存塊復用策略，根據第N個節點組中的第t個節點在所述拓撲結構上的前2個節點的內存塊，為所述第t個節點分配內存塊，生成至少一個第五內存塊復用策略；

將所述第五內存塊復用策略作為所述內存塊復用策略；

其中，t大于或等于1且小于或等于Q，Q為所述第N個節點組的節點的總數量。

7.一種內存分配裝置，其特征在于，包括：

獲取模塊，用于獲取神經網絡的全局信息；

分配模塊，用于基于所述全局信息獲取所述神經網絡的內存塊復用策略；

確定模塊，用于基于所述內存塊復用策略確定所述神經網絡的內存分配策略。