Rebasing initial implementation

.pre-commit-config.yaml

@@ -37,5 +37,6 @@ repos:
       additional_dependencies: [
         pytest,
         pytket==1.20.1,
+        pytket-quantinuum==0.23.0,
         types-setuptools,
       ]

pytket/phir/api.py

@@ -0,0 +1,32 @@
+from pytket.circuit import Circuit
+
+from .qtm_machine import QtmMachine
+from .rebasing.rebaser import rebase_to_qtm_machine
+from .sharding.sharder import Sharder
+
+
+def pytket_to_phir(
+    circuit: Circuit,
+    qtm_machine: QtmMachine | None = None,
+) -> str:
+    """Converts a pytket circuit into its PHIR representation.
+
+    This can optionally include rebasing against a Quantinuum machine architecture.
+
+    :param circuit: Circuit object to be converted
+    :param qtm_machine: (Optional) Quantinuum machine architecture to rebase against
+
+    Returns:
+    -------
+        PHIR JSON as a str
+    """
+    if qtm_machine:
+        circuit = rebase_to_qtm_machine(circuit, qtm_machine.value)
+
+    sharder = Sharder(circuit)
+    shards = sharder.shard()
+
+    # TODO: Pass shards[] into placement, routing, etc
+    # TODO: Convert to PHIR JSON spec and return
+
+    return str(shards)  # Just returning fake string for now

pytket/phir/qtm_machine.py

@@ -0,0 +1,9 @@
+from enum import Enum
+
+
+class QtmMachine(Enum):
+    """Available machine architectures."""
+
+    H1_1 = "H1-1"
+    H1_2 = "H1-2"
+    H2_1 = "H2-1"

pytket/phir/rebasing/rebaser.py

@@ -0,0 +1,18 @@
+from pytket.circuit import Circuit
+from pytket.extensions.quantinuum.backends.api_wrappers import QuantinuumAPIOffline
+from pytket.extensions.quantinuum.backends.quantinuum import (
+    QuantinuumBackend,
+)
+
+
+def rebase_to_qtm_machine(circuit: Circuit, qtm_machine: str) -> Circuit:
+    """Rebases a circuit's gate to the gate set appropriate for the given machine."""
+    qapi_offline = QuantinuumAPIOffline()
+    backend = QuantinuumBackend(
+        device_name=qtm_machine,
+        machine_debug=False,
+        api_handler=qapi_offline,  # type: ignore [arg-type]
+    )
+    # Optimization level 0 includes rebasing and little else
+    # see: https://cqcl.github.io/pytket-quantinuum/api/#default-compilation
+    return backend.get_compiled_circuit(circuit, 0)

requirements.txt

@@ -5,4 +5,5 @@ pre-commit==3.4.0
 pytest==7.4.2
 ruff==0.0.292
 pytket==1.20.1
+pytket-quantinuum==0.23.0
 wheel==0.41.2

tests/test_api.py

@@ -0,0 +1,24 @@
+from pytket.phir.api import pytket_to_phir
+from pytket.phir.qtm_machine import QtmMachine
+
+from .sample_data import QasmFiles, get_qasm_as_circuit
+
+
+class TestApi:
+    def test_pytket_to_phir_no_machine(self) -> None:
+        circuit = get_qasm_as_circuit(QasmFiles.baby)
+
+        phir = pytket_to_phir(circuit)
+
+        # TODO: Make this test more valuable once PHIR is actually returned
+        print(phir)
+        assert len(phir) > 0
+
+    def test_pytket_to_phir_h1_1(self) -> None:
+        circuit = get_qasm_as_circuit(QasmFiles.baby)
+
+        phir = pytket_to_phir(circuit, QtmMachine.H1_1)
+
+        # TODO: Make this test more valuable once PHIR is actually returned
+        print(phir)
+        assert len(phir) > 0

tests/test_rebaser.py

@@ -0,0 +1,23 @@
+from pytket.circuit import Circuit, OpType
+from pytket.phir.rebasing.rebaser import rebase_to_qtm_machine
+
+from .sample_data import QasmFiles, get_qasm_as_circuit
+
+EXPECTED_GATES = [
+    OpType.Measure,
+    OpType.Rz,
+    OpType.PhasedX,
+    OpType.ZZPhase,
+]
+
+
+class TestRebaser:
+    def test_rebaser_happy_path_arc1a(self) -> None:
+        circ = get_qasm_as_circuit(QasmFiles.baby)
+        rebased: Circuit = rebase_to_qtm_machine(circ, "H1-1")
+
+        print(rebased)
+        for command in rebased.get_commands():
+            print(command)
+            if command.op.is_gate():
+                assert command.op.type in EXPECTED_GATES