GPU-Acceleration of Tensor Renormalization with PyTorch using CUDA

Raghav G. Jha, Abhishek Samlodia
Thomas Jefferson National Accelerator Facility, Newport News, VA 23606, USA
arXiv:2306.00358 [hep-lat], (1 Jun 2023)


   title={GPU-Acceleration of Tensor Renormalization with PyTorch using CUDA},

   author={Raghav G. Jha and Abhishek Samlodia},






Download Download (PDF)   View View   Source Source   



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 and its scaling with bond dimension for two-dimensional systems. Our results establish that the utilization of GPU resources is essential for future precision computations with TRG.
No votes yet.
Please wait...

* * *

* * *

HGPU group © 2010-2023 hgpu.org

All rights belong to the respective authors

Contact us: