Meta Platforms, Inc.
HARDWARE ACCELERATED MATRIX MANIPULATION OPERATIONS USING PROCESSOR INSTRUCTIONS

Abstract:

A processor system comprises a shared memory and a processing element. The processing element includes a matrix processor unit and is in communication with the shared memory. The processing element is configured to receive a processor instruction specifying a data matrix and a matrix manipulation operation. A manipulation matrix based on the processor instruction is identified. The data matrix and the manipulation matrix are loaded into the matrix processor unit and a matrix operation is performed to determine a result matrix. The result matrix is outputted to a destination location.