Acceleration and optimization of quantum computing simulator implemented on new Sunway supercomputer

doi:10.11772/j.issn.1001-9081.2022091456

Journal of Computer Applications ›› 2023, Vol. 43 ›› Issue (8): 2486-2492.DOI: 10.11772/j.issn.1001-9081.2022091456

Special Issue: 先进计算；第一届CCF量子计算大会 (CQCC 2022)

• Advanced computing • Previous Articles Next Articles

Acceleration and optimization of quantum computing simulator implemented on new Sunway supercomputer

Xinmin SHI¹, Yong LIU², Yaojian CHEN³, Jiawei SONG², Xin LIU²()

^1.School of Cyber Security，Information Engineering University，Zhengzhou Henan 450001，China
^2.National Research Center of Parallel Computer Engineering and Technology，Beijing 100190，China
^3.Department of Computer Science and Technology，Tsinghua University，Beijing 100084，China

Received:2022-09-29 Revised:2023-02-07 Accepted:2023-02-10 Online:2023-08-07 Published:2023-08-10
Contact: Xin LIU
About author:SHI Xinmin， born in 1997， M. S. candidate. His research interests include quantum simulator， parallel algorithm optimization.
LIU Yong， born in 1980， Ph. D.， associate research fellow. His research interests include parallel algorithm optimization， quantum computing simulation.
CHEN Yaojian， born in 1999， Ph. D. candidate. His research interests include parallel algorithm optimization， quantum computing simulation.
SONG Jiawei， born in 1989， engineer. His research interests include parallel algorithm optimization， quantum computing simulation.
Supported by:
National Science and Technology Major Project(2017-I-0004-0004)

基于新一代神威超算的量子计算模拟器加速和优化

史新民¹, 刘勇², 陈垚键³, 宋佳伟², 刘鑫²()

^1.信息工程大学网络空间安全学院, 郑州 450001
^2.国家并行计算机工程技术研究中心, 北京 100190
^3.清华大学计算机科学与技术系, 北京 100084

通讯作者: 刘鑫
作者简介:史新民（1997—），男，河北唐山人，硕士研究生，主要研究方向：量子模拟器、并行算法优化
刘勇（1980—），男，湖南邵阳人，副研究员，博士，主要研究方向：并行算法优化、量子计算模拟
陈垚键（1999—），男，广西桂林人，博士研究生，主要研究方向：并行算法优化、量子计算模拟
宋佳伟（1989—），男，江苏无锡人，工程师，主要研究方向：并行算法优化、量子计算模拟；
基金资助:
国家科技重大专项(2017?I?0004?0004)

Abstract

Abstract:

Two optimization methods for quantum simulator implemented on Sunway supercomputer were proposed aiming at the problems of gradual scaling of quantum hardware and insufficient classical simulation speed. Firstly， the tensor contraction operator library SWTT was reconstructed by improving the tensor transposition strategy and computation strategy， which improved the computing kernel efficiency of partial tensor contraction and reduced redundant memory access. Secondly， the balance between complexity and efficiency of path computation was achieved by the contraction path adjustment method based on data locality optimization. Test results show that the improvement method of operator library can improve the simulation efficiency of the "Sycamore" quantum supremacy circuit by 5.4% and the single-step tensor contraction efficiency by up to 49.7 times； the path adjustment method can improve the floating-point efficiency by about 4 times with the path computational complexity inflated by a factor of 2. The two optimization methods have the efficiencies of single-precision and mixed-precision floating-point operations for the simulation of Google’s 53-bit， 20-layer quantum chip random circuit with a million amplitude sampling improved from 3.98% and 1.69% to 18.48% and 7.42% respectively， and reduce the theoretical estimated simulation time from 470 s to 226 s for single-precision and 304 s to 134 s for mixed-precision， verifying that the two methods significantly improve the quantum computational simulation speed.

Key words: quantum simulator, Sunway supercomputer architecture, contraction path, tensor contraction operator library, tensor network contraction

摘要：

