[发明专利]一种基于硬件计算图的深度学习编译器在审
申请号: | 202210625568.7 | 申请日: | 2022-06-02 |
公开(公告)号: | CN114936015A | 公开(公告)日: | 2022-08-23 |
发明(设计)人: | 李武军;李俊;王炜 | 申请(专利权)人: | 南京大学 |
主分类号: | G06F8/30 | 分类号: | G06F8/30;G06F8/41;G06N3/10;G06N3/04;G06N3/08 |
代理公司: | 南京乐羽知行专利代理事务所(普通合伙) 32326 | 代理人: | 李玉平 |
地址: | 210023 江苏*** | 国省代码: | 江苏;32 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本发明公开了一种基于硬件计算图的深度学习编译器,包括软件计算图生成模块,硬件计算图生成模块,软件计算图优化模块,硬件计算图优化模块,以及代码生成模块;软件计算图生成模块用于将外部框架定义的深度学习模型转化为内部软件计算图;所述硬件计算图生成模块,在软件计算图的指导下,先生成基础节点集合,再使用贪心搜索对节点集合调整直到满足整个深度学习模型所需的计算要求,最后生成硬件计算图的边;代码生成模块用于生成深度学习模型对应的硬件代码以及仿真代码。本发明可以对复杂拓扑的模型进行高效,且高性能的部署,减少了部署所需的成本。并且本发明能够应用于CPU、GPU、FPGA以及人工智能专用芯片等多种硬件平台。 | ||
搜索关键词: | 一种 基于 硬件 计算 深度 学习 编译器 | ||
【主权项】:
暂无信息
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于南京大学,未经南京大学许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/202210625568.7/,转载请声明来源钻瓜专利网。