Tehrani Alireza, Richer Michelle, Heidar-Zadeh Farnaz
Department of Chemistry, Queen's University, Kingston, Ontario K7L-3N6, Canada.
J Chem Phys. 2024 Aug 21;161(7). doi: 10.1063/5.0216781.
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.
CuGBasis是一个免费的开源CUDA®/Python库,用于高效计算电子结构计算后处理中至关重要的标量、向量和矩阵量。CuGBasis将高性能图形处理单元(GPU)计算与Python编程的简便性和灵活性相结合,使其与大量的库生态系统兼容。我们展示了它作为Python库的实用性,并演示了它与现有Python软件的无缝互操作性,以便从量子化学计算中获得化学见解。利用GPU加速代码,cuGBasis表现出卓越的性能,使其非常适用于更大的系统或大型数据库。我们的基准测试显示,与其他软件包相比,性能提升了100倍,包括串行/多线程中央处理器和GPU实现。本文概述了导致cuGBasis性能增强的各种特性和计算策略,为GPU加速代码的开发者提供指导。