qiskit/test/benchmarks/qasm3_exporter.py

84 lines
2.1 KiB
Python

# This code is part of Qiskit.
#
# (C) Copyright IBM 2024
#
# This code is licensed under the Apache License, Version 2.0. You may
# obtain a copy of this license in the LICENSE.txt file in the root directory
# of this source tree or at http://www.apache.org/licenses/LICENSE-2.0.
#
# Any modifications or derivative works of this code must retain this
# copyright notice, and modified files need to carry a notice indicating
# that they have been altered from the originals.
# pylint: disable=missing-docstring
# pylint: disable=attribute-defined-outside-init
from qiskit.circuit import Parameter
from qiskit import QuantumCircuit
from qiskit import qasm3
from .utils import random_circuit
class RandomBenchmarks:
params = ([20], [256, 1024], [0, 42])
param_names = ["n_qubits", "depth", "seed"]
timeout = 300
def setup(self, n_qubits, depth, seed):
self.circuit = random_circuit(
n_qubits,
depth,
measure=True,
conditional=True,
reset=True,
seed=seed,
max_operands=3,
)
def time_dumps(self, _, __, ___):
qasm3.dumps(self.circuit)
class CustomGateBenchmarks:
params = ([200], [100])
param_names = ["n_qubits", "depth"]
timeout = 300
def setup(self, n_qubits, depth):
custom_gate = QuantumCircuit(2, name="custom_gate")
custom_gate.h(0)
custom_gate.x(1)
qc = QuantumCircuit(n_qubits)
for _ in range(depth):
for i in range(n_qubits - 1):
qc.append(custom_gate.to_gate(), [i, i + 1])
self.circuit = qc
def time_dumps(self, _, __):
qasm3.dumps(self.circuit)
class ParameterizedBenchmarks:
params = ([20, 50], [1, 5, 10])
param_names = ["n_qubits", "n_params"]
timeout = 300
def setup(self, n_qubits, n_params):
qc = QuantumCircuit(n_qubits)
params = [Parameter(f"angle{i}") for i in range(n_params)]
for n in range(n_qubits - 1):
for i in params:
qc.rx(i, n)
self.circuit = qc
def time_dumps(self, _, __):
qasm3.dumps(self.circuit)