[发明专利]一种矩阵乘法的计算方法及装置

摘要

本申请提出一种矩阵乘法器。由于现有的矩阵乘法器所包含的全连接网络占用芯片空间较大，且进行矩阵乘法计算时需要进行大量的存储访问，导致流多处理器进行矩阵乘法计算效率较低。基于提高图形处理器进行矩阵乘法计算的效率的目的，本申请所提出的矩阵乘法器在进行矩阵乘法时，利用不同组的存储库可以同时访问的特性，每次将作为被乘数的矩阵的一行元素以及作为乘数的矩阵的一列元素加载到相应的计算单元中，同时进行计算。通过使用这种矩阵乘法器，可以减少完成矩阵乘法计算所需要的步骤，降低了所需进行的存储访问的次数，从而提高了图形处理器进行矩阵乘法计算的效率。

主权项

1.一种矩阵乘法器，其特征在于，所述矩阵乘法器包括：N*N个计算单元，所述N*N个计算单元组成N*N的矩阵，N为大于等于2的正整数；两个存储库集合，每个存储库集合包括N个存储库，第一存储库集合用于存储输入矩阵中的第一乘法矩阵，第二存储库集合用于存储输入矩阵中的第二乘法矩阵，所述第一存储库集合中第M个存储库与所述N*N的矩阵中第M行的每个计算单元相连接，所述第二存储库集合中第M个存储库与所述N*N的矩阵中第M列的每个计算单元相连接，其中，M为变量，取值为1≤M≤N；在每个时钟周期，所述N*N的矩阵中的每一行的每个计算单元用于接收与自身相连接的所述第一存储库集合中的存储库所广播的第一输入数据，所述N*N的矩阵中的每一列的每个计算单元用于接收与自身相连接的所述第二存储库集合中的存储库所广播的第二输入数据；在每个时钟周期，所述N*N的矩阵中的每个计算单元根据接收到的所述第一输入数据和所述第二输入数据进行乘法计算；在第N个时钟周期结束后，所述矩阵乘法器完成所述第一乘法矩阵与所述第二乘法矩阵的乘法运算。