高效！启科量子线路模拟器 QuSprout 与 Amazon HPC 集成，赋能量子计算

news2024/11/28 8:47:36

KY1, Bertran Shao2, Adam Sun

Amazon HERO；2.开发者生态负责人；3.Solutions Architect2022.11.17，启科量子正式开源其内部团队研发的量子线路模拟器：QuSprout 软件。结合早前其开源的启科量子编程框架 QuTrunk，开发者可以基于启科的量子计算平台实现真正的量子编程。因为 QuSprout 的性能水平直接影响量子编程计算的效率，而 QuSprout 本身又严重依赖硬件平台。众所周知亚马逊云科技是全球著名云计算服务提供商，提供了丰富的硬件平台，故如何将 QuSprout 与亚马逊云科技高性能服务实现有机的融合，将变得非常有意义。今天我们会将 QuSprout 部署在亚马逊云科技高性能计算（HPC）平台上，并对比分析 QuSprout 在 CPU 以及 GPU 两种不同硬件平台的计算效率。

1、QuSprout 简介

QuSprout 是启科量子自主研发的一款免费、开源的量子计算模拟后端，用户在 QuTrunk 量子编程框架生成量子线路后，需要连接到 QuSprout 计算后端来进行模拟计算。

同时，QuSprout 支持多个量子线路的任务管理、MPI 多进程并行计算。目前支持 Kylin，Ubuntu，CentOS 操作系统，支持在 x86，arm64 体系上编译部署，（建议使用 KylinV10，Ubuntu22.04，CentOS7.9）

量子模拟方面，QuSprout 通过 RPC 与 QuTrunk 量子编程框架连接，主要负责量子线路的任务管理、

QuSprout 使用 C++ 作为宿主语言，目前已经实现40+个量子门，包括 H, CH, P, CP, R, CR, Rx, Ry, Rz等，为量子线路的计算提供了更高的性能和更强大的灵活性。

QuSprout 已经支持 CPU 和 GPU 两种类型的硬件平台，目前开源的版本为 CPU 版本，GPU 版本目前还在内测中，相信不久也会实现开源。

2、关于 Amazon HPC 平台

P3 实例：

Amazon EC2 P3 类型实例基于 NVIDIA Tesla V100 Tensor Core GPU，最多支持8 GPUs，可为用户提供高性能计算平台，支持基于CUDA的加速程序，通常用于 HPC 程序、机器学习等场景。

本文采用 p3.2xlarge 实例作为 QuSprout 底层 GPU 的硬件平台，它为单 GPU 显卡实例，显卡型号为 NVIDIA Tesla V100-SXM2，拥有640 Tensor Cores，5120 CUDA Cores，其双精度浮点（FP64）运算性能可达7TFLOPS，单精度（FP32）可达14TFLOPS，具体实例的详细参数如下表所示：

M5 实例：

M5 家族的实例属于计算、内存、带宽相对均衡的一种资源类型，可以用来作为 web 服务器，计算集群，游戏服务器等。拥有 Intel Xeon Platinum 8000 以及 AMD EPYC 7000 系列的 CPU 可供选择。

本文采用 m5.8xlarge 类型的实例作为 QuSprout 底层 CPU 的硬件平台，它的 CPU 类型为 Intel® Xeon® Platinum 8259CL CPU @ 2.50GHz，具体实例的详细参数如下表所示：

3、部署 QuSprout

首先在 Amazon Console 中申请创建 m5.8xlarge和p3.2xlarge 实例，待实例初始化成功后，可通过 SSH 远程连接到对应的 EC2 实例。然后从 Github 拉取最新的 QuSprout 代码，根据 README 文档，进行编译和安装即可，整个过程通过脚本执行，简单方便。

最后安装 QuTrunk，因为 QuSprout 是后端的量子线路模拟器，需要与前端量子编程框架 QuTrunk 联合使用。同样从 Github 拉取最新的 QuTrunk 代码，根据 README 文档，通过脚本安装即可。

注：本文测试为了避免网络带宽对计算效率的影响，选择将 QuTrunk 与 QuSprout 部署在同一个 EC2 实例上。通过在 QuTrunk 中配置 QuSprout 服务程序的 IP 和端口，即可实现两者的通信。

4、测试 QuSprout 性能

QuTrunk 和 QuSprout 部署完成后，就可以进行量子编程的开发了。此处我们以 grover 搜索算法为例，记录 QuSprout 分别在 m5.8xlarge（CPU）以及p3.2xlarge（GPU）两个平台上的运行效率，通过模拟不同数量的量子比特，统计相应平台的计算时间和资源消耗信息。具体内容如下所示：

4.1 grover 搜索算法源码：

"""Grover's search algorithm."""  
  
import math  
import random  
  
from numpy import pi  
  
from qutrunk.circuit import QCircuit  
from qutrunk.circuit.gates import Measure, All  
from qutrunk.circuit.ops import PLUS, QAA  
  
  
def run_grover(qubits=10, backend=None):  
    # Quantum qubits  
    num_qubits = qubits  
  
    # Number of amplitudes  
 num_elems = 2**num_qubits  
  
    # Count of iteration  
    num_reps = math.ceil(pi / 4 * math.sqrt(num_elems))  
    print("num_qubits:", num_qubits, "num_elems:", num_elems, "num_reps:", num_reps)  
  
    # Choose target state randomly  
    sol_elem = random.randint(0, num_elems - 1)  
    print(f"target state: |{str(sol_elem)}>")  
  
    # Create quantum circuit with local python simulator  
    circuit = QCircuit(backend=backend, resource=True)  
  
    # Allocate quantum qubits  
    qureg = circuit.allocate(num_qubits)  
  
    # Set inital amplitudes to plus state  
    PLUS * qureg  
  
    # Apply quantum operator(gates)  
    QAA(num_reps, sol_elem) * qureg  
  
    # Measure for all qubits  
    All(Measure) * qureg  
  
    # Run circuit in local simulator  
    res = circuit.run()  
  
    # Get measure result and print as int  
 out = res.get_outcome()  
    print("measure result: " + str(int(out, base=2)))  
  
    # Print quantum circuit resource information  
    circuit.show_resource()  
  
    # Print quantum circuit execution information  
    print(res.excute_info())  
  
    return circuit  
  
  
if __name__ == "__main__":  
    # Run locally  
    circuit = run_grover()