[发明专利]编译优化方法、装置、计算机设备以及存储介质

说明书

本公开涉及一种编译优化方法、计算机设备以及存储介质。该方法能够确定张量的维度折叠信息，并根据该张量维度折叠信息对张量数据进行维度折叠，从而可以在优化张量数据的读写访问，进而可以优化目标处理器在运行阶段对张量数据的数据处理过程。同时，本公开的编译优化方法，可以减少程序中的循环次数，降低程序中的控制流，提高程序的执行效率。

技术领域

本公开涉及计算机技术领域，特别是涉及一种编译优化方法、装置、计算机设备以及存储介质。

背景技术

在任务(例如，图像处理任务、视频处理任务、文本处理任务和语音处理任务等)处理过程中，通常涉及对任务中的数据进行处理，这些数据可能是数值和张量，其中，张量可以包括一维张量(即向量)、二维张量(即矩阵)以及更高维度的张量)等。在数据较复杂的情况下，多维张量的处理通常可通过多维循环的方式来处理，但多维循环的处理方式引入了更多控制流，增大了处理资源的占用，且并不能充分利用处理器和存储器的带宽，导致处理效率较低。

发明内容

基于此，有必要针对上述技术问题，提供一种能够提升运算效率的编译优化方法、装置、计算机设备和存储介质。

根据本公开的一方面，提供了一种编译优化方法，包括：

获取待处理张量数据的维度信息，其中，所述维度信息包含所述待处理张量数据的至少一个维度的维度值；

根据所述待处理张量数据的维度信息，确定所述待处理张量数据的维度折叠信息；

根据所述待处理张量数据的维度折叠信息，对所述待处理张量数据进行维度折叠。

本公开一方面还提供了一种计算机设备，包括：

处理器；

用于存储处理器计算机可执行程序的存储器；

其中，所述处理器被配置为执行所述计算机可执行程序，以实现上述任意一项所述的方法。

本公开一方面还提供了一种计算机可读存储介质，其上存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现上述任意一项所述的方法。

本公开的编译优化方法，通过张量数据的维度信息确定张量的维度折叠信息(包括维度折叠方向以及可折叠维数等)，并根据该张量维度折叠信息对张量数据进行维度折叠，从而可以在优化张量数据的读写访问，进而可以优化目标处理器在运行阶段对张量数据的数据处理过程。同时，本公开的编译优化方法，可以减少程序中的循环次数，降低程序中的控制流，提高程序的执行效率。

附图说明

包含在说明书中并且构成说明书的一部分的附图与说明书一起示出了本公开的示例性实施例、特征和方面，并且用于解释本公开的原理。

图1为本公开实施例中用于执行编译优化方法的处理器的示意图；

图2为本公开一个实施例的编译优化方法的流程图；

图3为本公开实施例的张量数据维度折叠的一个示意图；

图4为本公开实施例中通过语法树判断张量数据的维度是否连续的示意图；

图5为本公开另一实施例的编译优化方法的流程图；

图6为本公开一实施例的组合计算装置的框图。

具体实施方式

下面将结合本公开实施例中的附图，对本公开实施例中的技术方案进行清楚、完整地描述，显然，所描述的实施例是本公开一部分实施例，而不是全部的实施例。基于本公开中的实施例，本领域技术人员在没有做出创造性劳动前提下所获得的所有其他实施例，都属于本公开保护的范围。