Saved in:
| Main Authors: | , , , , , , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2026
|
| Subjects: | |
| Online Access: | https://arxiv.org/abs/2605.10363 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866913121947353088 |
|---|---|
| author | Wei, Xinran Pan, Yan Ju, Fusong Zhou, Zehao Zhang, Yihong Huang, Lin Zhu, Jianwei Zhang, Jia Xia, Huanhuan Shao, Bin Qin, Tao |
| author_facet | Wei, Xinran Pan, Yan Ju, Fusong Zhou, Zehao Zhang, Yihong Huang, Lin Zhu, Jianwei Zhang, Jia Xia, Huanhuan Shao, Bin Qin, Tao |
| contents | Locality-driven integration is a pervasive computational pattern in quantum chemistry, arising whenever spatially localized basis functions interact through numerical quadrature or integral screening. The dominant matrix multiplications in these tasks exhibit dynamic, structured sparsity driven by spatial locality, posing significant challenges for both dense batched kernels and generic sparse formats on GPUs. We present KerneLDI, a GPU-oriented framework that addresses this regime by co-designing data layout, screening logic, and matrix-computation operators to realize block-structured matrix multiplication for locality-driven integration. KerneLDI reorganizes operand matrices into a unified block-filtered representation that retains only spatially relevant blocks, and executes the resulting contractions with customized dense block multipliers that adapt proven dense-matmul optimizations to retained block pairs. We develop and evaluate KerneLDI on exchange--correlation (EXC) integration in Kohn--Sham density functional theory, a representative and computationally critical instance of this pattern. Across diverse molecular systems, KerneLDI preserves numerical accuracy while delivering up to 10$\times$ speedup for EXC evaluation over a dense GPU baseline, scales favorably with increasing system size and multi-GPU parallelism, accelerates end-to-end self-consistent field calculations, and yields nearly 6$\times$ throughput improvement for ab initio molecular dynamics. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2605_10363 |
| institution | arXiv |
| publishDate | 2026 |
| record_format | arxiv |
| spellingShingle | Accelerating Locality-Driven Integration in Quantum Chemistry with Block-Structured Matrix Multiplication Wei, Xinran Pan, Yan Ju, Fusong Zhou, Zehao Zhang, Yihong Huang, Lin Zhu, Jianwei Zhang, Jia Xia, Huanhuan Shao, Bin Qin, Tao Computational Physics Distributed, Parallel, and Cluster Computing Chemical Physics Locality-driven integration is a pervasive computational pattern in quantum chemistry, arising whenever spatially localized basis functions interact through numerical quadrature or integral screening. The dominant matrix multiplications in these tasks exhibit dynamic, structured sparsity driven by spatial locality, posing significant challenges for both dense batched kernels and generic sparse formats on GPUs. We present KerneLDI, a GPU-oriented framework that addresses this regime by co-designing data layout, screening logic, and matrix-computation operators to realize block-structured matrix multiplication for locality-driven integration. KerneLDI reorganizes operand matrices into a unified block-filtered representation that retains only spatially relevant blocks, and executes the resulting contractions with customized dense block multipliers that adapt proven dense-matmul optimizations to retained block pairs. We develop and evaluate KerneLDI on exchange--correlation (EXC) integration in Kohn--Sham density functional theory, a representative and computationally critical instance of this pattern. Across diverse molecular systems, KerneLDI preserves numerical accuracy while delivering up to 10$\times$ speedup for EXC evaluation over a dense GPU baseline, scales favorably with increasing system size and multi-GPU parallelism, accelerates end-to-end self-consistent field calculations, and yields nearly 6$\times$ throughput improvement for ab initio molecular dynamics. |
| title | Accelerating Locality-Driven Integration in Quantum Chemistry with Block-Structured Matrix Multiplication |
| topic | Computational Physics Distributed, Parallel, and Cluster Computing Chemical Physics |
| url | https://arxiv.org/abs/2605.10363 |