Search papers, labs, and topics across Lattice.
This paper reformulates mass matrix assembly, a key kernel in implicit particle methods, as a sequence of matrix products optimized for hardware MMA tiles (tensor cores). This reformulation is general across interpolation orders and hardware platforms, and applies to both scalar and tensorial block mass matrices. By introducing particle batching and support-group decomposition, the authors achieve up to 3x speedup over optimized conventional implementations on NVIDIA tensor cores, reducing end-to-end ECSIM runtime by 15%.
Implicit particle methods get a 3x speed boost by recasting mass matrix assembly into tensor core-friendly matrix multiplications.
Matrix-multiply-accumulate (MMA) units, or tensor cores, are now widespread across modern computing architectures. Yet, their use for particle-grid operators remains limited. In implicit particle methods, mass-matrix assembly is a reduction-dominated kernel in which weighted outer products of interpolation weights are accumulated over particle support. We show that this operation can be reformulated exactly, cell by cell, as a sequence of matrix products matched to hardware MMA tiles. The formulation is general with respect to interpolation order and hardware platform, and applies to both scalar mass matrices and the tensorial block mass matrix arising in implicit in the Energy-Conserving Semi-Implicit Method (ECSIM) for Particle-in-Cell simulations. We introduce particle batching and a support-group decomposition for higher-order shape functions whose stencil extends beyond a single cell, specialize the method to first- and second-order B-spline interpolation, and implement it on NVIDIA tensor cores. The resulting kernels achieve up to 3x over optimized conventional implementations and reduce end-to-end ECSIM runtime by 15%.
