公开(公告)号：US20230236888A1

公开(公告)日：2023-07-27

申请号：US18127300

申请日：2023-03-28

Applicant: Huawei Technologies Co., Ltd.

Inventor： Zhen ZHANG ,  Ioannis LAMPROU ,  Javier DE JUAN ,  Chang LIU

IPC: G06F9/50

CPC classification number: G06F9/5016 ,  G06F9/5033

Abstract: This application provides a memory allocation method. The method includes: obtaining a computation graph corresponding to a neural network; sequentially allocating memory space to M pieces of tensor data based on a sorting result of the M pieces of tensor data, where if at least a part of the allocated memory space can be reused for one of the M pieces of tensor data, the at least a part of the memory space that can be reused for the tensor data is allocated to the tensor data, the allocated memory space is memory space that has been allocated to the M pieces of tensor data before the tensor data, the sorting result indicates a sequence of allocating memory space to the M pieces of tensor data, and the sorting result is related to information about each of the M pieces of tensor data.