• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

计算机工程与科学 ›› 2021, Vol. 43 ›› Issue (01): 17-23.

• 高性能计算 • 上一篇    下一篇

量子线路模拟器QuEST在多GPU平台上的性能优化

张亮,常旭,秦志楷,沈立   

  1. (国防科技大学计算机学院,湖南 长沙 410073)
  • 收稿日期:2020-06-11 修回日期:2020-07-15 接受日期:2021-01-25 出版日期:2021-01-25 发布日期:2021-01-22
  • 基金资助:
    国家自然科学基金(61972407)

Performance optimization of quantum circuit simulator QuEST on multi-GPU platform

ZHANG Liang,CHANG Xu,QIN Zhi-kai,SHEN Li   

  1. (College of Computer Science and Technology,National University of Defense Technology,Changsha 410073,China )

  • Received:2020-06-11 Revised:2020-07-15 Accepted:2021-01-25 Online:2021-01-25 Published:2021-01-22

摘要: 在当前量子计算的研究中,量子线路模拟器作为重要的研究工具,一直受到研究者们的高度重视。QuEST是一款开源的通用量子线路模拟器,能在单个CPU结点、多个CPU结点和单个GPU等多种测试平台上灵活运行。量子线路模拟固有的并行性使其非常适合在GPU上运行,并能获得较大的性能加速。但是其缺点在于所消耗的内存空间巨大,单个GPU受显存容量限制,无法模拟具有更多量子位的量子系统。设计并实现了多GPU版本的QuEST模拟器,解决了单个GPU显存不足的问题,能够使用多个GPU模拟更多的量子位。而且,与单CPU版本相比可获得7~9倍的性能加速,与多CPU版本相比可获得3倍的性能加速。

关键词: 量子计算, 量子线路模拟器, QuEST, 多GPU, 显存

Abstract: In the current quantum computing research, as an important research tool, quantum circuit simulators have always been highly valued by researchers. QuEST is an open source general-purpose quantum circuit simulator that can run flexibly on multiple test platforms such as a single CPU node, multiple CPU nodes, and a single GPU. The inherent parallelism of quantum circuit simulator makes it very suitable for running on the GPU, and obtain greater performance acceleration. However, the disadvantage is that the memory space consumed is huge. A single GPU is limited by the memory capacity and cannot simulate a quantum system with more qubits. This paper designs and implements a multi-GPU version of the QuEST simulator, which solves the problem of insufficient memory of single GPU and can simulate more qubits. Moreover, compared with the single CPU version, it can achieve 7~9x perfor- mance acceleration, and compared with the multi-CPU version, it achieves 3x performance acceleration.   




Key words: quantum computing, quantum circuit simulator, QuEST, multi-GPU, GPU memory