- 专利标题: 基于Tensor访问的深度学习内存管理方法及系统
-
申请号: CN202011619848.4申请日: 2020-12-31
-
公开(公告)号: CN112306697B公开(公告)日: 2021-04-27
- 发明人: 何水兵 , 陈帅犇 , 陈平 , 杨斯凌 , 陈伟剑 , 孙贤和 , 陈刚 , 银燕龙 , 毛旷
- 申请人: 之江实验室 , 浙江大学
- 申请人地址: 浙江省杭州市余杭区文一西路1818号人工智能小镇10号楼;
- 专利权人: 之江实验室,浙江大学
- 当前专利权人: 之江实验室,浙江大学
- 当前专利权人地址: 浙江省杭州市余杭区文一西路1818号人工智能小镇10号楼;
- 代理机构: 杭州求是专利事务所有限公司
- 代理商 邱启旺
- 主分类号: G06F9/50
- IPC分类号: G06F9/50 ; G06N3/04 ; G06N3/08
摘要:
本发明提供了一种基于Tensor访问的深度学习内存管理方法,该方法通过收集神经网络的执行信息和硬件平台的性能信息获得相关决策下的内存空间开销和时间开销,并建立整数线性规划模型,通过在约束条件下优化求解最优的Tensor调度策略,从而解决内存不足问题的同时获得较高的深度学习训练性能。相比于现有技术,相同的硬件性能下,本发明可以实现更大的batchsize的神经网络训练。本发明同时还提出了一种内存管理系统,包括profile模块、决策模块和执行模块;该系统可直接添加在深度学习框架上,使用方便。
公开/授权文献
- CN112306697A 基于Tensor访问的深度学习内存管理方法及系统 公开/授权日:2021-02-02