arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2005.06410cs.PF
收藏

面向ARM多核处理器的高性能可移植卷积算子

High Performance and Portable Convolution Operators for ARM-based Multicore Processors

Pablo San Juan, Adrián Castelló, Manuel F. Dolz, Pedro Alonso-Jordá, Enrique S. Quintana-Ortí

摘要:

卷积神经网络对许多人工智能任务的巨大影响已经导致了针对这种类型的网络中存在的卷积算子的各种高性能算法的发展。这些方法之一利用\imcol变换,然后是通用矩阵乘法(GEMM),以便利用许多线性代数库中GEMM内核的高度优化实现。这种方法的主要问题是1)托管由IM 2COL变换生成的中间矩阵所需的大存储器工作空间;以及2)执行IM 2COL变换的时间,这对于复杂的神经网络是不可忽略的。本文提出了一种基于BLIS实现GEMM内核的可移植高性能卷积算法,该算法利用BLIS结构避免了中间存储器的使用。此外,该算法消除了显式IM 2COL变换的成本,同时保持了BLIS中GEMM底层实现的可移植性和性能。

英文摘要:

The considerable impact of Convolutional Neural Networks on many Artificial Intelligence tasks has led to the development of various high performance algorithms for the convolution operator present in this type of networks. One of these approaches leverages the \imcol transform followed by a general matrix multiplication (GEMM) in order to take advantage of the highly optimized realizations of the GEMM kernel in many linear algebra libraries. The main problems of this approach are 1) the large memory workspace required to host the intermediate matrices generated by the IM2COL transform; and 2) the time to perform the IM2COL transform, which is not negligible for complex neural networks. This paper presents a portable high performance convolution algorithm based on the BLIS realization of the GEMM kernel that avoids the use of the intermediate memory by taking advantage of the BLIS structure. In addition, the proposed algorithm eliminates the cost of the explicit IM2COL transform, while maintaining the portability and performance of the underlying realization of GEMM in BLIS.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