Element-wise multiplication augmented general matrix multiplication computing core

Abstract

Systems and techniques are described herein for data processing. For example, a computing device can process, using a computing core of the at least one processor, a plurality of data matrices to generate a first matrix representation and a second matrix representation of element-wise multiplication of two or more matrices of the plurality of data matrices. The computing device can store the first matrix representation and the second matrix representation in local memory of the at least one processor. The computing device can process, using the computing core, the first matrix representation and the second matrix representation using a general matrix multiplication (GEMM) kernel to generate an augmented matrix associated with matrix multiplication of the first matrix representation and the second matrix representation.

Images