[发明专利]一种神经网络的编译方法、装置、计算机设备及存储介质在审
申请号: | 202111652713.2 | 申请日: | 2021-12-30 |
公开(公告)号: | CN114356340A | 公开(公告)日: | 2022-04-15 |
发明(设计)人: | 周桓;田志仲;何山;勾志宏 | 申请(专利权)人: | 上海阵量智能科技有限公司 |
主分类号: | G06F8/41 | 分类号: | G06F8/41;G06N3/04 |
代理公司: | 北京中知恒瑞知识产权代理事务所(普通合伙) 11889 | 代理人: | 袁忠林 |
地址: | 200235 上海*** | 国省代码: | 上海;31 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 一种 神经网络 编译 方法 装置 计算机 设备 存储 介质 | ||
1.一种神经网络的编译方法,其特征在于,包括:
对中间程序进行解析,得到所述中间程序的内部数据结构;所述内部数据结构包括对象以及对象之间的关联关系;所述对象包括:操作语句、以及与所述操作语句对应的变量和变量定义语句;
其中,所述中间程序是利用预设中间语言和目标高级语言之间的转换关系信息,将基于目标高级语言编写的目标神经网络的原始程序转换为预设中间语言编写的程序;
基于所述内部数据结构,生成所述操作语句的配置信息;
基于所述操作语句对数据处理芯片的指令封装关系、以及所述配置信息,生成所述数据处理芯片在运行所述目标神经网络时的机器指令。
2.根据权利要求1所述的编译方法,其特征在于,所述操作语句的配置信息包括:与所述操作语句对应的变量的目标存储地址;
所述基于所述内部数据结构,生成每条所述操作语句的配置信息,包括:
针对所述操作语句中的目标操作语句,基于所述目标操作语句对应变量的变量定义语句,为所述目标操作语句对应的变量分配对应的目标存储地址;所述目标存储地址包括内存中的存储地址、和/或高速缓存中的存储地址。
3.根据权利要求2所述的编译方法,其特征在于,所述基于所述内部数据结构,生成每条所述操作语句的配置信息之前,还包括:
基于所述操作语句的类型、以及所述操作语句对应的变量,从所述操作语句中,确定对预设存储空间进行重复访存的待删除操作语句;
从所述操作语句中删除所述待删除操作语句,得到所述目标操作语句。
4.根据权利要求3所述的编译方法,其特征在于,所述操作语句的类型包括:变量存储、以及变量读取;
所述基于所述操作语句的类型、以及所述操作语句对应的变量,从所述操作语句中,确定对预设存储空间进行重复访存的待删除操作语句,包括:
针对每个类型为变量存储的第一操作语句,基于所述第一操作语句对应的变量,确定是否存在与所述第一操作语句对应的变量为同一变量、且类型为变量读取的第二操作语句;其中,所述第一操作语句和所述第二操作语句分别属于相邻的不同网络层;
若存在,则将所述第一操作语句与对应的第二操作语句确定为所述待删除操作语句。
5.根据权利要求2-4任一项所述的编译方法,其特征在于,所述变量包括:各个网络层分别对应的输入变量、输出变量、以及中间变量;
所述基于所述目标操作语句对应变量的变量定义语句,为所述目标操作语句对应的变量分配对应的目标存储地址,包括:
响应于所述目标操作语句对应变量的变量定义语句指示所述目标操作语句对应变量为输入变量,为所述目标操作语句对应的变量分配内存、以及高速缓存中的存储地址;
响应于所述目标操作语句对应变量的变量定义语句指示所述目标操作语句对应变量为输出变量,为所述目标操作语句对应的变量分配内存、以及高速缓存中的存储地址;
响应于所述目标操作语句对应变量的变量定义语句指示所述目标操作语句对应变量为中间变量,为所述目标操作语句对应的变量分配高速缓存中的存储地址。
6.根据权利要求1-5任一项所述的编译方法,其特征在于,所述基于所述操作语句对数据处理芯片的指令封装关系、以及所述配置信息,生成所述数据处理芯片在运行所述目标神经网络时的机器指令之前,还包括:
针对每条操作语句,根据所述每条操作语句和其他操作语句之间的关联关系,为所述每条操作语句添加等待信息;所述等待信息用于指示所述每条操作指令的执行顺序。
7.根据权利要求6所述的编译方法,其特征在于,所述针对每条操作语句,根据所述每条操作语句和其他操作语句之间的关联关系,为所述每条操作语句添加等待信息之后,还包括:
基于操作语句、以及为所述操作语句添加的等待信息,生成第一调试信息;
所述第一调试信息用于对所述目标神经网络的原始程序进行调试。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于上海阵量智能科技有限公司,未经上海阵量智能科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202111652713.2/1.html,转载请声明来源钻瓜专利网。