针对量子硬件规模逐步扩大、当下量子计算经典模拟速度不高的问题，提出了基于神威超算量子模拟器的两种优化方法。首先，通过改进张量转置策略和计算策略重新构建了张量收缩算子库SWTT，从而提高了部分张量收缩的计算内核效率并减少了冗余访存；其次，通过提高数据局部性的收缩路径调整方法实现了路径计算复杂度和计算效率之间的均衡。测试结果表明，该算子库改进方法可将“悬铃木”量子霸权电路模拟效率提升5.4%，单步张量收缩效率最高提升49.7倍；该路径调整方法可在路径计算复杂度膨胀2倍条件下提升约4倍的浮点效率。两种优化方法使神威超算整机模拟谷歌53量子比特20层量子芯片随机电路百万振幅采样的单精度和混合精度浮点运算效率分别从3.98%和1.69%提升至18.48%和7.42%，理论估计模拟时间从单精度的470 s降至226 s，混合精度的304 s降至134 s，证明两种方法大幅提高了量子计算模拟速度。

关键词: 量子模拟器, 神威超算体系结构, 收缩路径, 张量收缩算子库, 张量网络收缩

CLC Number:

TP38

Xinmin SHI, Yong LIU, Yaojian CHEN, Jiawei SONG, Xin LIU. Acceleration and optimization of quantum computing simulator implemented on new Sunway supercomputer[J]. Journal of Computer Applications, 2023, 43(8): 2486-2492.

史新民, 刘勇, 陈垚键, 宋佳伟, 刘鑫. 基于新一代神威超算的量子计算模拟器加速和优化[J]. 《计算机应用》唯一官方网站, 2023, 43(8): 2486-2492.

Figures/Tables 10

References 26

