SC2020Top-tier venue
Distributed-memory DMRG via sparse and dense parallel tensor contractions
Ryan Levy, Edgar Solomonik, Bryan K. Clark
Abstract
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.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 1ec9630e-e240-461e-b726-068cd5ad3d96Cited by top-tier papers2
- Sparta: high-performance, element-wise sparse tensor contraction on heterogeneous memoryJiawen Liu, Jie Ren, Roberto Gioiosa, Dong Li et al.PPoPP 2021 · 31 citations
- Merchandiser: Data Placement on Heterogeneous Memory for Task-Parallel HPC Applications with Load-Balance AwarenessZhen Xie, Jie Liu, Jiajia Li, Dong LiPPoPP 2023 · 18 citations
Related papers
- Efficient 2D tensor network simulation of quantum systemsYuchen Pang, Tianyi Hao, Annika Dugad, Yiqing Zhou et al.SC 2020 · 19 citations
- Many-Body Electronic Correlation Energy using Krylov Subspace Linear SolversShikhar Shah, Boqin Zhang, Hua Huang, John E. Pask et al.SC 2024 · 2 citations
- Legate Sparse: Distributed Sparse Computing in PythonRohan Yadav, Wonchan Lee, Melih Elibol, Manolis Papadakis et al.SC 2023 · 8 citations
- Automatic Generation of Distributed-Memory Mappings for Tensor ComputationsMartin Kong, Raneem Abu Yosef, Atanas Rountev, P. SadayappanSC 2023 · 9 citations
- 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 citations
