[发明专利]子矩阵运算装置及方法

权利要求书

1.一种子矩阵运算方法，其特征在于，所述方法包括如下步骤：

获取子矩阵运算指令，其中，所述子矩阵运算指令包括张量运算指令、子矩阵乘向量指令、向量乘子矩阵指令、子矩阵加法指令、子矩阵减法指令以及子矩阵乘法指令中的至少一种；

根据所述子矩阵运算指令分别从寄存器单元中获取第一子矩阵信息和第二子矩阵信息，所述第一子矩阵信息包括所述第一子矩阵数据在存储单元中的起始地址、所述第一子矩阵数据的行宽、所述第一子矩阵数据的列宽以及所述第一子矩阵数据的行间隔，其中，所述第一子矩阵数据的行间隔是指所述第一子矩阵数据相邻两行间，上一行的行末数据到下一行的行首数据的数据间隔；

根据所述第一子矩阵信息从存储单元中获取第一子矩阵数据，根据所述第二子矩阵信息从所述存储单元中获取第二子矩阵数据，第一子矩阵为二维矩阵的二维子矩阵，所述根据所述第一子矩阵信息从存储单元中获取第一子矩阵数据，包括：从所述第一子矩阵数据在存储单元中的起始地址开始，每读取所述第一子矩阵数据的行宽个数据后跳过所述第一子矩阵数据的行间隔个数据再读取所述第一子矩阵数据的行宽个数据，重复所述第一子矩阵数据的列宽次，得到所述第一子矩阵数据；

根据所述第一子矩阵数据和所述第二子矩阵数据进行子矩阵运算，获得子矩阵运算结果。

2.根据权利要求1所述的方法，其特征在于，所述第二子矩阵信息包括向量地址及向量长度；

根据所述第一子矩阵数据和所述第二子矩阵数据进行子矩阵运算，获得子矩阵运算结果的步骤包括：

将所述第一子矩阵数据作为被乘数，将所述第二子矩阵数据作为乘数进行子矩阵乘向量运算，获得子矩阵乘向量运算结果；

或者，将所述第一子矩阵数据作为乘数，将所述第二子矩阵数据作为被乘数进行向量乘子矩阵运算，获得向量乘子矩阵运算结果。

3.根据权利要求1所述的方法，其特征在于，所述第二子矩阵信息包括所述第二子矩阵数据在所述存储单元中的起始地址、所述第二子矩阵数据的行宽、所述第二子矩阵数据的列宽以及所述第二子矩阵数据的行间隔，其中，所述第二子矩阵数据的行间隔是指所述第二子矩阵数据相邻两行间，上一行的行末数据到下一行的行首数据的数据间隔；

根据所述第一子矩阵数据和所述第二子矩阵数据进行子矩阵运算，获得子矩阵运算结果的步骤包括：

根据所述第一子矩阵数据和所述第二子矩阵数据进行矩阵加法运算或减法运算。

4.根据权利要求1所述的方法，其特征在于，所述第二子矩阵信息包括所述第二子矩阵数据在所述存储单元中的起始地址、所述第二子矩阵数据的行宽、所述第二子矩阵数据的列宽及所述第二子矩阵数据的行间隔，其中，所述第二子矩阵数据的行间隔是指所述第二子矩阵数据相邻两行间，上一行的行末数据到下一行的行首数据的数据间隔；

根据所述第一子矩阵数据和所述第二子矩阵数据进行子矩阵运算，获得子矩阵运算结果的步骤包括：

根据所述第一子矩阵数据和所述第二子矩阵数据进行对位乘法运算，获得子矩阵乘法运算结果。

5.根据权利要求1所述的方法，其特征在于，所述第二子矩阵信息包括所述第二子矩阵数据在所述存储单元中的起始地址、所述第二子矩阵数据的行宽、所述第二子矩阵数据的列宽及所述第二子矩阵数据的行间隔，其中，所述第二子矩阵数据的行间隔是指所述第二子矩阵数据相邻两行之间，上一行的行末数据到下一行的行首数据的数据间隔；

根据所述第一子矩阵数据和所述第二子矩阵数据进行子矩阵运算，获得子矩阵运算结果的步骤包括：

根据所述第一子矩阵数据和所述第二子矩阵数据进行张量运算，获得张量运算结果。

6.根据权利要求1-5任一项所述的子矩阵运算方法，其特征在于，所述子矩阵运算指令包括操作码和至少一个操作域，其中，所述操作码用于指示所述子矩阵运算指令的功能，操作域用于指示所述子矩阵运算指令的数据信息；

所述子矩阵运算指令的数据信息包括所述寄存器单元的编号，从而能够根据寄存器单元的编号访问对应的寄存器单元，获取所述第一子矩阵信息和所述第二子矩阵信息。