[发明专利]通过人工智能处理器处理一维复数数组的方法和设备

说明书

本公开记载了一种通过人工智能处理器处理一维复数数组的方法、电子设备和计算装置，其中计算装置可以包括在组合处理装置中，该组合处理装置还可以包括通用互联接口和其他处理装置。所述计算装置与其他处理装置进行交互，共同完成用户指定的计算操作。组合处理装置还可以包括存储装置，该存储装置分别与计算装置和其他处理装置连接，用于计算装置和其他处理装置的数据。

技术领域

本发明涉及数据处理领域，更具体地，涉及人工智能处理器上的矩阵运算领域。

背景技术

离散傅里叶变换和离散傅里叶逆变换在雷达、数字通信、数字图像处理和计算机视觉等领域中有着很广泛的作用，这使得离散傅里叶变换的快速工程实现具有十分重要的意义。但是随着人工智能技术的发展，及更前沿领域的需要，图像处理、雷达信号处理和视频处理等应用系统中对大规模数据的离散傅里叶变换需求越来越高。在雷达领域，随着高分辨大测绘带宽的合成孔径雷达的飞速发展，对大规模数据的离散傅里叶快速实现提出了更高的要求。然而，针对大规模数据的离散傅里叶快速实现对资源的消耗随着数据长度的增加而增加，且对大规模数据的离散傅里叶快速实现的优化效率往往和单个处理器的计算资源开销的程度直接相关，使得单个处理器的计算资源成为实现针对大规模的离散傅里叶变换快速实现的瓶颈。因此，充分应用处理器的计算资源，以及多处理器并行执行时很有必要的。

在CPU上只能进行标量的计算，随着数据规模的增加，计算时间呈指数级增加，所以如果可以直接进行一组数据的计算，即张量的计算，那么计算性能可以显著提升。之前常用快速傅里叶变换进行离散傅里叶变换的计算，通常也是进行的标量的计算。虽然这也可以减少算法复杂度，但是执行快速傅里叶变换的计算量很大，并且和系统的性能有很大关系。

发明内容

本公开的目的在于克服现有技术中无法进行张量计算的缺陷，提供种通过人工智能处理器处理一维复数数组的方法。

根据本公开的第一方面，提供一种通过人工智能处理器处理一维复数数组的方法，其中，所述一维复数数组的长度为N，与所述一维复数数组对应的系数矩阵的尺寸为N×N，所述方法包括：将所述一维复数数组加载至所述人工智能处理器上的片上存储单元的第一存储区；将所述系数矩阵加载至所述人工智能处理器上的片上存储单元的第二存储区；所述人工智能处理器利用利用所述一维复数数组和所述系数矩阵进行傅里叶变换，以得到运算结果；以及所述人工智能处理器将所述运算结果传送至所述片外存储单元以进行存储。

根据本公开的第二方面，提供一种电子设备，包括：一个或多个处理器；以及存储器，所述存储器中存储有计算机可执行指令，当所述计算机可执行指令由所述一个或多个处理器运行时，使得所述电子设备执行如上所述的方法。

根据本公开的第三方面，提供一种计算机可读存储介质，包括计算机可执行指令，当所述计算机可执行指令由一个或多个处理器运行时，执行如上所述的方法。

本公开的技术方案可以充分利用硬件资源，将数据一次性的加载，然后在片上内存上进行高速计算，在片外内存上进行存储，降低内存间的缓存耗时，从而提升访存效率，总体提升算法的性能。

附图说明

通过参考附图阅读下文的详细描述，本公开示例性实施方式的上述以及其他目的、特征和优点将变得易于理解。在附图中，以示例性而非限制性的方式示出了本公开的若干实施方式，并且相同或对应的标号表示相同或对应的部分其中：

图1a示出了本公开的方法可以应用的一个处理器组的内部结构示意图；

图1b示出了本公开的方法可以应用的人工智能处理器的结构示意图；

图2示出了根据本公开的一个实施方式的一种通过人工智能处理器处理一维复数数组的方法；

图3示出了根据本公开的一个实施方式的一维复数数组的示意图；