摘要
本公开提供了一种深度学习任务的计算图的分割方法、装置、设备及介质,涉及人工智能技术领域,尤其涉及深度学习、深度学习编译技术领域。实现方案为:基于深度学习任务生成计算图;将多个算子中属于第一算子种类的算子的目标设备标记为计算设备,并将多个算子中不属于第一算子种类的算子的目标设备标记为主机;针对多个算子中用于处理元组的第一算子,基于第一算子所处理的元组中包含的各个数据的来源,更新第一算子的目标设备;以及对多个算子中目标设备相同的相邻算子进行合并,获得多个子图,以使多个子图中的每个子图在对应的目标设备上执行。