一种基于AI处理器的通用矩阵乘算子的处理方法和装置

摘要

本发明涉及矩阵处理技术领域，具体是涉及一种基于AI处理器的通用矩阵乘算子的处理方法和装置。依据原矩阵中的设定元素，在CPU上计算出输入矩阵；将所述输入矩阵搬运至所述全局内存；将搬运至所述全局内存的输入矩阵和位于所述全局内存上的矩阵作为通用矩阵乘算子的输入，得到所述通用矩阵乘算子的输出矩阵。本发明在AI处理器上通过通用矩阵乘算子处理矩阵，能够使得AI处理器的计算资源得到充分利用，从而提高了处理矩阵的速度和效率。