Treffer: Evaluating multi-GPU computing capabilities of Numba and CuPy.

In this paper, we evaluate the performance of Numba and CuPy in multi-GPU configurations, focusing on both strong and weak scalings. We employ two benchmark problems: pseudo-random number generation and one-dimensional Monte Carlo radiation transport in a purely absorbing medium. In the experiments, we compare Numba, CuPy, and CUDA C implementations under both default and optimized conditions implemented on the NVIDIA DGX-2 server platform. In the default setup, CUDA C delivers better performance and the highest energy efficiency. However, we demonstrate that CuPy can achieve substantial speedups in optimized mode, though this requires extensive code modifications. Numba shows competitive performance in cases with minimal data transfer to global memory, but its scalability and energy efficiency are limited by CPU-side random number generator state initialization bottlenecks, and it benefits less from optimizations compared to CuPy. [ABSTRACT FROM AUTHOR]

Copyright of Cluster Computing is the property of Springer Nature and its content may not be copied or emailed to multiple sites without the copyright holder's express written permission. Additionally, content may not be used with any artificial intelligence tools or machine learning technologies. However, users may print, download, or email articles for individual use. This abstract may be abridged. No warranty is given about the accuracy of the copy. Users should refer to the original published version of the material for the full abstract. (Copyright applies to all Abstracts.)