main.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Local Mini-Circuit-Cutting"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Found solution using 2 cuts with a sampling overhead of 14.128980038696172.\n",
      "Gate Cut at circuit instruction index 13\n",
      "Gate Cut at circuit instruction index 19\n",
      "Sampling overhead: 14.128980038696168\n",
      "68 total subexperiments to run on backend.\n",
      "Running subcircuit 0...\n",
      "Subcircuit 0 took 0.004372119903564453 seconds to run.\n",
      "Running subcircuit 1...\n",
      "Subcircuit 1 took 0.0012557506561279297 seconds to run.\n",
      "Running subcircuit 2...\n",
      "Subcircuit 2 took 0.0015749931335449219 seconds to run.\n",
      "Running subcircuit 3...\n",
      "Subcircuit 3 took 0.0012869834899902344 seconds to run.\n",
      "Running subcircuit 4...\n",
      "Subcircuit 4 took 0.0011260509490966797 seconds to run.\n",
      "Running subcircuit 5...\n",
      "Subcircuit 5 took 0.0011637210845947266 seconds to run.\n",
      "Running subcircuit 6...\n",
      "Subcircuit 6 took 0.001251220703125 seconds to run.\n",
      "Running subcircuit 7...\n",
      "Subcircuit 7 took 0.0010752677917480469 seconds to run.\n",
      "Running subcircuit 8...\n",
      "Subcircuit 8 took 0.0011510848999023438 seconds to run.\n",
      "Running subcircuit 9...\n",
      "Subcircuit 9 took 0.0011818408966064453 seconds to run.\n",
      "Running subcircuit 10...\n",
      "Subcircuit 10 took 0.0013828277587890625 seconds to run.\n",
      "Running subcircuit 11...\n",
      "Subcircuit 11 took 0.0012698173522949219 seconds to run.\n",
      "Running subcircuit 12...\n",
      "Subcircuit 12 took 0.001077890396118164 seconds to run.\n",
      "Running subcircuit 13...\n",
      "Subcircuit 13 took 0.0011792182922363281 seconds to run.\n",
      "Running subcircuit 14...\n",
      "Subcircuit 14 took 0.0029058456420898438 seconds to run.\n",
      "Running subcircuit 15...\n",
      "Subcircuit 15 took 0.0011038780212402344 seconds to run.\n",
      "Running subcircuit 16...\n",
      "Subcircuit 16 took 0.0010991096496582031 seconds to run.\n",
      "Running subcircuit 17...\n",
      "Subcircuit 17 took 0.0016160011291503906 seconds to run.\n",
      "Running subcircuit 18...\n",
      "Subcircuit 18 took 0.0011239051818847656 seconds to run.\n",
      "Running subcircuit 19...\n",
      "Subcircuit 19 took 0.001110076904296875 seconds to run.\n",
      "Running subcircuit 20...\n",
      "Subcircuit 20 took 0.0012149810791015625 seconds to run.\n",
      "Running subcircuit 21...\n",
      "Subcircuit 21 took 0.0011000633239746094 seconds to run.\n",
      "Running subcircuit 22...\n",
      "Subcircuit 22 took 0.0012407302856445312 seconds to run.\n",
      "Running subcircuit 23...\n",
      "Subcircuit 23 took 0.0010938644409179688 seconds to run.\n",
      "Running subcircuit 24...\n",
      "Subcircuit 24 took 0.0011861324310302734 seconds to run.\n",
      "Running subcircuit 25...\n",
      "Subcircuit 25 took 0.0013210773468017578 seconds to run.\n",
      "Running subcircuit 26...\n",
      "Subcircuit 26 took 0.0010890960693359375 seconds to run.\n",
      "Running subcircuit 27...\n",
      "Subcircuit 27 took 0.0012469291687011719 seconds to run.\n",
      "Running subcircuit 28...\n",
      "Subcircuit 28 took 0.0010769367218017578 seconds to run.\n",
      "Running subcircuit 29...\n",
      "Subcircuit 29 took 0.0011029243469238281 seconds to run.\n",
      "Running subcircuit 30...\n",
      "Subcircuit 30 took 0.0021369457244873047 seconds to run.\n",
      "Running subcircuit 31...\n",
      "Subcircuit 31 took 0.0013298988342285156 seconds to run.\n",
      "Running subcircuit 32...\n",
      "Subcircuit 32 took 0.0012259483337402344 seconds to run.\n",
      "Running subcircuit 33...\n",
      "Subcircuit 33 took 0.0013179779052734375 seconds to run.\n",
      "Running subcircuit 0...\n",
      "Subcircuit 0 took 0.0013949871063232422 seconds to run.\n",
      "Running subcircuit 1...\n",
      "Subcircuit 1 took 0.0022220611572265625 seconds to run.\n",
      "Running subcircuit 2...\n",
      "Subcircuit 2 took 0.001138925552368164 seconds to run.\n",
      "Running subcircuit 3...\n",
      "Subcircuit 3 took 0.0014510154724121094 seconds to run.\n",
      "Running subcircuit 4...\n",
      "Subcircuit 4 took 0.0012960433959960938 seconds to run.\n",
      "Running subcircuit 5...\n",
      "Subcircuit 5 took 0.0013589859008789062 seconds to run.\n",
      "Running subcircuit 6...\n",
      "Subcircuit 6 took 0.0042269229888916016 seconds to run.\n",
      "Running subcircuit 7...\n",
      "Subcircuit 7 took 0.0035409927368164062 seconds to run.\n",
      "Running subcircuit 8...\n",
      "Subcircuit 8 took 0.0013871192932128906 seconds to run.\n",
      "Running subcircuit 9...\n",
      "Subcircuit 9 took 0.0018508434295654297 seconds to run.\n",
      "Running subcircuit 10...\n",
      "Subcircuit 10 took 0.002119302749633789 seconds to run.\n",
      "Running subcircuit 11...\n",
      "Subcircuit 11 took 0.0025589466094970703 seconds to run.\n",
      "Running subcircuit 12...\n",
      "Subcircuit 12 took 0.009794950485229492 seconds to run.\n",
      "Running subcircuit 13...\n",
      "Subcircuit 13 took 0.0025157928466796875 seconds to run.\n",
      "Running subcircuit 14...\n",
      "Subcircuit 14 took 0.0030379295349121094 seconds to run.\n",
      "Running subcircuit 15...\n",
      "Subcircuit 15 took 0.00368499755859375 seconds to run.\n",
      "Running subcircuit 16...\n",
      "Subcircuit 16 took 0.003704071044921875 seconds to run.\n",
      "Running subcircuit 17...\n",
      "Subcircuit 17 took 0.002954244613647461 seconds to run.\n",
      "Running subcircuit 18...\n",
      "Subcircuit 18 took 0.0011920928955078125 seconds to run.\n",
      "Running subcircuit 19...\n",
      "Subcircuit 19 took 0.0019481182098388672 seconds to run.\n",
      "Running subcircuit 20...\n",
      "Subcircuit 20 took 0.0013010501861572266 seconds to run.\n",
      "Running subcircuit 21...\n",
      "Subcircuit 21 took 0.0013949871063232422 seconds to run.\n",
      "Running subcircuit 22...\n",
      "Subcircuit 22 took 0.0013837814331054688 seconds to run.\n",
      "Running subcircuit 23...\n",
      "Subcircuit 23 took 0.00128173828125 seconds to run.\n",
      "Running subcircuit 24...\n",
      "Subcircuit 24 took 0.0012159347534179688 seconds to run.\n",
      "Running subcircuit 25...\n",
      "Subcircuit 25 took 0.0011861324310302734 seconds to run.\n",
      "Running subcircuit 26...\n",
      "Subcircuit 26 took 0.0029141902923583984 seconds to run.\n",
      "Running subcircuit 27...\n",
      "Subcircuit 27 took 0.004910945892333984 seconds to run.\n",
      "Running subcircuit 28...\n",
      "Subcircuit 28 took 0.0012447834014892578 seconds to run.\n",
      "Running subcircuit 29...\n",
      "Subcircuit 29 took 0.0013518333435058594 seconds to run.\n",
      "Running subcircuit 30...\n",
      "Subcircuit 30 took 0.0012531280517578125 seconds to run.\n",
      "Running subcircuit 31...\n",
      "Subcircuit 31 took 0.0014851093292236328 seconds to run.\n",
      "Running subcircuit 32...\n",
      "Subcircuit 32 took 0.0024747848510742188 seconds to run.\n",
      "Running subcircuit 33...\n",
      "Subcircuit 33 took 0.0017521381378173828 seconds to run.\n",
      "Running subcircuit 0...\n",
      "Subcircuit 0 took 0.0011358261108398438 seconds to run.\n",
      "Running subcircuit 1...\n",
      "Subcircuit 1 took 0.002315044403076172 seconds to run.\n",
      "Running subcircuit 2...\n",
      "Subcircuit 2 took 0.003565073013305664 seconds to run.\n",
      "Running subcircuit 3...\n",
      "Subcircuit 3 took 0.0011479854583740234 seconds to run.\n",
      "Running subcircuit 4...\n",
      "Subcircuit 4 took 0.0012080669403076172 seconds to run.\n",
      "Running subcircuit 5...\n",
      "Subcircuit 5 took 0.0010428428649902344 seconds to run.\n",
      "Running subcircuit 6...\n",
      "Subcircuit 6 took 0.001277923583984375 seconds to run.\n",
      "Running subcircuit 7...\n",
      "Subcircuit 7 took 0.0009982585906982422 seconds to run.\n",
      "Running subcircuit 8...\n",
      "Subcircuit 8 took 0.0011577606201171875 seconds to run.\n",
      "Running subcircuit 9...\n",
      "Subcircuit 9 took 0.0011379718780517578 seconds to run.\n",
      "Running subcircuit 10...\n",
      "Subcircuit 10 took 0.0028960704803466797 seconds to run.\n",
      "Running subcircuit 11...\n",
      "Subcircuit 11 took 0.0016770362854003906 seconds to run.\n",
      "Running subcircuit 12...\n",
      "Subcircuit 12 took 0.0012729167938232422 seconds to run.\n",
      "Running subcircuit 13...\n",
      "Subcircuit 13 took 0.0011429786682128906 seconds to run.\n",
      "Running subcircuit 14...\n",
      "Subcircuit 14 took 0.0016450881958007812 seconds to run.\n",
      "Running subcircuit 15...\n",
      "Subcircuit 15 took 0.0017118453979492188 seconds to run.\n",
      "Running subcircuit 16...\n",
      "Subcircuit 16 took 0.0012061595916748047 seconds to run.\n",
      "Running subcircuit 17...\n",
      "Subcircuit 17 took 0.0012319087982177734 seconds to run.\n",
      "Running subcircuit 18...\n",
      "Subcircuit 18 took 0.0019240379333496094 seconds to run.\n",
      "Running subcircuit 19...\n",
      "Subcircuit 19 took 0.003008127212524414 seconds to run.\n",
      "Running subcircuit 20...\n",
      "Subcircuit 20 took 0.0011639595031738281 seconds to run.\n",
      "Running subcircuit 21...\n",
      "Subcircuit 21 took 0.002207040786743164 seconds to run.\n",
      "Running subcircuit 22...\n",
      "Subcircuit 22 took 0.0017781257629394531 seconds to run.\n",
      "Running subcircuit 23...\n",
      "Subcircuit 23 took 0.0031280517578125 seconds to run.\n",
      "Running subcircuit 24...\n",
      "Subcircuit 24 took 0.0010731220245361328 seconds to run.\n",
      "Running subcircuit 25...\n",
      "Subcircuit 25 took 0.000978231430053711 seconds to run.\n",
      "Running subcircuit 26...\n",
      "Subcircuit 26 took 0.0010530948638916016 seconds to run.\n",
      "Running subcircuit 27...\n",
      "Subcircuit 27 took 0.0013010501861572266 seconds to run.\n",
      "Running subcircuit 28...\n",
      "Subcircuit 28 took 0.0012621879577636719 seconds to run.\n",
      "Running subcircuit 29...\n",
      "Subcircuit 29 took 0.0011658668518066406 seconds to run.\n",
      "Running subcircuit 30...\n",
      "Subcircuit 30 took 0.0033037662506103516 seconds to run.\n",
      "Running subcircuit 31...\n",
      "Subcircuit 31 took 0.0009670257568359375 seconds to run.\n",
      "Running subcircuit 32...\n",
      "Subcircuit 32 took 0.0010709762573242188 seconds to run.\n",
      "Running subcircuit 33...\n",
      "Subcircuit 33 took 0.002743959426879883 seconds to run.\n",
      "Running original circuit...\n",
      "Original circuit took 0.002792835235595703 seconds to run.\n",
      "Cuts took 0.17928004264831543 seconds to run.\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "({0: {'0 0': 524, '0 1': 500},\n",
       "  1: {'0 0': 503, '1 1': 521},\n",
       "  2: {'1 1': 519, '0 0': 505},\n",
       "  3: {'0 1': 523, '0 0': 501},\n",
       "  4: {'0 0': 505, '0 1': 519},\n",
       "  5: {'0 0': 501, '0 1': 523},\n",
       "  6: {'0 1': 467, '0 0': 557},\n",
       "  7: {'0 1': 516, '0 0': 508},\n",
       "  8: {'0 0': 522, '0 1': 502},\n",
       "  9: {'0 1': 530, '0 0': 494},\n",
       "  10: {'1 1': 539, '0 0': 485},\n",
       "  11: {'0 0': 500, '1 1': 524},\n",
       "  12: {'0 0': 529, '0 1': 495},\n",
       "  13: {'0 1': 502, '0 0': 522},\n",
       "  14: {'0 0': 535, '1 1': 489},\n",
       "  15: {'0 0': 513, '1 1': 511},\n",
       "  16: {'0 0': 501, '0 1': 523},\n",
       "  17: {'0 0': 522, '0 1': 502},\n",
       "  18: {'0 0': 529, '1 1': 495},\n",
       "  19: {'1 1': 514, '0 0': 510},\n",
       "  20: {'0 0': 483, '0 1': 541},\n",
       "  21: {'0 0': 515, '0 1': 509},\n",
       "  22: {'0 0': 506, '1 1': 518},\n",
       "  23: {'1 1': 495, '0 0': 529},\n",
       "  24: {'0 0': 480, '0 1': 544},\n",
       "  25: {'0 1': 532, '0 0': 492},\n",
       "  26: {'0 0': 512, '0 1': 512},\n",
       "  27: {'0 0': 524, '0 1': 500},\n",
       "  28: {'0 1': 534, '0 0': 490},\n",
       "  29: {'0 0': 509, '0 1': 515},\n",
       "  30: {'0 0': 540, '1 1': 484},\n",
       "  31: {'0 0': 496, '0 1': 528},\n",
       "  32: {'0 1': 524, '0 0': 500},\n",
       "  33: {'1 1': 510, '0 0': 514},\n",
       "  'original': {'0 0': 487, '1 1': 537}},\n",
       " 0.17928004264831543,\n",
       " 0.002792835235595703)"
      ]
     },
     "execution_count": 1,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import numpy as np\n",
    "from qiskit import transpile\n",
    "from qiskit_aer import AerSimulator\n",
    "from qiskit.circuit.random import random_circuit\n",
    "from qiskit.quantum_info import SparsePauliOp\n",
    "from circuit_knitting.cutting.automated_cut_finding import (\n",
    "    find_cuts,\n",
    "    OptimizationParameters,\n",
    "    DeviceConstraints,\n",
    ")\n",
    "from circuit_knitting.cutting import (\n",
    "    cut_wires,\n",
    "    expand_observables,\n",
    "    partition_problem,\n",
    "    generate_cutting_experiments,\n",
    ")\n",
    "import json\n",
    "import time\n",
    "\n",
    "def experiment_with_circuit(num_qubits, depth):\n",
    "    circuit = random_circuit(num_qubits, depth, max_operands=2, seed=1242)\n",
    "    \n",
    "    # Adjust observable strings to match the number of qubits\n",
    "    observable_strings = [\"Z\" + \"I\" * (num_qubits - 1),\n",
    "                          \"I\" * (num_qubits // 2) + \"Z\" + \"I\" * (num_qubits // 2 - 1),\n",
    "                          \"I\" * (num_qubits - 1) + \"Z\"]\n",
    "    observable_strings = [obs if len(obs) == num_qubits else \"I\" * num_qubits for obs in observable_strings]\n",
    "    observable = SparsePauliOp(observable_strings)\n",
    "\n",
    "    optimization_settings = OptimizationParameters(seed=111)\n",
    "    device_constraints = DeviceConstraints(qubits_per_subcircuit=4)\n",
    "\n",
    "    cut_circuit, metadata = find_cuts(circuit, optimization_settings, device_constraints)\n",
    "    print(f'Found solution using {len(metadata[\"cuts\"])} cuts with a sampling overhead of {metadata[\"sampling_overhead\"]}.')\n",
    "    for cut in metadata[\"cuts\"]:\n",
    "        print(f\"{cut[0]} at circuit instruction index {cut[1]}\")\n",
    "\n",
    "    qc_w_ancilla = cut_wires(cut_circuit)\n",
    "    observables_expanded = expand_observables(observable.paulis, circuit, qc_w_ancilla)\n",
    "\n",
    "    partitioned_problem = partition_problem(circuit=qc_w_ancilla, observables=observables_expanded)\n",
    "    subcircuits = partitioned_problem.subcircuits\n",
    "    subobservables = partitioned_problem.subobservables\n",
    "\n",
    "    print(f\"Sampling overhead: {np.prod([basis.overhead for basis in partitioned_problem.bases])}\")\n",
    "\n",
    "    subexperiments, coefficients = generate_cutting_experiments(circuits=subcircuits, observables=subobservables, num_samples=1000)\n",
    "    print(f\"{len(subexperiments[0]) + len(subexperiments[1])} total subexperiments to run on backend.\")\n",
    "\n",
    "    backend = AerSimulator()\n",
    "    results = {}\n",
    "\n",
    "    total_time = 0\n",
    "    for subexperiment in subexperiments.values():\n",
    "        for i, circuit in enumerate(subexperiment):\n",
    "            print(f\"Running subcircuit {i}...\")\n",
    "            transpiled_circuit = transpile(circuit, backend)\n",
    "            subexperiment_start = time.time()\n",
    "            result = backend.run(transpiled_circuit).result()\n",
    "            total_time += time.time() - subexperiment_start\n",
    "            counts = result.get_counts()\n",
    "            results[i] = counts\n",
    "            print(f\"Subcircuit {i} took {time.time() - subexperiment_start} seconds to run.\")\n",
    "\n",
    "    print(\"Running original circuit...\")\n",
    "    original_start = time.time()\n",
    "    original_counts = backend.run(circuit).result().get_counts()\n",
    "    original_time = time.time() - original_start\n",
    "    results[\"original\"] = original_counts\n",
    "\n",
    "    print(f\"Original circuit took {original_time} seconds to run.\")\n",
    "    print(f\"Cuts took {total_time} seconds to run.\")\n",
    "\n",
    "    with open(\"results.json\", \"w\") as f:\n",
    "        json.dump(results, f)\n",
    "\n",
    "    return results, total_time, original_time\n",
    "\n",
    "# Example usage with optimized parameters (8 qubits, depth 3, 1000 samples)\n",
    "results, total_time, original_time = experiment_with_circuit(8, 4)\n",
    "\n",
    "results, total_time, original_time"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# With IBMQ Mini-Circuit-Cutting "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Found solution using 1 cuts with a sampling overhead of 1.783275571221539.\n",
      "Gate Cut at circuit instruction index 13\n",
      "Sampling overhead: 1.783275571221539\n",
      "12 total subexperiments to run on backend.\n",
      "Subexperiments: {0: [<qiskit.circuit.quantumcircuit.QuantumCircuit object at 0x319c13dd0>, <qiskit.circuit.quantumcircuit.QuantumCircuit object at 0x16a550190>, <qiskit.circuit.quantumcircuit.QuantumCircuit object at 0x319c13a90>, <qiskit.circuit.quantumcircuit.QuantumCircuit object at 0x319c10cd0>, <qiskit.circuit.quantumcircuit.QuantumCircuit object at 0x319b9e350>, <qiskit.circuit.quantumcircuit.QuantumCircuit object at 0x16a5314d0>], 1: [<qiskit.circuit.quantumcircuit.QuantumCircuit object at 0x16a5cd7d0>, <qiskit.circuit.quantumcircuit.QuantumCircuit object at 0x16a493c10>, <qiskit.circuit.quantumcircuit.QuantumCircuit object at 0x1110f4510>, <qiskit.circuit.quantumcircuit.QuantumCircuit object at 0x16a4662d0>, <qiskit.circuit.quantumcircuit.QuantumCircuit object at 0x16a47c350>, <qiskit.circuit.quantumcircuit.QuantumCircuit object at 0x1111139d0>]}\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "/var/folders/w5/qtgfh_kx375_3385y0kbwgy40000gn/T/ipykernel_6724/3653426976.py:62: DeprecationWarning: The Sampler and Estimator V1 primitives have been deprecated as of qiskit-ibm-runtime 0.23.0 and will be removed no sooner than 3 months after the release date. Please use the V2 Primitives. See the `V2 migration guide <https://docs.quantum.ibm.com/api/migration-guides/v2-primitives>`_. for more details\n",
      "  sampler = Sampler(session=session)\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Batch 0 took 291.1208908557892 seconds to run.\n",
      "Batch 1 took 63.281394958496094 seconds to run.\n",
      "Running original circuit...\n",
      "Running subcircuit 0...\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "/var/folders/w5/qtgfh_kx375_3385y0kbwgy40000gn/T/ipykernel_6724/3653426976.py:104: DeprecationWarning: backend.run() and related sessions methods are deprecated  as of qiskit-ibm-runtime 0.23 and will be removed no sooner than 6 months after the release date. More details can be found in the primitives migration guide https://docs.quantum.ibm.com/api/migration-guides/qiskit-runtime.\n",
      "  result = backend.run(transpiled_circuit).result()\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Subcircuit 0 took 98.17586588859558 seconds to run.\n",
      "Running subcircuit 1...\n",
      "Subcircuit 1 took 27.412906885147095 seconds to run.\n",
      "Running subcircuit 2...\n",
      "Subcircuit 2 took 20.862608909606934 seconds to run.\n",
      "Running subcircuit 3...\n",
      "Subcircuit 3 took 28.183209896087646 seconds to run.\n",
      "Running subcircuit 4...\n",
      "Subcircuit 4 took 35.09509873390198 seconds to run.\n",
      "Running subcircuit 5...\n",
      "Subcircuit 5 took 20.407121181488037 seconds to run.\n",
      "Running subcircuit 0...\n",
      "Subcircuit 0 took 39.35172772407532 seconds to run.\n",
      "Running subcircuit 1...\n",
      "Subcircuit 1 took 21.97625494003296 seconds to run.\n",
      "Running subcircuit 2...\n",
      "Subcircuit 2 took 20.65148687362671 seconds to run.\n",
      "Running subcircuit 3...\n",
      "Subcircuit 3 took 24.904088258743286 seconds to run.\n",
      "Running subcircuit 4...\n",
      "Subcircuit 4 took 32.534209966659546 seconds to run.\n",
      "Running subcircuit 5...\n",
      "Subcircuit 5 took 19.189459085464478 seconds to run.\n",
      "Running original circuit...\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "/var/folders/w5/qtgfh_kx375_3385y0kbwgy40000gn/T/ipykernel_6724/3653426976.py:112: DeprecationWarning: backend.run() and related sessions methods are deprecated  as of qiskit-ibm-runtime 0.23 and will be removed no sooner than 6 months after the release date. More details can be found in the primitives migration guide https://docs.quantum.ibm.com/api/migration-guides/qiskit-runtime.\n",
      "  original_counts = backend.run(circuit).result().get_counts()\n"
     ]
    },
    {
     "ename": "RuntimeJobFailureError",
     "evalue": "'Unable to retrieve job result. Instruction s on qubits (0,) from the 0-th circuit is not supported by the target. -- \\\\n        Transpile your circuits for the target before submitting a primitive query. For\\\\n        example, you can use the following code block given an IBMBackend object `backend`\\\\n        and circuits of type `List[QuantumCircuit]`:\\\\n            from qiskit.transpiler.preset_passmanagers import generate_preset_pass_manager\\\\n            pm = generate_preset_pass_manager(optimization_level=1, target=backend.target)\\\\n            isa_circuits = pm.run(circuits)\\\\n        Then pass `isa_circuits` to the Sampler or Estimator.\\\\n         -- https://ibm.biz/error_codes#1517'",
     "output_type": "error",
     "traceback": [
      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
      "\u001b[0;31mRuntimeJobFailureError\u001b[0m                    Traceback (most recent call last)",
      "Cell \u001b[0;32mIn[2], line 126\u001b[0m\n\u001b[1;32m    122\u001b[0m     \u001b[38;5;28;01mreturn\u001b[39;00m results, total_time, original_time\n\u001b[1;32m    125\u001b[0m \u001b[38;5;66;03m# For IBM Quantum hardware:\u001b[39;00m\n\u001b[0;32m--> 126\u001b[0m results, total_time, original_time \u001b[38;5;241m=\u001b[39m \u001b[43mexperiment_with_circuit\u001b[49m\u001b[43m(\u001b[49m\u001b[38;5;241;43m8\u001b[39;49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;241;43m4\u001b[39;49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43muse_ibmq\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;28;43;01mTrue\u001b[39;49;00m\u001b[43m)\u001b[49m\n",
      "Cell \u001b[0;32mIn[2], line 112\u001b[0m, in \u001b[0;36mexperiment_with_circuit\u001b[0;34m(num_qubits, depth, use_ibmq)\u001b[0m\n\u001b[1;32m    110\u001b[0m \u001b[38;5;28mprint\u001b[39m(\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mRunning original circuit...\u001b[39m\u001b[38;5;124m\"\u001b[39m)\n\u001b[1;32m    111\u001b[0m original_start \u001b[38;5;241m=\u001b[39m time\u001b[38;5;241m.\u001b[39mtime()\n\u001b[0;32m--> 112\u001b[0m original_counts \u001b[38;5;241m=\u001b[39m \u001b[43mbackend\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mrun\u001b[49m\u001b[43m(\u001b[49m\u001b[43mcircuit\u001b[49m\u001b[43m)\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mresult\u001b[49m\u001b[43m(\u001b[49m\u001b[43m)\u001b[49m\u001b[38;5;241m.\u001b[39mget_counts()\n\u001b[1;32m    113\u001b[0m original_time \u001b[38;5;241m=\u001b[39m time\u001b[38;5;241m.\u001b[39mtime() \u001b[38;5;241m-\u001b[39m original_start\n\u001b[1;32m    114\u001b[0m results[\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124moriginal\u001b[39m\u001b[38;5;124m\"\u001b[39m] \u001b[38;5;241m=\u001b[39m original_counts\n",
      "File \u001b[0;32m/opt/miniconda3/envs/iisc-test/lib/python3.11/site-packages/qiskit_ibm_runtime/runtime_job.py:164\u001b[0m, in \u001b[0;36mRuntimeJob.result\u001b[0;34m(self, timeout, decoder)\u001b[0m\n\u001b[1;32m    162\u001b[0m     \u001b[38;5;28;01mif\u001b[39;00m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39m_reason \u001b[38;5;241m==\u001b[39m \u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mRAN TOO LONG\u001b[39m\u001b[38;5;124m\"\u001b[39m:\n\u001b[1;32m    163\u001b[0m         \u001b[38;5;28;01mraise\u001b[39;00m RuntimeJobMaxTimeoutError(error_message)\n\u001b[0;32m--> 164\u001b[0m     \u001b[38;5;28;01mraise\u001b[39;00m RuntimeJobFailureError(\u001b[38;5;124mf\u001b[39m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mUnable to retrieve job result. \u001b[39m\u001b[38;5;132;01m{\u001b[39;00merror_message\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m\"\u001b[39m)\n\u001b[1;32m    165\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39m_status \u001b[38;5;129;01mis\u001b[39;00m JobStatus\u001b[38;5;241m.\u001b[39mCANCELLED:\n\u001b[1;32m    166\u001b[0m     \u001b[38;5;28;01mraise\u001b[39;00m RuntimeInvalidStateError(\n\u001b[1;32m    167\u001b[0m         \u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mUnable to retrieve result for job \u001b[39m\u001b[38;5;132;01m{}\u001b[39;00m\u001b[38;5;124m. \u001b[39m\u001b[38;5;124m\"\u001b[39m \u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mJob was cancelled.\u001b[39m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;241m.\u001b[39mformat(\u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mjob_id())\n\u001b[1;32m    168\u001b[0m     )\n",
      "\u001b[0;31mRuntimeJobFailureError\u001b[0m: 'Unable to retrieve job result. Instruction s on qubits (0,) from the 0-th circuit is not supported by the target. -- \\\\n        Transpile your circuits for the target before submitting a primitive query. For\\\\n        example, you can use the following code block given an IBMBackend object `backend`\\\\n        and circuits of type `List[QuantumCircuit]`:\\\\n            from qiskit.transpiler.preset_passmanagers import generate_preset_pass_manager\\\\n            pm = generate_preset_pass_manager(optimization_level=1, target=backend.target)\\\\n            isa_circuits = pm.run(circuits)\\\\n        Then pass `isa_circuits` to the Sampler or Estimator.\\\\n         -- https://ibm.biz/error_codes#1517'"
     ]
    }
   ],
   "source": [
    "import numpy as np\n",
    "import json\n",
    "import time\n",
    "from qiskit import transpile, QuantumCircuit, ClassicalRegister, QuantumRegister\n",
    "from qiskit_ibm_runtime import QiskitRuntimeService, Sampler, Session\n",
    "from qiskit.quantum_info import SparsePauliOp\n",
    "from qiskit.circuit.random import random_circuit\n",
    "from circuit_knitting.cutting.automated_cut_finding import (\n",
    "    find_cuts,\n",
    "    OptimizationParameters,\n",
    "    DeviceConstraints,\n",
    ")\n",
    "from circuit_knitting.cutting import (\n",
    "    cut_wires,\n",
    "    expand_observables,\n",
    "    partition_problem,\n",
    "    generate_cutting_experiments,\n",
    ")\n",
    "\n",
    "def experiment_with_circuit(num_qubits, depth, use_ibmq=False):\n",
    "    circuit = random_circuit(num_qubits, depth, max_operands=2, seed=1242)\n",
    "    \n",
    "    # Adjust observable strings to match the number of qubits\n",
    "    observable_strings = [\"Z\" + \"I\" * (num_qubits - 1),\n",
    "                          \"I\" * (num_qubits // 2) + \"Z\" + \"I\" * (num_qubits // 2 - 1),\n",
    "                          \"I\" * (num_qubits - 1) + \"Z\"]\n",
    "    observable_strings = [obs if len(obs) == num_qubits else \"I\" * num_qubits for obs in observable_strings]\n",
    "    observable = SparsePauliOp(observable_strings)\n",
    "\n",
    "    optimization_settings = OptimizationParameters(seed=111)\n",
    "    device_constraints = DeviceConstraints(qubits_per_subcircuit=7)\n",
    "\n",
    "    cut_circuit, metadata = find_cuts(circuit, optimization_settings, device_constraints)\n",
    "    print(f'Found solution using {len(metadata[\"cuts\"])} cuts with a sampling overhead of {metadata[\"sampling_overhead\"]}.')\n",
    "    for cut in metadata[\"cuts\"]:\n",
    "        print(f\"{cut[0]} at circuit instruction index {cut[1]}\")\n",
    "\n",
    "    qc_w_ancilla = cut_wires(cut_circuit)\n",
    "    observables_expanded = expand_observables(observable.paulis, circuit, qc_w_ancilla)\n",
    "\n",
    "    partitioned_problem = partition_problem(circuit=qc_w_ancilla, observables=observables_expanded)\n",
    "    subcircuits = partitioned_problem.subcircuits\n",
    "    subobservables = partitioned_problem.subobservables\n",
    "\n",
    "    print(f\"Sampling overhead: {np.prod([basis.overhead for basis in partitioned_problem.bases])}\")\n",
    "\n",
    "    subexperiments, coefficients = generate_cutting_experiments(circuits=subcircuits, observables=subobservables, num_samples=1000)\n",
    "    print(f\"{len(subexperiments[0]) + len(subexperiments[1])} total subexperiments to run on backend.\")\n",
    "    \n",
    "    print(\"Subexperiments:\", subexperiments)\n",
    "\n",
    "    results = {}\n",
    "    total_time = 0\n",
    "\n",
    "    if use_ibmq:\n",
    "        # IBM Quantum setup\n",
    "        service = QiskitRuntimeService(channel=\"ibm_quantum\", token=\"f4578dfc2792a51df5907e166130b002695c4de4f2c694baa77b060328ac5dec1638d0930e8ebf00927adcfc57c660c7a1d8ce26b2ed7113230f6c23914599bf\")\n",
    "        backend = service.backend(\"ibm_brisbane\")\n",
    "        \n",
    "        # Create a session with the ibm_brisbane backend\n",
    "        session = Session(service=service, backend='ibm_brisbane')\n",
    "        sampler = Sampler(session=session)\n",
    "        \n",
    "        for batch, subcircuits in subexperiments.items():\n",
    "            transpiled_subcircuits = []\n",
    "            for subcircuit in subcircuits:\n",
    "                # Add classical register and measurement\n",
    "                creg = ClassicalRegister(subcircuit.num_qubits, 'creg')\n",
    "                subcircuit.add_register(creg)\n",
    "                for q in range(subcircuit.num_qubits):\n",
    "                    subcircuit.measure(q, creg[q])\n",
    "                # Transpile the subcircuit\n",
    "                transpiled_subcircuit = transpile(subcircuit, backend, optimization_level=1)\n",
    "                transpiled_subcircuits.append(transpiled_subcircuit)\n",
    "            \n",
    "            subexperiment_start = time.time()\n",
    "            job = sampler.run(transpiled_subcircuits)\n",
    "            result = job.result()\n",
    "            total_time += time.time() - subexperiment_start\n",
    "            \n",
    "            for i, counts in enumerate(result.quasi_dists):\n",
    "                results[f\"{batch}_{i}\"] = counts\n",
    "            \n",
    "            print(f\"Batch {batch} took {time.time() - subexperiment_start} seconds to run.\")\n",
    "\n",
    "        print(\"Running original circuit...\")\n",
    "        creg = ClassicalRegister(circuit.num_qubits, 'creg')\n",
    "        circuit.add_register(creg)\n",
    "        for q in range(circuit.num_qubits):\n",
    "            circuit.measure(q, creg[q])\n",
    "        transpiled_circuit = transpile(circuit, backend, optimization_level=1)\n",
    "        \n",
    "        original_start = time.time()\n",
    "        job = sampler.run(transpiled_circuit, shots=1000)\n",
    "        result = job.result()\n",
    "        original_time = time.time() - original_start\n",
    "        results[\"original\"] = result.quasi_dists[0]\n",
    "\n",
    "        for subexperiment in subexperiments.values():\n",
    "            for i, circuit in enumerate(subexperiment):\n",
    "                print(f\"Running subcircuit {i}...\")\n",
    "                transpiled_circuit = transpile(circuit, backend)\n",
    "                subexperiment_start = time.time()\n",
    "                result = backend.run(transpiled_circuit).result()\n",
    "                total_time += time.time() - subexperiment_start\n",
    "                counts = result.get_counts()\n",
    "                results[i] = counts\n",
    "                print(f\"Subcircuit {i} took {time.time() - subexperiment_start} seconds to run.\")\n",
    "\n",
    "        print(\"Running original circuit...\")\n",
    "        original_start = time.time()\n",
    "        original_counts = backend.run(circuit).result().get_counts()\n",
    "        original_time = time.time() - original_start\n",
    "        results[\"original\"] = original_counts\n",
    "\n",
    "    print(f\"Original circuit took {original_time} seconds to run.\")\n",
    "    print(f\"Cuts took {total_time} seconds to run.\")\n",
    "\n",
    "    with open(\"results.json\", \"w\") as f:\n",
    "        json.dump(results, f)\n",
    "\n",
    "    return results, total_time, original_time\n",
    "\n",
    "\n",
    "# For IBM Quantum hardware:\n",
    "results, total_time, original_time = experiment_with_circuit(8, 4, use_ibmq=True)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "iisc-test",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.11.9"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}