摘要
发明公开了基于多级存储芯片的计算图的编译优化方法和装置。该方法包括:对运行在多级存储芯片上的计算图进行筛选获取候选算子节点;获取当前各候选算子节点的动态类型;根据动态类型对计算图中的候选算子节点进行前向搜索和后向搜索获取融合算子节点,并根据融合算子节点对计算图进行更新获取优化计算图,通过从计算图中筛选出符合融合条件的候选算子节点,以及各候选算子节点的动态类型,并根据动态类型对候选算子节点进行搜索获取融合算子节点,由于位于同一融合算子节点中的候选算子节点仅在共享存储层进行张量搬运,从而避免了在不同存储层之间张量数据的搬运次数,提高了计算图的数据处理效率和计算性能。