北京航空航天大学杨海龙获国家专利权
买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
龙图腾网获悉北京航空航天大学申请的专利一种面向张量加速器的模板计算代码生成方法及系统获国家发明授权专利权,本发明授权专利权由国家知识产权局授予,授权公告号为:CN119536701B 。
龙图腾网通过国家知识产权局官网在2025-09-12发布的发明授权授权公告中获悉:该发明授权的专利申请号/专利号为:202411676995.3,技术领域涉及:G06F8/30;该发明授权一种面向张量加速器的模板计算代码生成方法及系统是由杨海龙;卫林泉;刘笑妍;游心;钱德沛设计研发完成,并于2024-11-22向国家知识产权局提交的专利申请。
本一种面向张量加速器的模板计算代码生成方法及系统在说明书摘要公布了:本发明涉及一种面向张量加速器的模板计算代码生成方法及系统,其方法包括:S1:将模板(Stencil)计算转换为本发明所设计的领域特定编程语言;S2:将模板计算的权重矩阵进行行切分与列切分;S3:将切分后的权重行列进行堆叠重排形成适用于矩阵乘的权重矩阵;S4:启动自动调优代码生成器,通过进化搜索形成矩阵乘循环优化配置,并采用代价模型筛选候选配置;S5:将筛选出的候选配置用于真机实测性能,并将实测结果用于更新代价模型;S6:检查进化搜索迭代次数是否达到迭代次数阈值,如果是,则程序结束,选择最优配置,否则转至S4。本发明构建访存与计算的多级软流水,使得访存开销能够更好掩盖。
本发明授权一种面向张量加速器的模板计算代码生成方法及系统在权利要求书中公布了:1.一种面向张量加速器的模板计算代码生成方法,其特征在于,包括: 步骤S1:将模板计算转换为领域特定编程语言,所述领域特定编程语言包括:模板计算的基本公式和用户指定的循环优化配置;设置模板计算代码生成程序的总迭代次数为m次; 步骤S2:根据领域特定编程语言所表达的模板计算模式,将模板计算的权重矩阵进行行切分与列切分,提取模板计算权重矩阵中的非零元,当所述非零元位于一行中,则采用行切分,位于一列中,则采用列切分; 步骤S3:将切分后的权重行或列进行堆叠重排形成适用于矩阵乘的权重矩阵,选择适用于张量计算单元的矩阵乘形状,判断所述权重行或列能否在矩阵乘形状范围内完成堆叠重排,并对其进行重排转换; 步骤S4:启动自动调优代码生成器,设置迭代次数i=0,开始迭代搜索模板计算代码循环优化的最优配置,将所述重排转换后的模板计算进行循环优化,通过进化搜索生成矩阵乘循环优化配置,构建候选配置队列,并采用代价模型筛选候选配置; 步骤S5:将筛选出的候选配置用于真机实测,获取实测性能结果后,更新最优配置,并将实测结果用于更新代价模型,迭代次数i加1; 步骤S6:检查进化搜索迭代次数是否达到总迭代次数,如果此时i=m,则程序结束,并将当前最优配置返回,得到最优代码生成样本,否则,转至步骤S4。
如需购买、转让、实施、许可或投资类似专利技术,可联系本专利的申请人或专利权人北京航空航天大学,其通讯地址为:100191 北京市海淀区学院路37号;或者联系龙图腾网官方客服,联系龙图腾网可拨打电话0551-65771310或微信搜索“龙图腾网”。
以上内容由龙图腾AI智能生成。
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。

皖公网安备 34010402703815号
请提出您的宝贵建议,有机会获取IP积分或其他奖励