[发明专利]计算图优化方法及装置、电子设备、计算机可读存储介质

说明书

本公开提供了一种计算图优化方法及装置、电子设备、计算机可读介质，该方法包括：对待处理的第一计算图进行遍历，确定所述第一计算图的第一子图；将所述第一子图与图模板库中的模板图进行匹配，确定所述模板图中是否存在与第一子图匹配的目标子图；在存在与所述第一子图匹配的目标子图的情况下，根据所述目标子图，对所述第一计算图进行优化，得到所述第一计算图的优化结果。根据本公开的实施例能够提升计算图优化效率。

技术领域

本公开涉及计算机技术领域，特别涉及一种计算图优化方法及装置、电子设备、计算机可读存储介质。

背景技术

随着人工智能技术的不断发展，越来越多的神经网络被应用到了生产实践中。这些神经网络在编程、部署、执行的过程中往往使用计算图的形式去描述，计算图作为IR(Intermediate Representation，中间表示)也被用于高性能计算等其他领域，因此，针对计算图的优化(如编译优化与执行优化)已经越来越受到人们的重视。

发明内容

本公开提供一种基于众核系统的计算图优化方法及装置、处理核、电子设备、计算机可读存储介质。

第一方面，本公开提供了一种计算图优化方法，该计算图优化方法包括：对待处理的第一计算图进行遍历，确定所述第一计算图的第一子图；将所述第一子图与图模板库中的模板图进行匹配，确定所述模板图中是否存在与第一子图匹配的目标子图；在存在与所述第一子图匹配的目标子图的情况下，根据所述目标子图，对所述第一计算图进行优化，得到所述第一计算图的优化结果。

第二方面，本公开提供了一种计算图优化装置，该计算图优化装置包括：遍历模块，用于对待处理的第一计算图进行遍历，确定所述第一计算图的第一子图；匹配模块，用于将所述第一子图与图模板库中的模板图进行匹配，确定所述模板图中是否存在与第一子图匹配的目标子图；优化模块，用于在存在与所述第一子图匹配的目标子图的情况下，根据所述目标子图，对所述第一计算图进行优化，得到所述第一计算图的优化结果。

第三方面，本公开提供了一种电子设备，该电子设备包括：至少一个处理器；以及与所述至少一个处理器通信连接的存储器；其中，所述存储器存储有可被所述至少一个处理器执行的一个或多个计算机程序，一个或多个所述计算机程序被所述至少一个处理器执行，以使所述至少一个处理器能够执行上述的计算图优化方法。

第四方面，本公开提供了一种计算机可读存储介质，其上存储有计算机程序，其中，所述计算机程序在被处理器/处理核执行时实现上述的计算图优化方法。

本公开所提供的实施例，将用于传统的基于一维文本的编译体系的模板匹配方法用于对计算图进行优化，提升了计算图的优化效率，通过一系列流程，提升了计算图的编译效率。

应当理解，本部分所描述的内容并非旨在标识本公开的实施例的关键或重要特征，也不用于限制本公开的范围。本公开的其它特征将通过以下的说明书而变得容易理解。

附图说明

附图用来提供对本公开的进一步理解，并且构成说明书的一部分，与本公开的实施例一起用于解释本公开，并不构成对本公开的限制。通过参考附图对详细示例实施例进行描述，以上和其他特征和优点对本领域技术人员将变得更加显而易见，在附图中：

图1为相关技术中对计算图进行优化的流程的示意图；

图2为本公开实施例提供的一种计算图优化方法的流程图；

图3为本公开实施例提供的一种计算图优化方法的部分步骤的流程图；

图4为本公开实施例提供的一种计算图优化方法的示意图；

图5为本公开实施例提供的一种计算图优化方法的部分步骤的流程图；

图6为本公开实施例提供的一种计算图优化方法的部分步骤的流程图；