单通道卷积层及多通道卷积层处理方法和装置

摘要

本发明提出单通道卷积层处理及多通道卷积层处理方法和装置。该单通道卷积层处理方法包括：将待处理数据划分为多个数据块；多个图形处理器线程组将多个数据块读取到多个图形处理器线程组的本地存储器中；多个图形处理器线程组将多个过滤器读取到多个图形处理器线程组的本地存储器中；以及多个图形处理器线程组中的多个线程同时计算多个数据块中对应的数据点与多个过滤器的卷积。该多通道卷积处理方法包括：在单通道卷积处理方法的基础上，多个图形处理器线程组中的多个线程依次进行多个通道的卷积，并将每个通道的卷积结果进行累加。从而，无需额外空间,在存储空间利用上达到最优，增加了存储器的数据重用，具有极高的性能和性能功耗比。

主权项

一种单通道卷积层处理方法，其特征在于，包括：将待处理数据划分为多个数据块；多个图形处理器线程组将所述多个数据块读取到所述多个图形处理器线程组的本地存储器中；所述多个图形处理器线程组将多个过滤器读取到所述多个图形处理器线程组的本地存储器中；以及所述多个图形处理器线程组中的多个线程同时计算所述多个数据块中对应的数据点与所述多个过滤器的卷积。