返回搜索
档案
基于昇腾芯片的矩阵乘法计算方法、装置、设备及介质
发明专利有效
10权利要求 · 4 独立
§ Ⅰ
卷宗概要
发明人
徐柏松; 徐鹏翔; 范登栋; 杨凯; 刘勇翔; 许芬; 田永鸿
IPC 分类
G06F 9/38 (2018.01)G06F 9/30 (2018.01)G06F 9/345 (2018.01)G06F 9/355 (2018.01)G06F 17/16 (2006.01)G06F 7/544 (2006.01)
本申请实施例提供了一种基于昇腾芯片的矩阵乘法计算方法、装置、设备及介质。方法包括:通过多个向量计算核获取矩阵乘法描述信息,并对矩阵乘法描述信息包含的第一矩阵和第二矩阵进行补齐处理,得到目标第一矩阵和目标第二矩阵;获取多个第一矩阵分块和多个第二矩阵分块;针对每个第一矩阵分块与对应的第二矩阵分块构成的矩阵分块对,通过矩阵计算核计算得到矩阵计算子结果;通过对应分配的向量计算核对标量系数和矩阵计算子结果进行向量计算,得到每个矩阵分块对的向量计算子结果;基于多个向量计算核计算得到的多个向量计算子结果进行融合,得到矩阵乘法描述信息对应的矩阵乘法计算结果。以此,能够提高矩阵乘法计算的效率和准确性。