1	ARUTE F， ARYA K， BABBUSH R， et al. Quantum supremacy using a programmable superconducting processor［J］. Nature， 2019， 574（7779）：505-510. 10.1038/s41586-019-1666-5
2	ZHU Q L， CAO S R， CHEN F S， et al. Quantum computational advantage via 60-qubit 24-cycle random circuit sampling［J］. Science Bulletin， 2022， 67（3）：240-245. 10.1016/j.scib.2021.10.017
3	PEDNAULT E， GUNNELS J A， NANNICINI G， et al. Leveraging secondary storage to simulate deep 54-qubit sycamore circuits［EB/OL］. （2019-10-22）［2023-02-04］..
4	PRESKILL J. Quantum computing and the entanglement frontier［EB/OL］. （2012-11-10）［2023-02-04］..
5	ZLOKAPA A， VILLALONGA B， BOIXO S， et al. Boundaries of quantum supremacy via random circuit sampling［J］. npj Quantum Information， 2023， 9： No.36. 10.1038/s41534-023-00703-x
6	PAN F， ZHANG P. Simulating the Sycamore quantum supremacy circuits［EB/OL］. （2021-03-04）［2023-02-04］..
7	WRIGHT K， BECK K M， DEBNATH S， et al. Benchmarking an 11-qubit quantum computer［J］. Nature Communications， 2019， 10： No.5464. 10.1038/s41467-019-13534-2
8	PRESKILL J. Quantum computing in the NISQ era and beyond［J］. Quantum， 2018， 2： No.79. 10.22331/q-2018-08-06-79
9	LIU Y， LIU X， LI F， et al. Closing the "quantum supremacy" gap： achieving real-time simulation of a random quantum circuit using a new Sunway supercomputer［C］// Proceedings of the 2021 International Conference for High Performance Computing， Networking， Storage and Analysis. New York： ACM， 2021： No.3. 10.1145/3458817.3487399
10	ZHONG H S， WANG H， DENG Y H， et al. Quantum computational advantage using photons［J］. Science， 2020， 370（6523）： 1460-1463. 10.1126/science.abe8770
11	CROSS A. The IBM Q experience and QISKit open-source quantum computing software［C］// Proceedings of the APS March Meeting 2018. College Park， MD： American Physical Society， 2018： No.L58.3.
12	MARKOV I L， FATIMA A， ISAKOV S V， et al. Quantum supremacy is both closer and farther than it appears［EB/OL］. （2018-09-26）［2023-02-04］..
13	HÄNER T， STEIGER D S. 0.5 petabyte simulation of a 45-qubit quantum circuit［C］// Proceedings of the 2017 International Conference for High Performance Computing， Networking， Storage and Analysis. New York： ACM， 2017： No.33. 10.1145/3126908.3126947
14	BOIXO S， ISAKOV S V， SMELYANSKIY V N， et al. Simulation of low-depth quantum circuits as complex undirected graphical models［EB/OL］. （2018-01-19）［2023-02-04］..
15	WU X C， DI S， DASGUPTA E M， et al. Full-state quantum circuit simulation by using data compression［C］// Proceedings of the 2019 International Conference for High Performance Computing， Networking， Storage and Analysis. New York： ACM， 2019： No.80. 10.1145/3295500.3356155
16	LI R L， WU B J， YING M S， et al. Quantum supremacy circuit simulation on Sunway TaihuLight［J］. IEEE Transactions on Parallel and Distributed Systems， 2020， 31（4）： 805-816. 10.1109/tpds.2019.2947511
17	CHEN Z Y， ZHOU Q， XUE C， et al. 64-qubit quantum circuit simulation［J］. Science Bulletin， 2018， 63（15）： 964-971. 10.1016/j.scib.2018.06.007
18	PEDNAULT E， GUNNELS J A， NANNICINI G， et al. Pareto-efficient quantum circuit simulation using tensor contraction deferral［EB/OL］. （2020-08-27）［2023-02-04］..
19	MARKOV I L， SHI Y Y. Simulating quantum computation by contracting tensor networks［J］. SIAM Journal on Computing， 2008， 38（3）：963-981. 10.1137/050644756
20	HUANG C， ZHANG F， NEWMAN M， et al. Classical simulation of quantum supremacy circuits［EB/OL］. （2020-05-14）［2023-02-04］..
21	KALACHEV G， PANTELEEV P， YUNG M H. Multi-tensor contraction for XEB verification of quantum circuits［EB/OL］. （2022-05-19）［2023-02-04］..
22	GUO C， LIU Y， XIONG M， et al. General-purpose quantum circuit simulator with projected entangled-pair states and the quantum supremacy frontier［J］. Physical Review Letters， 2019， 123（19）： No.190501. 10.1103/physrevlett.123.190501
23	VILLALONGA B， BOIXO S， NELSON B， et al. A flexible high-performance simulator for verifying and benchmarking quantum circuits implemented on real hardware［J］. npj Quantum Information， 2019， 5： No.86. 10.1038/s41534-019-0196-1
24	CHEN Y J， LIU Y， SHI X M， et al. Lifetime-based method for quantum simulation on a new Sunway supercomputer［C］// Proceedings of the 28th ACM SIGPLAN Annual Symposium on Principles and Practice of Parallel Programming. New York： ACM， 2023： 148-159. 10.1145/3572848.3577529
25	GRAY J， KOURTIS S. Hyper-optimized tensor network contraction［J］. Quantum， 2021， 5： No.410. 10.22331/q-2021-03-15-410
26	LIU Y， CHEN Y J， GUO C， et al. Validating quantum-supremacy experiments with exact and fast tensor network contraction［EB/OL］. （2022-12-09）［2023-02-04］..

模拟编号	路径信息		子任务模拟时长（单精度）/s
模拟编号	总计算复杂度	子任务计算复杂度	子任务模拟时长（单精度）/s
1（优化前）	3.44×10¹⁸	8.20×10¹¹	11.76
2（优化后）	3.44×10¹⁸	8.20×10¹¹	11.12

模拟编号	路径信息		子任务模拟时长（单精度）/s
模拟编号	总计算复杂度	子任务计算复杂度	子任务模拟时长（单精度）/s
1（优化前）	3.44×10¹⁸	8.20×10¹¹	11.76
2（优化后）	3.44×10¹⁸	8.20×10¹¹	11.12

路径	浮点效率/%		总模拟时长/s
路径	单精度	混合精度	单精度	混合精度
初始路径	4.21	1.69	439.2	260.90
一轮调整	12.38	4.97	260.7	154.86
两轮调整	18.48	7.42	225.9	134.20

路径	浮点效率/%		总模拟时长/s
路径	单精度	混合精度	单精度	混合精度
初始路径	4.21	1.69	439.2	260.90
一轮调整	12.38	4.97	260.7	154.86
两轮调整	18.48	7.42	225.9	134.20

