Aws backend

.github/workflows/unit_test.yml

@@ -45,7 +45,7 @@ jobs:
         run: poetry install --no-interaction --no-root
         if: steps.cache.outputs.cache-hit != 'true'
       - name: Install project
-        run: poetry install --no-interaction --with test
+        run: poetry install --no-interaction --with test --all-extras
       - name: Install task runner
         run: pip install poethepoet
       - name: Lint

.gitignore

@@ -157,5 +157,5 @@ cython_debug/
 #  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
-#.idea/
+.idea/
 .devenv

pyproject.toml

@@ -21,10 +21,11 @@ packages = [{ include = "qibo_cloud_backends", from = "src" }]
 
 [tool.poetry.dependencies]
 python = ">=3.9,<3.12"
-qibo = ">=0.2.4"
-qibo_client = ">=0.0.3"
+qibo = "^0.2.8"
+qibo_client = "^0.0.5"
 qiskit_ibm_runtime = ">=0.17"
 qiskit_ibm_provider = ">=0.8.0"
+amazon-braket-sdk = { version = "^1.77.3", optional = true }
 
 [tool.poetry.group.test.dependencies]
 pytest = "^7.4.3"
@@ -38,11 +39,13 @@ sphinx-copybutton = "^0.5.2"
 nbsphinx = "^0.9.3"
 furo = "^2023.9.10"
 
-
 [tool.poetry.group.dev.dependencies]
 ipython = "^7"
 pdbpp = "^0.10.3"
 
+[tool.poetry.extras]
+aws = ["amazon-braket-sdk"]
+
 [tool.poe.tasks]
 test = "pytest"
 lint = "pylint --errors-only src"

requirements-aws.txt

@@ -0,0 +1 @@
+amazon-braket-sdk

src/qibo_cloud_backends/aws_client.py

@@ -0,0 +1,125 @@
+from qibo.backends import NumpyBackend
+from qibo.config import raise_error
+from qibo.result import MeasurementOutcomes, QuantumState
+
+import re
+import importlib.util
+import sys
+
+# # Import Qiskit packages for transpiler
+# from qiskit import transpile
+# from qiskit import QuantumCircuit
+# from qiskit.transpiler import CouplingMap
+# from qiskit.circuit.random import random_circuit
+# from qiskit import qasm2
+
+from qibo import Circuit as QiboCircuit
+
+from braket.aws import AwsDevice, AwsQuantumTask
+from braket.circuits import Gate, observables
+from braket.circuits import Circuit as BraketCircuit
+from braket.devices import Devices, LocalSimulator
+
+_QASM_BRAKET_GATES = {
+    "id": "i",
+    "cx": "cnot",
+    "sx": "v",
+    "sxdg": "vi",
+    "sdg": "si",
+    "tdg": "ti",
+}
+
+class BraketClientBackend(NumpyBackend):
+    """Backend for the remote execution of AWS circuits on the AWS backends.
+
+    Args:
+        device_arn (str | None): The ARN of the Braket device.
+            If `None`, instantiates the `LocalSimulator("default")`.
+    """
+
+    def __init__(self, device=None, verbatim_circuit=False):
+        """Initializes BraketBackend.
+
+        Args:
+            device (str): Default device is Braket's statevector LocalSimulator, LocalSimulator("default").
+                Other devices are Braket's density matrix simulator, LocalSimulator("braket_dm"), or any other
+                QPUs.
+            verbatim_circuit (bool): If `True`, wrap the Braket circuit in a verbatim box to run it on the QPU
+                without any transpilation. Defaults to `False`.
+        """
+
+        super().__init__()
+
+        self.verbatim_circuit = verbatim_circuit
+
+        if device is None:
+            self.device = LocalSimulator("default")
+            # self.device = LocalSimulator("braket_dm")
+        else:
+            self.device = device
+        self.name = "aws"
+
+    def remove_qelib1_inc(self, qasm_string):
+        """To remove the 'includes qe1lib.inc' from the OpenQASM string.
+
+        Args: 
+            qasm_code (OpenQASM circuit, str): circuit given in the OpenQASM format.
+
+        Returns:
+            qasm_code (OpenQASM circuit, str): circuit given in the OpenQASM format.
+        """
+
+        # Remove the "include "qelib1.inc";\n" line
+        modified_code = re.sub(r'include\s+"qelib1.inc";\n', '', qasm_string)
+        return modified_code
+
+    def qasm_convert_gates(self, qasm_code):
+        """To replace the notation for certain gates in OpenQASM
+
+        Args: 
+            qasm_code (OpenQASM circuit, str): circuit given in the OpenQASM format.
+
+        Returns:
+            qasm_code (OpenQASM circuit, str): circuit given in the OpenQASM format.
+        """
+
+        lines = qasm_code.split('\n')
+        modified_code = ""
+        for line in lines:
+            for key in _QASM_BRAKET_GATES:
+                if key in line:
+                    line = line.replace(key, _QASM_BRAKET_GATES[key])
+                    break
+            modified_code += line + '\n'
+        return modified_code
+
+    def execute_circuit(self,
+                        circuit,
+                        nshots=1000):
+        """Executes a circuit (Qibo or Qibo-transpiled-to-Braket) on an AWS Braket device. The device defaults to the LocalSimulator().
+
+        Args:
+            circuit (quantum circuit): Either qibo.models.Circuit or braket.circuits.Circuit to execute on the Braket device.
+            nshots (int): Total number of shots.
+
+        Returns:
+            Measurement outcomes (qibo.measurement.MeasurementOutcomes): The outcome of the circuit execution.
+        """
+
+        measurements = circuit.measurements
+        if not measurements:
+            raise_error(RuntimeError, "No measurement found in the provided circuit.")
+        nqubits = circuit.nqubits
+        circuit_qasm = circuit.to_qasm()
+        circuit_qasm = self.remove_qelib1_inc(circuit_qasm)
+        circuit_qasm = self.qasm_convert_gates(circuit_qasm)
+        braket_circuit = BraketCircuit.from_ir(circuit_qasm)
+
+        # Wrap verbatim box, execute:
+        if self.verbatim_circuit:
+            braket_circuit = BraketCircuit().add_verbatim_box(braket_circuit)
+        result = self.device.run(braket_circuit, shots=nshots).result()
+        samples = result.measurements
+        return MeasurementOutcomes(
+            measurements=measurements, backend=self, samples=samples, nshots=nshots
+        )

