[发明专利]卷积运算的处理方法、电子设备及计算机可读存储介质

权利要求书

1.一种卷积运算的处理方法，其特征在于，所述方法包括：

获取待处理的卷积运算、硬件参数以及配置数据库；

若确定所述配置数据库中无所述卷积运算的卷积尺寸和所述硬件参数对应的配置参数，则根据所述卷积尺寸以及所述硬件参数定义一参数搜索空间；其中，所述配置数据库的配置参数基于Winograd算法确定；

根据所述Winograd算法将所述参数搜索空间中的配置参数生成多个运算代码，并利用所述多个运算代码对所述卷积运算进行计算，以得到多个运算结果；

将所述多个运算结果中满足预设条件的一个运算结果对应的运算代码的配置参数，存储至所述配置数据库。

2.根据权利要求1所述的处理方法，其特征在于，

所述若确定所述配置数据库中无所述卷积运算的卷积尺寸和所述硬件参数对应的配置参数，则根据所述卷积尺寸以及所述硬件参数定义一参数搜索空间，包括：

判断所述配置数据库中是否有所述卷积运算的卷积尺寸和所述硬件参数对应的配置参数；

若是，则按照所述配置参数生成运算代码，并进行计算得到运算结果；

若否，则根据所述卷积尺寸以及所述硬件参数定义多个配置参数，以形成所述参数搜索空间。

3.根据权利要求2所述的处理方法，其特征在于，

所述卷积尺寸和所述硬件参数对应的所述配置参数至少包括输入通道的数量、输出通道的数量、输入数据的尺寸、卷积核变换对应的寄存器块数量、卷积核输出通道的数据块、输入数据变换对应的寄存器块数量、输入数据变换对应的缓存块、卷积核转换标签、搜索空间标签；

其中，所述卷积核变换对应的寄存器块数量的取值范围为[2，7]，所述卷积核输出通道的数据块的取值范围为[0，K/n]，所述n为所述卷积核变换对应的寄存器块数量，所述K为所述输出通道的数量，所述输入数据变换对应的寄存器块数量的取值范围为[2，7]，所述输入数据变换对应的缓存块的取值范围为[0，tiles/m]，所述m为所述输入数据变换对应的寄存器块数量，所述tiles为所述缓存块的数量，所述卷积核转换标签为0或1，所述搜索空间标签为0、1、2或3。

4.根据权利要求2所述的处理方法，其特征在于，

所述根据所述卷积尺寸以及所述硬件参数定义多个配置参数，以形成所述参数搜索空间，包括：

根据所述卷积尺寸以及所述硬件参数定义多个参数组合，以得到所述配置参数；

基于所述配置参数形成所述参数搜索空间。

5.根据权利要求4所述的处理方法，其特征在于，

所述根据所述Winograd算法将所述参数搜索空间中的配置参数生成多个运算代码，并利用所述多个运算代码对所述卷积运算进行计算，以得到多个运算结果，包括：

对所述参数搜索空间中的每个参数组合生成一运算代码，得到多个运算代码；

利用多个运算代码分别对所述卷积运算进行计算，以得到多个运算结果。

6.根据权利要求5所述的处理方法，其特征在于，

所述将所述多个运算结果中满足预设条件的一个运算结果对应的运算代码的配置参数，存储至所述配置数据库，包括：

判断多个运算结果中是否存在满足预设条件的运算结果，所述预设条件至少包括所述运算结果对应的运算时间最短；

若是，则将满足预设条件的运算结果对应的所述配置参数存储至所述配置数据库。

7.根据权利要求5所述的处理方法，其特征在于，

所述利用多个运算代码分别对所述卷积运算进行计算，以得到多个运算结果，包括：

根据所述运算代码中的卷积核变换对应的寄存器块数量和卷积核输出通道的数据块进行卷积核变换，得到卷积核变换矩阵；

根据所述运算代码中的输入数据变换对应的寄存器块数量和输入数据变换对应的缓存块对输入数据进行变换，得到输入数据变换矩阵；

将所述卷积核变换矩阵和输入数据变换矩阵按照所述搜索空间标签进行矩阵乘法，得到预运算结果；

将所述预运算结果进行压缩，得到所述运算结果。