[发明专利]一种面向神经网络计算的内存优化方法和装置有效
| 申请号： | 202211177786.5 | 申请日： | 2022-09-27 |
| 公开（公告）号： | CN115269205B | 公开（公告）日： | 2022-12-27 |
| 发明（设计）人： | 王宏升;陈光 | 申请（专利权）人： | 之江实验室 |
| 主分类号： | G06F9/50 | 分类号： | G06F9/50;G06F12/02;G06N3/04 |
| 代理公司： | 北京志霖恒远知识产权代理有限公司 11435 | 代理人： | 戴莉 |
| 地址： | 311121 浙江*** | 国省代码： | 浙江;33 |
| 权利要求书： | 查看更多 | 说明书： | 查看更多 |
| 摘要： | 本发明公开了一种面向神经网络计算的内存优化方法和装置，包括以下步骤：步骤S1：将计算图重构为拓扑结构计算图；步骤S2：构建关于张量变量的生命周期区间；步骤S3：构建关于生命周期区间的扫描线；步骤S4：将张量变量分配到空闲寄存器；步骤S5：分配给超出寄存器需求数量的张量变量；步骤S6：将已过期的所述生命周期区间所分配的寄存器分配给超出寄存器需求数量的张量变量；步骤S7：将转移到内存中的张量变量添加回处于激活状态的所述生命周期区间并为其分配空闲寄存器。本发明优化用于神经网络计算的计算图的数据流的内存，减少数据流中张量变量所需的内存开销，降低大模型对于硬件内存资源的要求。 | ||
| 搜索关键词： | 一种 面向 神经网络 计算 内存 优化 方法 装置 | ||
【主权项】：
暂无信息
下载完整专利技术内容需要扣除积分，VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息，商用须获得专利权人授权。该专利全部权利属于之江实验室，未经之江实验室许可，擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作，请联系【客服】
本文链接：http://www.vipzhuanli.com/patent/202211177786.5/，转载请声明来源钻瓜专利网。
