Systolic array-based data processing method and apparatus, medium, and program product

An electronic device, comprising:

• one or more processors;
• and
• memory storing one or more programs, the one or more programs comprising instructions that, when executed by the one or more processors, cause the one or more processors to perform operations comprising:
• determining a standard input depth and a standard output depth of a systolic array, and determining a target input depth and a target output depth of a shallow depth model;
• determining an input splice quantity for the systolic array based on the target input depth and the standard input depth, and determining an output splice quantity for the systolic array based on the target output depth and the standard output depth;
• generating input data matching the input splice quantity, each piece of input data having the target input depth, inputting the input data matching the input splice quantity to an input buffer of the systolic array in batches, without overlaps in the input data, and processing, by the systolic array, the input data in the input buffer to generate output data corresponding to each piece of input data, each piece of output data having the target output depth;
• and
• in accordance with a determination that a quantity of output data received by an output buffer of the systolic array from the systolic array matches the output splice quantity, outputting, in the output buffer, output data having a quantity matching the output splice quantity in batches.

A non-transitory computer-readable storage medium, storing a computer program, the computer program, when executed by one or more processors of an electronic device, cause the one or more processors to perform operations comprising:

• determining a standard input depth and a standard output depth of a systolic array, and determining a target input depth and a target output depth of a shallow depth model;
• determining an input splice quantity for the systolic array based on the target input depth and the standard input depth, and determining an output splice quantity for the systolic array based on the target output depth and the standard output depth;
• generating input data matching the input splice quantity, each piece of input data having the target input depth, inputting the input data matching the input splice quantity to an input buffer of the systolic array in batches, without overlaps in the input data, and processing, by the systolic array, the input data in the input buffer to generate output data corresponding to each piece of input data, each piece of output data having the target output depth;
• and
• in accordance with a determination that a quantity of output data received by an output buffer of the systolic array from the systolic array matches the output splice quantity, outputting, in the output buffer, output data having a quantity matching the output splice quantity in batches.