Search papers, labs, and topics across Lattice.
This paper optimizes the spectral element method (SEM) for wave propagation on ARM multicore CPUs with Scalable Matrix Extension (SME), achieving a 4–6x performance improvement over the original SPECFEM3D implementation. The authors employ a batched small-matrix kernel and a hybrid MPI+OpenMP execution scheme tailored for limited high-bandwidth memory (HBM) systems, while also conducting a dispersion-based iso-accuracy analysis of the $(h,p)$ tradeoff. The findings reveal that SME not only enhances kernel efficiency but also shifts the optimal polynomial order for SEM, leading to reduced time-to-solution and memory usage.
SME-enabled optimizations can boost SEM performance by up to 6x on ARM CPUs, transforming how we approach wave propagation simulations.
Wave propagation based on the spectral element method (SEM) is a representative HPC workload, but existing SEM implementations are not well matched to emerging ARM multicore CPUs with Scalable Matrix Extension (SME). We present an SME-enabled optimization of \textsc{SPECFEM3D} on the emerging LX2 processor that combines an SME-aware batched small-matrix kernel for SEM tensor-product operators, a memory-aware hybrid MPI+OpenMP execution scheme for limited-HBM systems, and a dispersion-based iso-accuracy study of the $(h,p)$ tradeoff. At fixed polynomial order, the optimized implementation improves full-application performance by 4--6$\times$ over the original code and delivers clear gains over optimized non-SME CPU baselines. Beyond these implementation-level gains, our results suggest that SME shifts the performance-favorable operating point toward higher polynomial orders along the dispersion-based iso-accuracy frontier, further reducing time-to-solution and working-set size. These results indicate that SME affects not only kernel efficiency, but also the practical discretization tradeoff for SEM on modern ARM multicore platforms.
Tsinghua AI