Research

Paper

TESTING March 23, 2026

A Unified Heterogeneous Implementation of Numerical Atomic Orbitals-Based Real-Time TDDFT within the ABACUS Package

Authors

Taoni Bao, Yuanbo Li, Zichao Deng, Haotian Zhao, Denghui Lu, Yike Huang, Chao Lian, Lixin He, Mohan Chen

Abstract

We present a unified heterogeneous computing framework for real-time time-dependent density functional theory (RT-TDDFT) based on numerical atomic orbitals (NAOs), implemented in the ABACUS package. We introduce three co-designed abstraction layers, including unified data containers, unified linear algebra operators, and unified grid integration interfaces. These layers collectively accelerate the two most demanding parts of NAO-based RT-TDDFT: explicit real-time wavefunction propagation and real-space grid operations such as Hamiltonian construction and force evaluation under external fields. We validate the method by computing optical properties for systems ranging from finite molecules to periodic solids, showing excellent agreement with standard benchmarks. Performance evaluations on bulk silicon demonstrate that a single GPU can achieve substantial wall-clock speedup over a fully utilized dual-socket CPU node. Furthermore, distributed multi-GPU strong-scaling tests confirm high parallel efficiency over tens of GPUs. This work establishes a high-performance, portable platform for large-scale first-principles simulations of ultrafast electron dynamics.

Metadata

arXiv ID: 2603.21835
Provider: ARXIV
Primary Category: cond-mat.mtrl-sci
Published: 2026-03-23
Fetched: 2026-03-24 06:02

Related papers

Raw Data (Debug)
{
  "raw_xml": "<entry>\n    <id>http://arxiv.org/abs/2603.21835v1</id>\n    <title>A Unified Heterogeneous Implementation of Numerical Atomic Orbitals-Based Real-Time TDDFT within the ABACUS Package</title>\n    <updated>2026-03-23T11:21:44Z</updated>\n    <link href='https://arxiv.org/abs/2603.21835v1' rel='alternate' type='text/html'/>\n    <link href='https://arxiv.org/pdf/2603.21835v1' rel='related' title='pdf' type='application/pdf'/>\n    <summary>We present a unified heterogeneous computing framework for real-time time-dependent density functional theory (RT-TDDFT) based on numerical atomic orbitals (NAOs), implemented in the ABACUS package. We introduce three co-designed abstraction layers, including unified data containers, unified linear algebra operators, and unified grid integration interfaces. These layers collectively accelerate the two most demanding parts of NAO-based RT-TDDFT: explicit real-time wavefunction propagation and real-space grid operations such as Hamiltonian construction and force evaluation under external fields. We validate the method by computing optical properties for systems ranging from finite molecules to periodic solids, showing excellent agreement with standard benchmarks. Performance evaluations on bulk silicon demonstrate that a single GPU can achieve substantial wall-clock speedup over a fully utilized dual-socket CPU node. Furthermore, distributed multi-GPU strong-scaling tests confirm high parallel efficiency over tens of GPUs. This work establishes a high-performance, portable platform for large-scale first-principles simulations of ultrafast electron dynamics.</summary>\n    <category scheme='http://arxiv.org/schemas/atom' term='cond-mat.mtrl-sci'/>\n    <category scheme='http://arxiv.org/schemas/atom' term='physics.comp-ph'/>\n    <published>2026-03-23T11:21:44Z</published>\n    <arxiv:primary_category term='cond-mat.mtrl-sci'/>\n    <author>\n      <name>Taoni Bao</name>\n    </author>\n    <author>\n      <name>Yuanbo Li</name>\n    </author>\n    <author>\n      <name>Zichao Deng</name>\n    </author>\n    <author>\n      <name>Haotian Zhao</name>\n    </author>\n    <author>\n      <name>Denghui Lu</name>\n    </author>\n    <author>\n      <name>Yike Huang</name>\n    </author>\n    <author>\n      <name>Chao Lian</name>\n    </author>\n    <author>\n      <name>Lixin He</name>\n    </author>\n    <author>\n      <name>Mohan Chen</name>\n    </author>\n  </entry>"
}