[发明专利]多通道卷积层处理方法和装置

权利要求书

1.一种单通道卷积层处理方法，其特征在于，包括：

将待处理数据划分为多个数据块；

多个图形处理器线程组将所述多个数据块读取到所述多个图形处理器线程组的本地存储器中；

所述多个图形处理器线程组将多个过滤器读取到所述多个图形处理器线程组的本地存储器中；以及

所述多个图形处理器线程组中的多个线程同时计算所述多个数据块中对应的数据点与所述多个过滤器的卷积。

2.根据权利要求1所述的方法，其特征在于，所述多个数据块是非规则数据块。

3.根据权利要求1或2所述的方法，其特征在于，使用多个图形处理器线程组将所述多个数据块读取到图形处理器线程的本地存储器中包括：

将每个数据块划分为多个子数据块，其中至少一个子数据块的数据量与对应的图形处理器线程组中的线程数目匹配；以及

所述对应的图形处理器线程组依次将所述多个子数据块读取到所述图形处理器线程组的本地存储器中。

4.根据权利要求1或2所述的方法，其特征在于，在所述多个图形处理器线程组将所述多个数据块读取到所述多个图形处理器线程组的本地存储器中时，还将对应的填充数据读取到本地存储器中。

5.一种单通道卷积层处理装置，其特征在于，包括：

划分模块，用于将待处理数据划分为多个数据块；

第一读取模块，用于通过多个图形处理器线程组将所述多个数据块读取到所述多个图形处理器线程组的本地存储器中；

第二读取模块，用于通过所述多个图形处理器线程组将多个过滤器读取到所述多个图形处理器线程组的本地存储器中；以及

计算模块，用于通过所述多个图形处理器线程组中的多个线程同时计算所述多个数据块中对应的数据点与所述多个过滤器的卷积。

6.根据权利要求5所述的装置，其特征在于，所述多个数据块是非规则数据块。

7.根据权利要求5或6所述的装置，其特征在于，所述第一读取模块包括：

划分单元，用于将每个数据块划分为多个子数据块，其中至少一个子数据块的数据量与对应的图形处理器线程组中的线程数目匹配；以及

读取单元，用于通过所述对应的图形处理器线程组依次将所述多个子数据块读取到所述图形处理器线程组的本地存储器中。

8.根据权利要求5或6所述的装置，其特征在于，所述第一读取模块在所述多个图形处理器线程组将所述多个数据块读取到所述多个图形处理器线程组的本地存储器中时，还还用于将对应的填充数据读取到本地存储器中。

9.一种多通道卷积层处理方法，其特征在于，包括：

待处理数据及过滤器包含多个通道；

将每个通道的待处理数据划分为多个数据块；

多个图形处理器线程组同时将所述多个数据块读取到所述多个图形处理器线程组的本地存储器中；

所述多个图形处理器线程组同时将多个过滤器读取到所述多个图形处理器线程组的本地存储器中；

所述多个图形处理器线程组中的多个线程同时计算所述一个通道多个数据块中对应的数据点与所述多个过滤器的卷积；

所述多个图形处理器线程组中的多个线程依次进行多个通道的卷积，并将每个通道的卷积结果进行累加；

将最终卷积结果写入所述多个图形处理器线程组的全局存储器中。

10.根据权利要求9所述的方法，其特征在于，在所述多个图形处理器线程组同时将所述多个数据块读取到所述多个图形处理器线程组的本地存储器中时，还将对应的填充数据读取到本地存储器中，所述填充之后的多个数据块是非规则数据块。

11.根据权利要求9或10所述的方法，其特征在于，所述待处理数据为图像数据，所述方法进一步包括：

将所述多幅图像和多个过滤器读取到所述图形处理器的全局存储器中。

12.根据权利要求9或10所述的方法，其特征在于，使用多个图形处理器线程组将所述多个数据块读取到图形处理器线程的本地存储器中包括：

将每个数据块划分为多个子数据块，其中至少一个子数据块的数据量与对应的图形处理器线程组中的线程数目匹配；以及

所述对应的图形处理器线程组依次将所述多个子数据块读取到所述图形处理器线程组的本地存储器中。