<tt>CuGBasis</tt>: High-performance CUDA/Python library for efficient computation of quantum chemistry density-based descriptors for larger systems
CuGBasis is a free and open-source CUDA®/Python library for efficient computation of scalar, vector, and matrix quantities crucial for the post-processing of electronic structure calculations. CuGBasis integrates high-performance Graphical Processing Unit (GPU) computing with the ease and flexibility of Python programming, making it compatible with a vast ecosystem of libraries. We showcase its utility as a Python library and demonstrate its seamless interoperability with existing Python software to gain chemical insight from quantum chemistry calculations. Leveraging GPU-accelerated code, cuGBasis exhibits remarkable performance, making it highly applicable to larger systems or large databases. Our benchmarks reveal a 100-fold performance gain compared to alternative software packages, including serial/multi-threaded Central Processing Unit and GPU implementations. This paper outlines various features and computational strategies that lead to cuGBasis's enhanced performance, guiding developers of GPU-accelerated code.
Authors
- Alireza Tehrani (ORCID: https://orcid.org/0000-0003-1308-8614)
- Farnaz Heidar‐Zadeh (ORCID: https://orcid.org/0000-0002-2069-050X)
- Michelle Richer (ORCID: https://orcid.org/0009-0001-2563-4603)
Institutions
- Queen's University (CA)
Publication Details
- Journal
- The Journal of Chemical Physics
- Published
- 2024-08-19
- DOI
- https://doi.org/10.1063/5.0216781
- Citations
- 10
- Primary Topic
- Machine Learning in Materials Science
- Type
- article
- Field-Weighted Citation Impact
- 1.17
Funders
- Compute Canada
- Queen's University
- Natural Sciences and Engineering Research Council of Canada