路径	子任务计算复杂度	子任务收缩时长/s	单精度浮点效率/%
2（初始路径）	9.23×10¹¹	14.32	3.68
2（一轮调整）	1.44×10¹²	8.26	9.97
2（两轮调整）	2.07×10¹²	7.74	15.28
3（初始路径）	1.05×10¹²	14.90	4.02
3（一轮调整）	1.75×10¹²	8.78	11.41
3（两轮调整）	2.19×10¹²	7.88	15.86
4（初始路径）	1.12×10¹²	15.20	4.22
4（一轮调整）	2.01×10¹²	9.94	11.55
4（两轮调整）	2.19×10¹²	9.94	13.68
4（三轮调整）	2.38×10¹²	8.02	15.61

Acceleration and optimization of quantum computing simulator implemented on new Sunway supercomputer

基于新一代神威超算的量子计算模拟器加速和优化

RichHTML

PDF

Knowledge

Abstract

Cite this article

share this article

Figures/Tables 10

References 26

Related Articles 15

Recommended Articles

Metrics

[1]	LI Zhuoran, LI Hua, WANG Tong, JIANG Chaozhe. Lightweight human pose estimation based on fused feature state space model#br# #br# [J]. Journal of Computer Applications, 0, (): 0-0.
[2]	Han WANG, Yuan WAN, Dong WANG, Yiming DING. Robust weight matrix combination selection method of broad learning system [J]. Journal of Computer Applications, 2024, 44(10): 3032-3038.
[3]	Jing QIN, Zhiguang QIN, Fali LI, Yueheng PENG. Diagnosis of major depressive disorder based on probabilistic sparse self-attention neural network [J]. Journal of Computer Applications, 2024, 44(9): 2970-2974.
[4]	Fan YANG, Yao ZOU, Mingzhi ZHU, Zhenwei MA, Dawei CHENG, Changjun JIANG. Credit card fraud detection model based on graph attention Transformation neural network [J]. Journal of Computer Applications, 2024, 44(8): 2634-2642.
[5]	Zhi ZHANG, Xin LI, Naifu YE, Kaixi HU. DKP： defending against model stealing attacks based on dark knowledge protection [J]. Journal of Computer Applications, 2024, 44(7): 2080-2086.
[6]	Feiyu ZHAI, Handa MA. Hybrid classical-quantum classification model based on DenseNet [J]. Journal of Computer Applications, 2024, 44(6): 1905-1910.
[7]	. Adaptive Collaborative Optimization for Vehicle Recognition Algorithm based on FIPA Model [J]. Journal of Computer Applications, 0, (): 0-0.
[8]	. Lightweight human pose estimation based on decoupled attention and ghost convolution [J]. Journal of Computer Applications, 0, (): 0-0.
[9]	Haoran WANG, Dan YU, Yuli YANG, Yao MA, Yongle CHEN. Domain transfer intrusion detection method for unknown attacks on industrial control systems [J]. Journal of Computer Applications, 2024, 44(4): 1158-1165.
[10]	Sihang CHEN, Aiwen JIANG, Zhaoyang CUI, Mingwen WANG. Multi-channel multi-step integration model for generative visual dialogue [J]. Journal of Computer Applications, 2024, 44(1): 39-46.
[11]	Jiahang PAN, Jiahang WANG, Zhan SHI, Yingkun XU, Yongan XU, Xiaoxia HUANG. FPD-ViT： Facial pain detection Vision Transformer [J]. Journal of Computer Applications, 2023, 43(S2): 77-82.
[12]	Lei LIU, Peng WU, Kai XIE, Beizhi CHENG, Guanqun SHENG. Parking space detection method based on self-supervised learning HOG prediction auxiliary task [J]. Journal of Computer Applications, 2023, 43(12): 3933-3940.
[13]	Chuyuan WEI, Mengke WANG, Chuanhao HU, Guangqi ZHANG. Deep review attention neural network model for enhancing explainability of recommendation system [J]. Journal of Computer Applications, 2023, 43(11): 3443-3448.
[14]	Guohuan HE, Jiangping ZHU. WT-U-Net++： surface defect detection network based on wavelet transform [J]. Journal of Computer Applications, 2023, 43(10): 3260-3266.
[15]	Xiaomin ZHOU, Fei TENG, Yi ZHANG. Automatic international classification of diseases coding model based on meta-network [J]. Journal of Computer Applications, 2023, 43(9): 2721-2726.