[发明专利]矩阵乘法计算方法和计算设备

权利要求书

1.一种矩阵乘法计算方法，其特征在于，包括：

提供多个矩阵乘法实现方案，所述矩阵乘法实现方案对应第一矩阵的行参数、列参数和第二矩阵的列参数；

确定第一目标矩阵的行参数、列参数和第二目标矩阵的列参数；

根据所述第一目标矩阵的行参数、列参数和第二目标矩阵的列参数，确定对应的目标矩阵乘法实现方案；

利用确定出的目标矩阵乘法实现方案计算所述第一目标矩阵和第二目标矩阵的乘积；

所述方法还包括：

当未查找到对应的矩阵乘法实现方案时，对多个可选实现方案进行评估，根据计算时间，从所述可选实现方案中选择优选实现方案；

所述对多个可选实现方案进行评估，根据计算时间，从所述可选实现方案中选择优选实现方案的步骤包括：

生成第一测试矩阵和第二测试矩阵，所述第一测试矩阵和所述第二测试矩阵具有与所述第一目标矩阵和所述第二目标矩阵相同的行列参数；

分别利用多个可选实现方案计算所述第一测试矩阵和所述第二测试矩阵的乘积，获得每一个可选实现方案的计算时间；其中，所述分别利用多个可选实现方案计算所述第一测试矩阵和所述第二测试矩阵的乘积，包括：遍历所有的可选实现方案，分别计算所述第一测试矩阵和所述第二测试矩阵的乘积；

根据计算时间，从所述可选实现方案中选择优选实现方案；所述优选实现方案为所述可选实现方案中计算时间最短的实现方案；

将所述优选实现方案上传至GPU；

将所述优选实现方案对应的识别ID以及所述优选实现方案对应的所述第一矩阵的行参数、列参数和所述第二矩阵的列参数写入查找表中；

所述方法还包括：

根据所述查找表获取所述矩阵乘法实现方案，并通过对应的识别ID获取并调用上传至GPU中的所述矩阵乘法实现方案；

在计算结束之后，回收所述查找表。

2.如权利要求1所述的方法，其特征在于，在根据所述第一目标矩阵的行参数、列参数和第二目标矩阵的列参数，确定对应的目标矩阵乘法实现方案的步骤之前，所述方法还包括：

为所述矩阵乘法实现方案分配识别ID；

将所述矩阵乘法实现方案的识别ID和该实现方案对应的第一矩阵的行参数、列参数和第二矩阵的列参数存入查找表。

3.如权利要求2所述的方法，其特征在于，将所述矩阵乘法实现方案的识别ID和该实现方案对应的第一矩阵的行参数、列参数和第二矩阵的列参数存入查找表的步骤包括：

将所述矩阵乘法实现方案的第一矩阵的行参数、列参数和第二矩阵的列参数生成键值；

将所述矩阵乘法实现方案中的识别ID和键值组成键值对，存储于所述查找表中。

4.如权利要求2所述的方法，其特征在于，所述根据所述第一目标矩阵的行参数、列参数和第二目标矩阵的列参数，确定对应的目标矩阵乘法实现方案的步骤包括：

在所述查找表中查找是否存在与所述第一矩阵的行参数、列参数和第二矩阵的列参数相同的参数的矩阵乘法实现方案。

5.如权利要求2所述的方法，其特征在于，在根据计算时间，从所述可选实现方案中选择优选实现方案的步骤之后，所述方法还包括：

将该优选实现方案的第一矩阵的行参数、列参数和第二矩阵的列参数生成键值；

将该优选实现方案中的识别ID和键值组成键值对，存储于所述查找表中。