example_partition.py

Go to the documentation of this file.

from squander.partitioning.partition import PartitionCircuitQasm
from squander.partitioning.tools import get_qubits


def ExamplePartition():
    """
    Example usage of circuit partitioning and printing partition info
    """
    filename = "examples/partitioning/qasm_samples/heisenberg-16-20.qasm"

    max_partition_size = 4
    partitioned_circuit, parameters = PartitionCircuitQasm(
        filename, max_partition_size, "kahn"
    )

    print(f"{filename} Partitions ({len(partitioned_circuit.get_Gates())}):")
    total = 0

    for i, partition in enumerate(partitioned_circuit.get_Gates()):
        num_gates = len(partition.get_Gates())
        qubits = set.union(*(get_qubits(gate) for gate in partition.get_Gates()))
        total += num_gates
        print(f"#{i + 1}: {num_gates} {qubits}")

    print(f"{filename} Params {parameters}")

    print(f"Total gates: {total}\n")


if __name__ == "__main__":
    ExamplePartition()