GPU-acceleration of tensor renormalization with PyTorch using CUDA

被引：5
|
作者：
Jha, Raghav G. [1 ]
Samlodia, Abhishek [2 ]
机构：
[1] Thomas Jefferson Natl Accelerator Facil, Newport News, VA 23606 USA
[2] Syracuse Univ, Dept Phys, Syracuse, NY 13244 USA
来源：
COMPUTER PHYSICS COMMUNICATIONS | 2024年 / 294卷
关键词：
Tensor; Lattice field theory; Spin models; CUDA; !text type='Python']Python[!/text; Speedup;
D O I：
10.1016/j.cpc.2023.108941
中图分类号：
TP39 [计算机的应用];
学科分类号：
081203 ; 0835 ;
摘要：
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. Published by Elsevier B.V.
引用
收藏
页数：8
相关论文
共 50 条
  • [11] GPU Acceleration of Object Detection on Video Stream using CUDA
    Dawwd, Shefa A.
    Salim, Ula T.
    2013 FIRST INTERNATIONAL SCIENTIFIC CONFERENCE ON ELECTRICAL, COMMUNICATION, COMPUTER, POWER, AND CONTROL ENGINEERING (ICECCPCE'13), 2013, : 198 - 203
  • [12] GAPS: GPU-Acceleration of PDE Solvers for Wave Simulation
    Hanindhito, Bagus
    Gourounas, Dimitrios
    Fathi, Arash
    Trenev, Dimitar
    Gerstlauer, Andreas
    John, Lizy K.
    PROCEEDINGS OF THE 36TH ACM INTERNATIONAL CONFERENCE ON SUPERCOMPUTING, ICS 2022, 2022,
  • [13] GPU-Acceleration of Affine Prediction in the Versatile Video Coding
    Storch, Iago
    Palomino, Daniel
    Bampi, Sergio
    2022 IEEE INTERNATIONAL SYMPOSIUM ON CIRCUITS AND SYSTEMS (ISCAS 22), 2022, : 429 - 433
  • [14] Experiments with GPU-Acceleration for Solving a Radiative Transfer Problem
    Vasconcelos, Paulo B.
    Marques, Osni
    COMPUTATIONAL SCIENCE AND ITS APPLICATIONS - ICCSA 2014, PT IV, 2014, 8582 : 550 - +
  • [15] RSA Public Key Acceleration on CUDA GPU
    Tembhurne, Jitendra V.
    Sathe, S. R.
    ARTIFICIAL INTELLIGENCE AND EVOLUTIONARY COMPUTATIONS IN ENGINEERING SYSTEMS, ICAIECES 2015, 2016, 394 : 365 - 375
  • [16] Acceleration of the DGF-FDTD Method on GPU Using the CUDA Technology
    Dziubak, Tomasz
    Wiktor, Michal
    Orlowski, Slawomir
    Stefanski, Tomasz P.
    2015 9th European Conference on Antennas and Propagation (EuCAP), 2015,
  • [17] Acceleration of CUDA Programs for Non-GPU Users Using Cloud
    Pisal, Tejas
    Walunj, Sandip M.
    Shrimali, Aslam
    Gautam, Omprakash
    Patil, Lalit
    2015 INTERNATIONAL CONFERENCE ON GREEN COMPUTING AND INTERNET OF THINGS (ICGCIOT), 2015, : 365 - 370
  • [18] GPU-acceleration for Moving Particle Semi-Implicit method
    Hori, Chiemi
    Gotoh, Hitoshi
    Ikari, Hiroyuki
    Khayyer, Abbas
    COMPUTERS & FLUIDS, 2011, 51 (01) : 174 - 183
  • [19] GPU-Acceleration of Sequence Homology Searches with Database Subsequence Clustering
    Suzuki, Shuji
    Kakuta, Masanori
    Ishida, Takashi
    Akiyama, Yutaka
    PLOS ONE, 2016, 11 (08):
  • [20] GPU-acceleration of waveform relaxation methods for large differential systems
    Conte, Dajana
    D'Ambrosio, Raffaele
    Paternoster, Beatrice
    NUMERICAL ALGORITHMS, 2016, 71 (02) : 293 - 310
12345