[发明专利]内存分配方法、内存分配装置、电子设备和可读存储介质

权利要求书

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.一种内存分配装置，其特征在于，包括：

获取模块，用于获取神经网络的全局信息；

分配模块，用于基于所述全局信息获取所述神经网络的内存块复用策略；

确定模块，用于基于所述内存块复用策略确定所述神经网络的内存分配策略。