Alibaba Group Holding Limited
DATA PROCESSING METHOD AND APPARATUS APPLIED TO GRAPHICS PROCESSING UNIT, AND ELECTRONIC DEVICE

Abstract:

This application provides a data processing method applied to a GPU, including: reading a to-be-processed matrix from memory corresponding to a target GPU into registers in a target streaming multiprocessor, where the target streaming multiprocessor is a streaming multiprocessor adapted to perform a matrix computation on the to-be-processed matrix; in a process of reading the to-be-processed matrix from the registers into shared memory corresponding to the registers, performing a first preset operation on the to-be-processed matrix to obtain an initial matrix corresponding to the to-be-processed matrix; and in a process of reading the initial matrix from the shared memory into the registers, performing a second preset operation on the initial matrix to obtain a target matrix meeting a matrix multiplication operation requirement of a matrix-dedicated computation unit. The data processing method applied to a graphics processing unit (Graphics Processing Unit, GPU) can transform a to-be-processed matrix into a target matrix meeting a matrix multiplication operation requirement of the GPU, thereby improving applicability of the GPU for matrix multiplication operations.