[发明专利]数据处理方法、装置、电子设备以及介质在审
申请号: | 202011511725.9 | 申请日: | 2020-12-18 |
公开(公告)号: | CN112631955A | 公开(公告)日: | 2021-04-09 |
发明(设计)人: | 李建军;姚猛;王振江;周昱 | 申请(专利权)人: | 北京地平线机器人技术研发有限公司 |
主分类号: | G06F12/06 | 分类号: | G06F12/06;G06F12/0802;G06F12/10;G06N3/04 |
代理公司: | 北京众达德权知识产权代理有限公司 11570 | 代理人: | 李娇 |
地址: | 100086 北京市海淀*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 数据处理 方法 装置 电子设备 以及 介质 | ||
1.一种数据处理方法,所述方法包括:
当执行到神经网络模型的运算层时,基于预先存储的缓存分配关系,设置第一缓存的第一循环寻址地址范围,以及第二缓存的第二循环寻址地址范围,其中,所述第一缓存为所述运算层的输入数据对应的缓存,所述第二缓存为所述运算层的输出结果对应的缓存,所述缓存分配关系包括:所述运算层的特征数据对应的缓存以及在该缓存中对应的循环寻址范围,所述特征数据包括输入数据和输出结果;
基于所述第一循环寻址地址范围在所述第一缓存中进行循环寻址,读取所述运算层的输入数据;
执行所述运算层对应的操作指令,对所读取的输入数据进行处理,得到输出结果;
基于所述第二循环寻址地址范围在所述第二缓存中进行循环寻址,将所述输出结果写入所述第二缓存。
2.如权利要求1所述的方法,其中,当多个不同特征数据对应同一缓存时,所述多个不同特征数据在该缓存中对应的循环寻址范围互不重叠。
3.如权利要求1所述的方法,其中,所述方法还包括:
基于所述运算层的每个特征数据大小,为所述每个特征数据分配缓存,并确定所述每个特征数据在相应缓存中对应的循环寻址范围;
将所述每个特征数据与相应缓存以及循环寻址范围之间的对应关系,作为所述运算层对应的所述缓存分配关系。
4.如权利要求3所述的方法,其中,所述确定所述每个特征数据在相应缓存中对应的循环寻址范围,包括:
针对所述每个特征数据,确定所述特征数据在相应缓存中的起始地址,并确定对所述特征数据进行拆分处理后得到的目标数据块;
基于所述特征数据的起始地址以及所述目标数据块的大小,确定所述特征数据在相应缓存中对应的循环寻址范围。
5.如权利要求4所述的方法,其中,所述目标数据块为对所述特征数据进行拆分处理后得到的多个数据块中最大的数据块。
6.如权利要求1所述的方法,其中,所述基于预先存储的缓存分配关系,设置第一缓存的第一循环寻址地址范围,以及第二缓存的第二循环寻址地址范围,包括:
将所述缓存分配关系中,与所述运算层的输入数据对应的循环寻址范围存储到第一存储空间中,作为所述第一缓存的第一循环寻址地址范围,与所述运算层的输出结果对应的循环寻址范围存储到第二存储空间中,作为所述第二缓存的第二循环寻址地址范围,
其中,所述第一存储空间用于存储所述第一缓存的读缓存循环寻址地址范围,所述第二存储空间用于存储所述第二缓存的写缓存循环寻址地址范围。
7.一种数据处理装置,所述装置包括:
循环寻址设置模块,用于当执行到神经网络模型的运算层时,基于预先存储的缓存分配关系,设置第一缓存的第一循环寻址地址范围,以及第二缓存的第二循环寻址地址范围,其中,所述第一缓存为所述运算层的输入数据对应的缓存,所述第二缓存为所述运算层的输出结果对应的缓存,所述缓存分配关系包括:所述运算层的特征数据对应的缓存以及在该缓存中对应的循环寻址范围,所述特征数据包括输入数据和输出结果;
读取模块,用于基于所述第一循环寻址地址范围在所述第一缓存中进行循环寻址,读取所述运算层的输入数据;
处理模块,用于执行所述运算层对应的操作指令,对所读取的输入数据进行处理,得到输出结果;
写入模块,用于基于所述第二循环寻址地址范围在所述第二缓存中进行循环寻址。
8.一种电子设备,包括:
处理器,所述处理器包括运算单元以及一个或多个缓存;以及
存储器,在所述存储器中存储有计算机程序指令,所述计算机程序指令在被所述运算单元运行时使得所述运算单元执行如权利要求1-6中任一项所述的方法。
9.如权利要求8所述的电子设备,其中,所述处理器还包括分别对应于每个缓存设置的存储器件,每个缓存对应的存储器件均与所述运算单元连接,所述存储器件包括两组存储空间,其中一组存储空间用于存储相应缓存的读缓存循环寻址地址范围,另一组存储空间用于存储相应缓存的写缓存循环寻址地址范围。
10.一种计算机可读存储介质,所述存储介质存储有计算机程序,所述计算机程序用于执行上述权利要求1-6中任一所述的方法。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京地平线机器人技术研发有限公司,未经北京地平线机器人技术研发有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202011511725.9/1.html,转载请声明来源钻瓜专利网。