GPU-acceleration of tensor renormalization with PyTorch using CUDA
Abstract
We show that numerical computations based on tensor renormalization group (TRG) methods can be significantly accelerated with PYTORCH on graphics processing units (GPUs) by leveraging NVIDIA's Compute Unified Device Architecture (CUDA). We find improvement in the runtime (for a given accuracy) and its scaling with bond dimension for two-dimensional systems. Our results establish that utilization of GPU resources is essential for future precision computations with TRG.
- Publication:
-
Computer Physics Communications
- Pub Date:
- January 2024
- DOI:
- 10.1016/j.cpc.2023.108941
- arXiv:
- arXiv:2306.00358
- Bibcode:
- 2024CoPhC.29408941J
- Keywords:
-
- Tensor;
- Lattice field theory;
- Spin models;
- CUDA;
- Python;
- Speedup;
- High Energy Physics - Lattice;
- Physics - Computational Physics
- E-Print:
- v2: Added computation of critical exponents for three-state Potts and XY model. Comparison tests between CPU and GPU for Ising at T=Tc. Addressed the comments by the anonymous referee. Version matches the one accepted for publication in Computer Physics Communications. v1: 1+15 pages, comments welcome!