[发明专利]一种内存管理方法、装置、设备及计算机可读存储介质

说明书

本申请实施例提供了一种内存管理方法、装置、设备及计算机可读存储介质；方法包括：获取目标任务请求；响应目标任务请求，获取前次任务执行时所使用的前次内存块；其中，前次任务为执行目标任务之前所执行的任务，前次内存块为前次任务执行时所使用的各个内存块构成的集合，内存块为内存分配的最小分配单元；通过遍历前次内存块，确定出目标任务对应的目标内存块；基于目标内存块，执行目标任务。通过本申请实施例，能够降低任务执行的内存消耗。

技术领域

本申请涉及计算机应用技术领域中人工智能技术，尤其涉及一种内存管理方法、装置、设备及计算机可读存储介质。

背景技术

内存是处理器直接寻址的存储空间，是外存与处理器进行沟通的桥梁；计算机中所有任务的执行，都需要利用内存来暂时存放处理器中的运算数据，以及与硬盘等外存交换的数据；也就是说，内存分配是任务执行的重要部分，因此，在利用网络模型执行正向传播预测任务的过程中，内存分配也非常重要。

一般来说，在利用网络模型执行预测任务时，如果输入数据的尺寸不断变化，中间数据的尺寸也会不断发生变化，从而，所占用的内存空间也是不断变化的；此时，在为任务的执行分配内存空间时，通常是分配一块足够大的内存块，以满足最大尺寸的输入数据对应的任务的执行；然而，在上述每次任务的执行过程中，由于每次任务的执行都是利用这样一块足够大的内存块实现的，因此内存空间的占用量较大，造成了内存空间的浪费，从而，任务执行的内存消耗大。

发明内容

本申请实施例提供一种内存管理方法、装置、设备及计算机可读存储介质，能够降低任务执行的内存消耗。

本申请实施例的技术方案是这样实现的：

本申请实施例提供一种内存管理方法，包括：

获取目标任务请求；

响应所述目标任务请求，获取前次任务执行时所使用的前次内存块；

其中，所述前次任务为执行目标任务之前所执行的任务，所述前次内存块为所述前次任务执行时所使用的各个内存块构成的集合，内存块为内存分配的最小分配单元；

通过遍历所述前次内存块，确定出所述目标任务对应的目标内存块；

基于所述目标内存块，执行所述目标任务。

本申请实施例提供一种内存管理装置，包括：

请求获取模块，用于获取目标任务请求；

请求响应模块，用于响应所述目标任务请求，获取前次任务执行时所使用的前次内存块；其中，所述前次任务为执行目标任务之前所执行的任务，所述前次内存块为所述前次任务执行时所使用的各个内存块构成的集合，内存块为内存分配的最小分配单元；

内存分配模块，用于通过复用所述前次内存块，确定出所述目标任务对应的目标内存块；

任务执行模块，用于基于所述目标内存块，执行所述目标任务。

在本申请实施例中，所述内存分配模块，还用于获取所述目标任务对应的输入参数信息，所述输入参数信息为所述目标任务对应的输入数据；基于预设网络模型和所述输入参数信息，计算所述目标任务对应的输入张量信息集合；遍历所述前次内存块，确定当前输入张量信息的内存块分配信息，所述当前输入张量信息为所述输入张量信息集合中的任一输入张量信息；当完成对所述输入张量信息集合的处理时，将与所述输入张量信息集合对应的各个内存块分配信息组合为所述目标内存块。