Exploiting floating point emulation in tensor contractions in relativistic coupled cluster theory
Maxime Le, Ryan Benazzouk, Jan Brandejs, Andre Severo Pereira Gomes
Abstract
In this manuscript we outline a computational strategy to leverage a vendor-optimized implementation of the Ozaki scheme I for floating point emulation as part of an implementation of the Transpose-Transpose-GEMM-Transpose (TTGT) pattern for tensor contractions, and assess its potential to accelerate two of the most expensive tensor contractions appearing in relativistic coupled cluster theory in high-end accelerators such as the NVIDIA Blackwell B200 system in which there is a strong differential in the performance of computations for double and lower precision datatypes. Our results indicate that this emulation-based TTGT strategy can provide significant speedups compared to runs without emulation, and can be competitive with state-of-the-art, vendor-optimized tensor contraction libraries using native double precision as compute datatype. More importantly, our scheme provides a path towards easily allowing users to tune the desired accuracy in the tensor contraction steps, including cases where precision between single and double is sufficient for the application.
Create a lesson
Related papers
When optimized geometries are not optimal: Implications for inverted singlet-triplet gap predictions
Atreyee Majumdar, Elamathi Selvan Samynathan, Raghunathan Ramakrishnan
Open-shell Configuration State Functions as trial states for phaseless auxiliary-field quantum Monte Carlo
Don Danilov, Hugh G. A. Burton, James Shee
Machine Learning the Complete Active-Space Wavefunction Reconstruction
Aniruddha Seal, Bhavnesh Jangid, Helen S. Clifford et al.
Brueckner Orbitals in Local Pair Natural Orbital Coupled-Cluster Theory: DLPNO-BCCD(T) and DLPNO-BCCD(T)Lambda2
Andy Jiang, Tibor Gyori, Elizabeth C. Toth et al.
Exact periodic electrostatics for latent-charge machine-learning potentials
Eslam Ibrahim
Effective Harmonic and Anharmonic Models for Vibronic Spectra
Vincenzo Barone