Distributed-memory DMRG via sparse and dense parallel tensor contractions
Ryan Levy, Edgar Solomonik, Bryan K. Clark
摘要
The density matrix renormalization group (DMRG) algorithm is a powerful tool for solving eigenvalue problems to model quantum systems. DMRG relies on tensor contractions and dense linear algebra to compute properties of condensed matter physics systems. However, its efficient parallel implementation is challenging due to limited concurrency, large memory footprint, and tensor sparsity. We mitigate these problems by implementing two new parallel approaches that handle block sparsity arising in DMRG, via Cyclops, a distributed memory tensor contraction library. We benchmark their performance on two physical systems using the Blue Waters and Stampede2 supercomputers. Our DMRG performance is improved by up to 5.9X in runtime and 99X in processing rate over ITensor, at roughly comparable computational resource use. This enables higher accuracy calculations via larger tensors for quantum state approximation. We demonstrate that despite having limited concurrency, DMRG is weakly scalable with the use of efficient parallel tensor contraction mechanisms.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
引用它的顶会 Paper2
- Sparta: high-performance, element-wise sparse tensor contraction on heterogeneous memoryJiawen Liu, Jie Ren, Roberto Gioiosa, Dong Li 等PPoPP 2021 · 被引用 31 次
- Merchandiser: Data Placement on Heterogeneous Memory for Task-Parallel HPC Applications with Load-Balance AwarenessZhen Xie, Jie Liu, Jiajia Li, Dong LiPPoPP 2023 · 被引用 18 次
相关 Paper
- Efficient 2D tensor network simulation of quantum systemsYuchen Pang, Tianyi Hao, Annika Dugad, Yiqing Zhou 等SC 2020 · 被引用 19 次
- Many-Body Electronic Correlation Energy using Krylov Subspace Linear SolversShikhar Shah, Boqin Zhang, Hua Huang, John E. Pask 等SC 2024 · 被引用 2 次
- Legate Sparse: Distributed Sparse Computing in PythonRohan Yadav, Wonchan Lee, Melih Elibol, Manolis Papadakis 等SC 2023 · 被引用 8 次
- Automatic Generation of Distributed-Memory Mappings for Tensor ComputationsMartin Kong, Raneem Abu Yosef, Atanas Rountev, P. SadayappanSC 2023 · 被引用 9 次
- A submatrix-based method for approximate matrix function evaluation in the quantum chemistry code CP2KMichael Lass, Robert Schade, Thomas D. Kühne, Christian PlesslSC 2020 · 被引用 7 次
