Parallel time integration using Batched BLAS (Basic Linear Algebra Subprograms) routines
arXiv:2108.07126 · doi:10.1016/j.cpc.2021.108181
Abstract
We present an approach for integrating the time evolution of quantum systems. We leverage the computation power of graphics processing units (GPUs) to perform the integration of all time steps in parallel. The performance boost is especially prominent for small to medium-sized quantum systems. The devised algorithm can largely be implemented using the recently-specified batched versions of the BLAS routines, and can therefore be easily ported to a variety of platforms. Our PARAllelized Matrix Exponentiation for Numerical Time evolution (PARAMENT) implementation runs on CUDA-enabled graphics processing units.
arXiv comment8 pages including source of the integration core in pseudo C code
References in corpus (7)
- QuTiP 2: A Python framework for the dynamics of open quantum systems
- The Magnus expansion and some of its applications
- Interactive Supercomputing on 40,000 Cores for Machine Learning and Data Analysis
- Quantum systems under frequency modulation
- High-order commutator-free exponential time-propagation of driven quantum systems
- Location of crossings in the Floquet spectrum of a driven two-level system
- Magnus integrators on multicore CPUs and GPUs