申请/专利权人:吴都
申请日:2023-08-18
公开(公告)日:2023-10-27
公开(公告)号:CN116954554A
主分类号:G06F7/523
分类号:G06F7/523;G06F17/15;G06F17/16
优先权:
专利状态码:在审-实质审查的生效
法律状态:2023.11.14#实质审查的生效;2023.10.27#公开
摘要:本发明公开了基于自动化搜索的加速卷积Im2col矩阵乘计算方法,包括如下步骤:1使用Im2col算法将卷积计算转换成矩阵乘法;2构造TVM用于搜索最优矩阵乘法的模板;3手工调优的小矩阵乘法库;4运行时自动代码生成的方法;本发明的有益效果如下:1本发明使用自动化搜索最优矩阵乘法的方式,从不同的参数组合中搜索到当前硬件架构和卷积形状下最优的参数组合,所以本发明可以针对不同的硬件架构和卷积形状生成性能较优的矩阵乘法代码,从而高效地完成卷积计算。2本发明使用类似运行时自动代码生成的方法调用经过专门调优的小矩阵乘法库,该方法可以解决长条形矩阵乘法性能较差的问题,得到性能较好的矩阵乘法的方式,从而高效地完成卷积计算。
主权项:1.基于自动化搜索的加速卷积Im2col矩阵乘计算方法,其特征在于,包括如下步骤:1使用Im2col算法将卷积计算转换成矩阵乘法;2构造TVM用于搜索最优矩阵乘法的模板;3构建手工调优的小矩阵乘法库;4运行时自动代码生成。
全文数据:
权利要求:
百度查询: 吴都 基于自动化搜索的加速卷积Im2col矩阵乘计算方法
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。