一种基于多级存储芯片的计算图的编译优化方法和装置

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
一种基于多级存储芯片的计算图的编译优化方法和装置
申请号:CN202510300602
申请日期:2025-03-14
公开号:CN119806543B
公开日期:2025-06-10
类型:发明专利
摘要
发明公开了基于多级存储芯片的计算图的编译优化方法和装置。该方法包括:对运行在多级存储芯片上的计算图进行筛选获取候选算子节点;获取当前各候选算子节点的动态类型;根据动态类型对计算图中的候选算子节点进行前向搜索和后向搜索获取融合算子节点,并根据融合算子节点对计算图进行更新获取优化计算图,通过从计算图中筛选出符合融合条件的候选算子节点,以及各候选算子节点的动态类型,并根据动态类型对候选算子节点进行搜索获取融合算子节点,由于位于同一融合算子节点中的候选算子节点仅在共享存储层进行张量搬运,从而避免了在不同存储层之间张量数据的搬运次数,提高了计算图的数据处理效率和计算性能。
技术关键词
节点 贪心算法 存储芯片 编译优化方法 动态 锚点 内存 计算机可执行指令 接口 处理器 优化装置 计算机设备 模块 关系 存储器 序列 程序 参数 数据