恭喜清华大学姚海龙获国家专利权
买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
龙图腾网恭喜清华大学申请的专利支持机器学习训练的专用后端代码生成方法及装置获国家发明授权专利权,本发明授权专利权由国家知识产权局授予,授权公告号为:CN113885871B 。
龙图腾网通过国家知识产权局官网在2025-03-28发布的发明授权授权公告中获悉:该发明授权的专利申请号/专利号为:202111071061.3,技术领域涉及:G06F8/41;该发明授权支持机器学习训练的专用后端代码生成方法及装置是由姚海龙;曾军;李啸宇;寇明阳设计研发完成,并于2021-09-13向国家知识产权局提交的专利申请。
本支持机器学习训练的专用后端代码生成方法及装置在说明书摘要公布了:本发明提供一种支持机器学习训练的专用后端代码生成方法及装置,所述方法包括:对所述第一计算图进行解析与优化处理,得到第二计算图;基于所述至少一个算子的算子类型将所述至少一个算子分发至算子映射模块,输出每个算子对应的算子调用代码;将所述至少一个算子的内存配置信息输入内存管理模块,输出内存管理代码;基于所述每个算子对应的算子调用代码与所述内存管理代码生成目标后端代码,基于算子库对所述目标后端代码进行编译处理,得到部署文件。本发明实现了一个端到端且支持神经网络训练的编译器,提高了GPU应用于神经网络训练与推理中的运算效率,扩展了GPU的应用范围。
本发明授权支持机器学习训练的专用后端代码生成方法及装置在权利要求书中公布了:1.一种支持机器学习训练的专用后端代码生成方法,其特征在于,包括:获取第一计算图,对所述第一计算图进行解析与优化处理,得到第二计算图,所述第一计算图为至少一个算子按照第一顺序组成的神经网络模型;获取第二计算图的至少一个算子,基于所述至少一个算子的算子类型将所述至少一个算子分发至算子映射模块,输出每个算子对应的算子调用代码;获取所述至少一个算子的内存配置信息,将所述至少一个算子的内存配置信息输入内存管理模块,输出内存管理代码;基于所述每个算子对应的算子调用代码与所述内存管理代码生成目标后端代码,基于算子库对所述目标后端代码进行编译处理,得到部署文件,所述部署文件用于执行所述第一计算图的计算任务;所述内存管理代码用于:基于所述至少一个算子的内存配置信息获取内存池;获取所述至少一个算子的输入输出内存请求,基于所述至少一个算子的输入输出内存请求对所述内存池进行分配;所述内存管理代码还用于:获取所述至少一个算子的输入输出内存请求对应的内存信息;基于所述内存信息获取所述至少一个算子对应的权重参数,基于权重参数更新机制对所述权重参数进行更新处理。
如需购买、转让、实施、许可或投资类似专利技术,可联系本专利的申请人或专利权人清华大学,其通讯地址为:100084 北京市海淀区双清路30号清华大学;或者联系龙图腾网官方客服,联系龙图腾网可拨打电话0551-65771310或微信搜索“龙图腾网”。
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。