2611.11860

Total: 1

#1 GPU acceleration of the numerical-atomic-orbital DFT code OpenMX with GEMMul8: performance and numerical agreement on consumer and HPC GPUs [PDF] [Copy] [Kimi] [REL]

Authors: Hiroyuki Kawai, Taisuke Ozaki, Yoshiaki Ōno

We integrate GEMMul8, a library that emulates FP64 matrix multiplication on INT8 Tensor Cores, into the GPU version of the numerical-atomic-orbital first-principles code OpenMX 4.0. The GEMMul8 configuration is compared against the cuBLAS FP64 configuration within each of two environments, a consumer GeForce RTX 5080 and an HPC-oriented H100 PCIe, for the four computational paths formed by collinear (CL) or noncollinear (NCL) spin and band or $Γ$-point-cluster solvers. In fixed-work measurements that completed 25 SCF iterations, the diagonalization section of a 384-atom CL band calculation on the RTX 5080 is accelerated by a factor of 2.1 and the total run time by 1.8; a 512-atom CL cluster calculation, which uses only real matrix products in contrast to the CL band calculation, still gains a factor of 1.5 in the diagonalization section. On the H100, GEMMul8 provides little additional acceleration of the diagonalization section. In an accuracy test on a 216-atom Si system in which one atom was displaced by 0.05 angstrom, all 24 configurations were converged using an SCF convergence criterion of $10^{-8}$; every configuration converged at the same iteration within each path, and the differences between GEMMul8 and cuBLAS are at most $2\times10^{-12}$ Ha in the total energy and $7\times10^{-12}$ Ha/Bohr in force components.

Subject: Computational Physics

Publish: 2026-11-09 04:31:59 UTC