CUDA extension - CuYao

Tutorial

CuYao is a CUDA extension of Yao, which allows you to run Yao circuits on GPU. The usage of CuYao is similar to Yao, but with some extra APIs to upload and download registers between CPU and GPU:

• cu(reg) to upload a registe reg to GPU, and
• cpu(cureg) to download a register cureg from GPU to CPU.

julia> using Yao, CUDA

# create a register on GPU
julia> cureg = rand_state(9; nbatch=1000) |> cu;   # or `curand_state(9; nbatch=1000)`.

# run a circuit on GPU
julia> cureg |> put(9, 2=>Z);

# measure the register on GPU
julia> measure!(cureg)
1000-element CuArray{DitStr{2, 9, Int64}, 1, CUDA.Mem.DeviceBuffer}:
 110110100 ₍₂₎
 000100001 ₍₂₎
 111111001 ₍₂₎
             ⋮
 010001101 ₍₂₎
 000100110 ₍₂₎

# download the register to CPU
julia> reg = cureg |> cpu;

Features

Supported gates:

• general U(N) gate
• general U(1) gate
• X, Y, Z gate
• T, S gate
• SWAP gate
• control gates

Supported register operations:

• measure!, measurereset!, measureremove!, select
• appendqudits!, appendqubits!
• insertqudit!, insertqubits!
• focus!, relax!
• join
• density_matrix
• fidelity
• expect

Autodiff:

• autodiff is supported when the only parameterized gates are rotation gates in a circuit.

API

cpu(cureg)

curand_state([T=ComplexF64], n::Int; nbatch=1)

cuzero_state([T=ComplexF64], n::Int; nbatch=1)

cuproduct_state([T=ComplexF64], total::Int, bit_config::Integer; nbatch=NoBatch())

cuuniform_state([T=ComplexF64], n::Int; nbatch=1)

cughz_state([T=ComplexF64], n::Int; nbatch=1)