tests/test_aws.py

@@ -0,0 +1,28 @@
+from qibo import Circuit
+from qibo import gates
+import numpy as np
+from qibo.quantum_info import random_clifford
+from qibo_cloud_backends import aws_client
+from qibo.backends import NumpyBackend
+
+NP_BACKEND = NumpyBackend()
+
+
+def test_aws_client_backend():
+    circuit_qibo = random_clifford(3, backend=NP_BACKEND)
+    circuit_qibo.add(gates.M(0, 2))
+
+    # Local simulator test, does not cost money
+    client = aws_client.BraketClientBackend()
+    # AWS device tests, cost money
+    # AWS = aws_client.BraketClientBackend(device = AwsDevice("arn:aws:braket:eu-west-2::device/qpu/oqc/Lucy"))
+    # AWS = aws_client.BraketClientBackend(device = AwsDevice("arn:aws:braket:::device/quantum-simulator/amazon/sv1"))
+
+    local_res = NP_BACKEND.execute_circuit(circuit_qibo)
+    remote_res = client.execute_circuit(circuit_qibo)
+
+    NP_BACKEND.assert_allclose(
+        local_res.probabilities(qubits=[0, 2]),
+        remote_res.probabilities(qubits=[0, 2]),
+        atol=1e-1,
+    )

tests/test_backends.py

@@ -11,17 +11,17 @@
 QIBO_TK = os.environ.get("QIBO_CLIENT_TII_TOKEN")
 
 
-def test_qiskit_client_backend():
-    c = random_clifford(3, backend=NP_BACKEND)
-    c.add(gates.M(0, 2))
-    client = QiskitClientBackend(
-        token=QISKIT_TK, provider="ibm-q", platform="ibmq_qasm_simulator"
-    )
-    local_res = NP_BACKEND.execute_circuit(c)
-    remote_res = client.execute_circuit(c)
-    NP_BACKEND.assert_allclose(
-        local_res.probabilities(qubits=[0, 2]), remote_res.probabilities(), atol=1e-1
-    )
+# def test_qiskit_client_backend():
+#     c = random_clifford(3, backend=NP_BACKEND)
+#     c.add(gates.M(0, 2))
+#     client = QiskitClientBackend(
+#         token=QISKIT_TK, provider="ibm-q", platform="ibmq_qasm_simulator"
+#     )
+#     local_res = NP_BACKEND.execute_circuit(c)
+#     remote_res = client.execute_circuit(c)
+#     NP_BACKEND.assert_allclose(
+#         local_res.probabilities(qubits=[0, 2]), remote_res.probabilities(), atol=1e-1
+#     )
 
 
 def test_qibo_client_backend():
@@ -35,3 +35,22 @@ def test_qibo_client_backend():
         remote_res.probabilities(qubits=[0, 2]),
         atol=1e-1,
     )
+
+# def test_aws_client_backend():
+#     circuit_qibo = random_clifford(3, backend=NP_BACKEND)
+#     circuit_qibo.add(gates.M(0, 2))
+# 
+#     # Local simulator test, does not cost money
+#     client = aws_client.BraketClientBackend()
+#     # AWS device tests, cost money
+#     # AWS = aws_client.BraketClientBackend(device = AwsDevice("arn:aws:braket:eu-west-2::device/qpu/oqc/Lucy"))
+#     # AWS = aws_client.BraketClientBackend(device = AwsDevice("arn:aws:braket:::device/quantum-simulator/amazon/sv1"))
+# 
+#     local_res = NP_BACKEND.execute_circuit(circuit_qibo)
+#     remote_res = client.execute_circuit(circuit_qibo)
+# 
+#     NP_BACKEND.assert_allclose(
+#         local_res.probabilities(qubits=[0, 2]),
+#         remote_res.probabilities(qubits=[0, 2]),
+#         atol=1e-1,
+#     )