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

说明书

本申请公开了一种内存分配方法、内存分配装置、电子设备和可读存储介质，属于通信技术领域。该内存分配方法包括：获取神经网络的全局信息；基于全局信息获取神经网络的内存块复用策略；基于内存块复用策略确定神经网络的内存分配策略。

技术领域

本申请属于通信技术领域，具体涉及一种内存分配方法、内存分配装置、电子设备和可读存储介质。

背景技术

随着人工智能和大数据的蓬勃发展，为了兼顾用户体验和隐私保护，越来越多深度神经网络算法部署到电子设备上。深度神经网络算法的前向推理计算需要申请大量的运行内存才能确保中间计算的正确性，而电子设备上运行内存的限制是阻碍移动端神经网络设计的一大因素。

发明内容

本申请实施例的目的是提供一种内存分配方法、内存分配装置、电子设备和可读存储介质，能够解决相关技术中深度神经网络算法的运行内存的限制的问题。

第一方面，本申请实施例提供了一种内存分配方法，该内存分配方法包括：

获取神经网络的全局信息；

基于全局信息获取神经网络的内存块复用策略；

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

第二方面，本申请实施例提供了一种内存分配装置，该内存分配装置包括：

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

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

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

第三方面，本申请实施例提供了一种电子设备，该电子设备包括处理器、存储器及存储在存储器上并可在处理器上运行的程序或指令，程序或指令被处理器执行时实现如第一方面的方法的步骤。

第四方面，本申请实施例提供了一种可读存储介质，可读存储介质上存储程序或指令，程序或指令被处理器执行时实现如第一方面的方法的步骤。

第五方面，本申请实施例提供了一种芯片，芯片包括处理器和通信接口，通信接口和处理器耦合，处理器用于运行程序或指令，实现如第一方面的方法。

在本申请实施例中，提出一种神经网络内存复用策略的实现方案，具体地，根据神经网络的全局信息，为神经网络的节点确定内存块复用策略，从而在对神经网络使用时按照内存块复用策略为神经网络的节点分配内存。通过上述方式，能够提高电子设备内存的复用率，降低神经网络前向推理所需的总运行内存，从而增强电子设备神经网络的复杂度、运行平台的多样性以及并行神经网络的个数。

附图说明

图1是本申请实施例的内存分配方法的流程示意图；

图2是本申请实施例的内存分配方式的示意图之一；

图3是本申请实施例的内存分配方式的示意图之二；

图4是本申请实施例的内存分配方式的示意图之三；

图5是本申请实施例的内存分配装置的示意框图；

图6是本申请实施例的电子设备的示意框图之一；

图7是本申请实施例的电子设备的示意框图之二。

具体实施方式

下面将结合本申请实施例中的附图，对本申请实施例中的技术方案进行清楚地描述，显然，所描述的实施例是本申请一部分实施例，而不是全部的实施例。基于本申请中的实施例，本领域普通技术人员获得的所有其他实施例，都属于本申请保护的范围。