公开(公告)号：US20230021472A1

公开(公告)日：2023-01-26

申请号：US17954695

申请日：2022-09-28

Applicant: HUAWEI TECHNOLOGIES CO., LTD.

Inventor： Anna BULANOVA ,  Jessica DAVIES ,  Xiong GAO

IPC: G06F30/398

Abstract: A method for optimizing a layout of a tensor memory defines at least one hard constraint for allocating a plurality of input/output (I/O) vectors for reading and writing data for a task in the tensor memory. The at least one hard constraint is applied to determine one or more potential conflicts between the plurality of I/O vectors. One or more soft constraints aimed at mitigating the one or more potential conflicts between the I/O vectors may also be generated. The at least one hard constraint is applied in a maximum satisfiability (MaxSAT) solver. The one or more soft constraints may also be applied in the MaxSAT solver. The MaxSAT solver determines locations of the data in the tensor memory. The starting addresses of the input data to be read and of output data to be written by each of the I/O vectors are updated in the tensor memory.