[发明专利]支持机器学习训练的专用后端代码生成方法及装置在审
| 申请号： | 202111071061.3 | 申请日： | 2021-09-13 |
| 公开（公告）号： | CN113885871A | 公开（公告）日： | 2022-01-04 |
| 发明（设计）人： | 姚海龙;曾军;李啸宇;寇明阳 | 申请（专利权）人： | 清华大学 |
| 主分类号： | G06F8/41 | 分类号： | G06F8/41;G06F8/30;G06N3/04;G06N3/08;G06N5/04;G06T1/20 |
| 代理公司： | 北京路浩知识产权代理有限公司 11002 | 代理人： | 杨明月 |
| 地址： | 100084 北*** | 国省代码： | 北京;11 |
| 权利要求书： | 查看更多 | 说明书： | 查看更多 |
| 摘要： | 本发明提供一种支持机器学习训练的专用后端代码生成方法及装置，所述方法包括：对所述第一计算图进行解析与优化处理，得到第二计算图；基于所述至少一个算子的算子类型将所述至少一个算子分发至算子映射模块，输出每个算子对应的算子调用代码；将所述至少一个算子的内存配置信息输入内存管理模块，输出内存管理代码；基于所述每个算子对应的算子调用代码与所述内存管理代码生成目标后端代码，基于算子库对所述目标后端代码进行编译处理，得到部署文件。本发明实现了一个端到端且支持神经网络训练的编译器，提高了GPU应用于神经网络训练与推理中的运算效率，扩展了GPU的应用范围。 | ||
| 搜索关键词： | 支持 机器 学习 训练 专用 后端 代码 生成 方法 装置 | ||
【主权项】：
暂无信息
下载完整专利技术内容需要扣除积分，VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息，商用须获得专利权人授权。该专利全部权利属于清华大学，未经清华大学许可，擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作，请联系【客服】
本文链接：http://www.vipzhuanli.com/patent/202111071061.3/，转载请声明来源钻瓜专利网。
- 上一篇：富氧燃烧送风系统控制方法
- 下一篇：一种出茶方法
