[发明专利]一种提高GEMM计算性能的方法及装置有效
申请号: | 201710795580.1 | 申请日: | 2017-09-06 |
公开(公告)号: | CN109460533B | 公开(公告)日: | 2021-10-26 |
发明(设计)人: | 齐霁;张邵敏;贾海鹏 | 申请(专利权)人: | 华为技术有限公司 |
主分类号: | G06F17/16 | 分类号: | G06F17/16;G06N3/04 |
代理公司: | 深圳市深佳知识产权代理事务所(普通合伙) 44285 | 代理人: | 王仲凯 |
地址: | 518129 广东*** | 国省代码: | 广东;44 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本申请提供了一种提高GEMM计算性能的方法及装置,获取待优化的通用矩阵‑矩阵乘GEMM计算的参数,从至少一个历史的GEMM计算的参数中查询目标参数,目标参数为与待优化的GEMM计算的参数满足预设关系的参数。依据参数与优化方法的预设对应关系,确定所述目标参数对应的优化方法。并使用所述目标参数对应的优化方法,优化所述待优化的GEMM计算。其中,GEMM计算的参数基于参与GEMM计算的矩阵的大小确定。因为将参与待优化的GEMM计算的矩阵的特征作为优化待优化的GEMM计算的依据,所以,在使用深度卷积神经网络对目标进行检测的过程中,即使矩阵的规模小或者形状不规则,也能提高GEMM计算的性能。 | ||
搜索关键词: | 一种 提高 gemm 计算 性能 方法 装置 | ||
【主权项】:
1.一种提高GEMM计算性能的方法,其特征在于,包括:获取待优化的通用矩阵‑矩阵乘GEMM计算的参数,所述GEMM计算为第一矩阵乘以第二矩阵的计算,所述GEMM计算的参数基于矩阵参数确定,所述矩阵参数包括矩阵规模参数和/或矩阵形状参数集,所述矩阵规模参数为所述第一矩阵的行数、所述第一矩阵的列数和所述第二矩阵的列数的最大公约数,所述矩阵形状参数集中包括:第一参数、第二参数和第三参数,所述第一参数为所述第一矩阵的行数与第一数值的比值,所述第二参数为所述第一矩阵的列数与所述第一数值的比值,所述第三参数为所述第二矩阵的列数与所述第一数值的比值,所述第一数值为所述矩阵规模参数或1;从参考的GEMM计算的参数中查询目标参数,所述参考的GEMM计算包括至少一个历史的GEMM计算,所述目标参数为与所述待优化的GEMM计算的参数满足预设关系的参数;依据参数与优化方法的预设对应关系,确定所述目标参数对应的优化方法;使用所述目标参数对应的优化方法,优化所述待优化的GEMM计算。
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于华为技术有限公司,未经华为技术有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/201710795580.1/,转载请声明来源钻瓜专利网。
- 上一篇:一种太阳直接辐射遥感计算方法和装置
- 下一篇:空间直角坐标转换参数计算方法