ConvStencil: Transform Stencil Computation to Matrix Multiplication on Tensor Cores

Authors: Yuetao Chen, Kun Li, Yuhao Wang, Donglin Bai, Lei Wang, Lingxiao Ma, Liang Yuan, Yunquan Zhang, Ting Cao, Mao Yang - Proceedings of the 29th ACM SIGPLAN Annual Symposium on Principles and Practice of Parallel Programming 2024 cited by 22

Tensor Core Unit (TCU) is increasingly integrated into modern high-performance processors to enhance matrix multiplication performance. However, constrained to its over-specification, its potential for improving other critical scientific operations like stencil computations remains untapped.