Towards electronic structure-based ab-initio molecular dynamics simulations with hundreds of millions of atoms

We push the boundaries of electronic structure-based ab-initio molecular dynamics (AIMD) beyond 100 million atoms. This scale is otherwise barely reachable with classical force-field methods or novel neural network and machine learning potentials. We achieve this breakthrough by combining innovation...

ver descrição completa

Detalhes bibliográficos
Autores: Schade, Robert, Kenter, Tobias, Elgabarty, Hossam, Lass, Michael, Schütt, Ole, Mohr, Stephan|||0000-0003-2510-5805
Tipo de documento: artigo
Data de publicação:2022
País:España
Recursos:Universitat Politècnica de Catalunya (UPC)
Repositório:UPCommons. Portal del coneixement obert de la UPC
Idioma:inglês
OAI Identifier:oai:upcommons.upc.edu:2117/364649
Acesso em linha:https://hdl.handle.net/2117/364649
https://dx.doi.org/10.1016/j.parco.2022.102920
Access Level:Acceso aberto
Palavra-chave:High performance computing
Computational science & engineering
Molecular dynamics.
Monte Carlo method--Computer programs
High-performance computing
Massively-parallel algorithms
Large-scale linear algebra
Ab-initio molecular dynamics
Approximate computing
Simulació per ordinador
Àrees temàtiques de la UPC::Informàtica::Aplicacions de la informàtica::Aplicacions informàtiques a la física i l‘enginyeria
Descrição
Resumo:We push the boundaries of electronic structure-based ab-initio molecular dynamics (AIMD) beyond 100 million atoms. This scale is otherwise barely reachable with classical force-field methods or novel neural network and machine learning potentials. We achieve this breakthrough by combining innovations in linear-scaling AIMD, efficient and approximate sparse linear algebra, low and mixed-precision floating-point computation on GPUs, and a compensation scheme for the errors introduced by numerical approximations. The core of our work is the non-orthogonalized local submatrix method (NOLSM), which scales very favorably to massively parallel computing systems and translates large sparse matrix operations into highly parallel, dense matrix operations that are ideally suited to hardware accelerators. We demonstrate that the NOLSM method, which is at the center point of each AIMD step, is able to achieve a sustained performance of 324 PFLOP/s in mixed FP16/FP32 precision corresponding to an efficiency of 67.7% when running on 1536 NVIDIA A100 GPUs.