Parallel implementation of the Density Matrix Renormalization Group method achieving a quarter petaFLOPS performance on a single DGX-H100 GPU node
Fuente:
arXiv
Salvato in:
| Autori principali: | , , , , , , , |
|---|---|
| Natura: | Preprint |
| Pubblicazione: |
2024
|
| Soggetti: | |
| Accesso online: | |
| Tags: |
Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
|
| _version_ | 1866917717818212352 |
|---|---|
| author | Menczer, Andor van Damme, Maarten Rask, Alan Huntington, Lee Hammond, Jeff Xantheas, Sotiris S. Ganahl, Martin Legeza, Örs |
| author_facet | Menczer, Andor van Damme, Maarten Rask, Alan Huntington, Lee Hammond, Jeff Xantheas, Sotiris S. Ganahl, Martin Legeza, Örs |
| contents | We report cutting edge performance results for a hybrid CPU-multi GPU implementation of the spin adapted ab initio Density Matrix Renormalization Group (DMRG) method on current state-of-the-art NVIDIA DGX-H100 architectures. We evaluate the performance of the DMRG electronic structure calculations for the active compounds of the FeMoco and cytochrome P450 (CYP) enzymes with complete active space (CAS) sizes of up to 113 electrons in 76 orbitals [CAS(113, 76)] and 63 electrons in 58 orbitals [CAS(63, 58)], respectively. We achieve 246 teraFLOPS of sustained performance, an improvement of more than 2.5x compared to the performance achieved on the DGX-A100 architectures and an 80x acceleration compared to an OpenMP parallelized implementation on a 128-core CPU architecture. Our work highlights the ability of tensor network algorithms to efficiently utilize high-performance GPU hardware and shows that the combination of tensor networks with modern large-scale GPU accelerators can pave the way towards solving some of the most challenging problems in quantum chemistry and beyond. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2407_07411 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Parallel implementation of the Density Matrix Renormalization Group method achieving a quarter petaFLOPS performance on a single DGX-H100 GPU node Menczer, Andor van Damme, Maarten Rask, Alan Huntington, Lee Hammond, Jeff Xantheas, Sotiris S. Ganahl, Martin Legeza, Örs Chemical Physics Strongly Correlated Electrons Computational Physics We report cutting edge performance results for a hybrid CPU-multi GPU implementation of the spin adapted ab initio Density Matrix Renormalization Group (DMRG) method on current state-of-the-art NVIDIA DGX-H100 architectures. We evaluate the performance of the DMRG electronic structure calculations for the active compounds of the FeMoco and cytochrome P450 (CYP) enzymes with complete active space (CAS) sizes of up to 113 electrons in 76 orbitals [CAS(113, 76)] and 63 electrons in 58 orbitals [CAS(63, 58)], respectively. We achieve 246 teraFLOPS of sustained performance, an improvement of more than 2.5x compared to the performance achieved on the DGX-A100 architectures and an 80x acceleration compared to an OpenMP parallelized implementation on a 128-core CPU architecture. Our work highlights the ability of tensor network algorithms to efficiently utilize high-performance GPU hardware and shows that the combination of tensor networks with modern large-scale GPU accelerators can pave the way towards solving some of the most challenging problems in quantum chemistry and beyond. |
| title | Parallel implementation of the Density Matrix Renormalization Group method achieving a quarter petaFLOPS performance on a single DGX-H100 GPU node |
| topic | Chemical Physics Strongly Correlated Electrons Computational Physics |
| url | https://arxiv.org/abs/2407.07411 |