首页 专利交易 科技果 科技人才 科技服务 商标交易 会员权益 IP管家助手 需求市场 关于龙图腾
 /  免费注册
到顶部 到底部
清空 搜索

执行神经网络模型的PIM架构的数据布局优化 

申请/专利权人:阿里巴巴集团控股有限公司

申请日:2021-01-14

公开(公告)日:2024-04-05

公开(公告)号:CN113139648B

主分类号:G06N3/063

分类号:G06N3/063;G06F17/16;G06F17/15;G06F7/523;G06F3/06

优先权:["20200117 US 16/746,419"]

专利状态码:有效-授权

法律状态:2024.04.05#授权;2021.08.06#实质审查的生效;2021.07.20#公开

摘要:本公开涉及一种执行神经网络模型的PIM架构的数据布局优化的技术方案。用于在包括存储块组件的PIM使能设备上布置计算图的方法。该方法包括将计算图的第一节点分配到存储块组件中的存储块的第一阵列的第一存储块上,以及将计算图的第二节点分配到存储块组件中的存储块的第二阵列的第二存储块上,其中,所述第一节点的输出数据用于提供给所述第二节点使用。所述存储块组件配置为支持经由所述存储块组件中的内部数据耦合从所述第一存储块到所述第二存储块的数据传输。

主权项:1.一种在包含存储块组件的PIM使能设备上布置计算图的方法,包括:将计算图的第一节点分配到存储块组件中的存储块的第一阵列的第一存储块上;以及,将计算图的第二节点分配到存储块组件中的存储块的第二阵列的第二存储块上,其中,所述第一节点的输出数据用于提供给所述第二节点使用;其中,所述存储块组件配置为支持经由所述存储块组件中的内部数据耦合从所述第一存储块到所述第二存储块的数据传输;基于第一节点和第二节点之间的依赖关系,为第一节点分配第一层编号,为第二节点分配第二层编号,其中第二层编号大于第一层编号;将第二节点重新分配到存储块组件中的存储块的第三阵列的第三存储块上;其中,所述存储块组件配置为支持通过由所述存储块组件的所有存储块共享的全局缓冲区从第一存储块到第三存储块的数据传输,并且来自包括第二节点且被分配给存储块第二阵列的一个或多个节点的输出边的数目,大于来自包括第一节点且被分配给存储块第一阵列的一个或多个节点的输出边的数目。

全文数据:

权利要求:

百度查询: 阿里巴巴集团控股有限公司 执行神经网络模型的PIM架构的数据布局优化

免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。