4 papers · 1 filter
ForgeStencil: Automating Per-Case Stencil Specialization from Kernels to 100+ Real Applications
Yaojian Chen, Yuxuan Li, Wubing Wan +3
On modern GPUs the fastest stencil kernel depends on the stencil's shape, precision, and host application, and a kernel tuned for one case is rarely fastest for another. Stencil DS…
Beyond Fast Contractions: Attenuation and Recovery of Matrix-Engine Speedups in High-Order Finite Elements
Yinuo Wang, Lin Gan, Tianqi Mao +8
Modern processors increasingly provide matrix engines whose peak arithmetic throughput greatly exceeds conventional SIMD, but scientific applications rarely realize this advantage…
StateFlow: Sequence Pipeline Parallelism for Long-Context Modeling with Linear Recurrence
Wenxuan Zhao, Yingfa Chen, Xu Han +7
Long-context training is increasingly important for large language models, and linear attention and state space models have become popular for improving long-context efficiency. Ho…
MMStencil: Optimizing High-order Stencils on Multicore CPU using Matrix Unit
Yinuo Wang, Tianqi Mao, Lin Gan +8
Matrix-accelerated stencil computation is a hot research topic, yet its application to three-dimensional (3D) high-order stencils and HPC remains underexplored. With the emergence…