ConvStencil: Transform Stencil Computation to Matrix Multiplication on Tensor Cores
Yuetao Chen, Kun Li, Yuhao Wang, Donglin Bai, Lei Wang, Lingxiao Ma, Liang Yuan, Yunquan Zhang, Ting Cao, Mao Yang
Abstract
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.
Topics & Concepts
StencilMatrix multiplicationComputationTensor (intrinsic definition)Computer scienceMultiplication (music)Computational scienceParallel computingMatrix (chemical analysis)MathematicsAlgorithmGeometryPhysicsMaterials scienceCombinatoricsQuantumComposite materialQuantum mechanicsParallel Computing and Optimization TechniquesTensor decomposition and applicationsInterconnection Networks and Systems