Optimized realization of Quantum Fourier Transform for domestic DCU accelerator

Abstract

Abstract Quantum Fourier Transform (QFT) is a key part of quantum algorithms such as Shor algorithm, Grover algorithm, and quantum phase estimation algorithm. The “Songshan” supercomputer system adopts the heterogeneous architecture of domestic CPU+DCU. The simulation implementation of the quantum Fourier transform on the “Songshan” supercomputer can promote the research and optimization of related quantum algorithms. Based on the HIP heterogeneous programming model, this paper first maps the phase shift transformation R gate operation of the QFT transformation to the DCU accelerator, and initially constructs the heterogeneous implementation of QFT on the “Songshan” supercomputer; then, the Hadamard gate transformation operation originally implemented on the host side was implemented on the DCU, which greatly reduces the communication overhead caused by the quantum information data transmission between the CPU and the DCU device; finally, the quantum logic gate is analyzed, and based on the operation of the phase shift transformation R gate on the probability amplitude information in the simulation algorithm, an optimization scheme to improve the activity of the DCU thread is proposed. During the experimental verification process, the realized 28-qubit quantum Fourier transform optimization has achieved 13.440 speedup compared with the serially realized QFT simulation on the CPU. The basic quantum gate optimization method based on DCU acceleration equipment in this paper can provide a reference for the efficient simulation of other quantum algorithms on domestic heterogeneous platforms.