Add example for distributed teleport

Author: dieriver

examples/distributed/teleport/network-alice.json

@@ -0,0 +1,35 @@
+{
+    "default": {
+        "nodes": {
+            "Alice": {
+                "app_socket": [
+                    "192.168.0.160",
+                    8001
+                ],
+                "qnodeos_socket": [
+                    "192.168.0.160",
+                    8005
+                ],
+                "vnode_socket": [
+                    "192.168.0.160",
+                    8020
+                ]
+            },
+            "Bob": {
+                "app_socket": [
+                    "192.168.20.249",
+                    8032
+                ],
+                "qnodeos_socket": [
+                    "192.168.20.249",
+                    8037
+                ],
+                "vnode_socket": [
+                    "192.168.20.249",
+                    8050
+                ]
+            }
+        },
+        "topology": null
+    }
+}

examples/distributed/teleport/network-bob.json

@@ -0,0 +1,35 @@
+{
+    "default": {
+        "nodes": {
+            "Alice": {
+                "app_socket": [
+                    "localhost",
+                    8001
+                ],
+                "qnodeos_socket": [
+                    "localhost",
+                    8005
+                ],
+                "vnode_socket": [
+                    "localhost",
+                    8020
+                ]
+            },
+            "Bob": {
+                "app_socket": [
+                    "localhost",
+                    8032
+                ],
+                "qnodeos_socket": [
+                    "localhost",
+                    8037
+                ],
+                "vnode_socket": [
+                    "localhost",
+                    8050
+                ]
+            }
+        },
+        "topology": null
+    }
+}

examples/distributed/teleport/teleport-alice.py

@@ -0,0 +1,51 @@
+from pathlib import Path
+
+from netqasm.runtime.settings import set_simulator
+from netqasm.runtime.application import default_app_instance
+
+from simulaqron.run.run import run_applications
+
+set_simulator("simulaqron")
+
+from netqasm.sdk.external import NetQASMConnection  # noqa: E402
+from netqasm.sdk import Qubit, EPRSocket  # noqa: E402
+
+
+def run_alice():
+    epr_socket: EPRSocket = EPRSocket("Bob")
+    with NetQASMConnection("Alice", epr_sockets=[epr_socket]) as alice:
+        # Create a qubit
+        q = Qubit(alice)
+        q.H()
+
+        # Create entanglement
+        epr = epr_socket.create_keep()[0]
+
+        # Teleport
+        q.cnot(epr)
+        q.H()
+        m1 = q.measure()
+        m2 = epr.measure()
+    return m1, m2
+
+
+if __name__ == "__main__":
+    apps = default_app_instance(
+        [
+            ("Alice", run_alice)
+        ]
+    )
+    network_cfg_path = Path() / "network-alice.json"
+    raw_results = run_applications(
+        apps, use_app_config=False, enable_logging=False, network_cfg=network_cfg_path.resolve()
+    )
+
+    results = {}
+
+    for name, raw_result in raw_results[0].items():
+        if isinstance(raw_result, tuple):
+            results[name] = tuple(int(result) for result in raw_result)
+        else:
+            results[name] = int(raw_result)
+
+    print(results)

examples/distributed/teleport/teleport-bob.py

@@ -0,0 +1,41 @@
+from pathlib import Path
+
+from netqasm.runtime.settings import set_simulator
+
+from netqasm.runtime.application import default_app_instance
+
+from simulaqron.run.run import run_applications
+
+set_simulator("simulaqron")
+from netqasm.sdk.external import NetQASMConnection  # noqa: E402
+from netqasm.sdk import EPRSocket  # noqa: E402
+
+
+def run_bob():
+    epr_socket: EPRSocket = EPRSocket("Alice")
+    with NetQASMConnection("Bob", epr_sockets=[epr_socket]):
+        entangled_qubit = epr_socket.recv_keep()[0]
+        meas = entangled_qubit.measure()
+    return meas
+
+
+if __name__ == "__main__":
+    apps = default_app_instance(
+        [
+            ("Bob", run_bob)
+        ]
+    )
+    network_cfg_path = Path() / "network-alice.json"
+    raw_results = run_applications(
+        apps, use_app_config=False, enable_logging=False, network_cfg=network_cfg_path.resolve()
+    )
+
+    results = {}
+
+    for name, raw_result in raw_results[0].items():
+        if isinstance(raw_result, tuple):
+            results[name] = tuple(int(result) for result in raw_result)
+        else:
+            results[name] = int(raw_result)
+
+    print(results)

simulaqron/run/run.py

@@ -5,7 +5,7 @@
 from os import PathLike
 from pathlib import Path
 from time import sleep
-from typing import Callable, Optional, Any, Dict, List
+from typing import Callable, Optional, Any, Dict, List, Union
 
 from netqasm.logging.glob import get_netqasm_logger
 from netqasm.logging.output import (reset_struct_loggers,
@@ -79,7 +79,7 @@ def run_sim_backend(node_names: List[str], sim_backend: SimBackend, network_conf
 def run_applications(
     app_instance: ApplicationInstance,
     num_rounds: int = 1,
-    network_cfg: str = None,  # WARNING - The type of this argument *cannot* be harmonized
+    network_cfg: Union[str, PathLike, Path] = None,  # WARNING - The type of this argument *cannot* be harmonized
     nv_cfg: Any = None,  # Unused; it's here for harmonization with squidasm "simulate_application"
     log_cfg: LogConfig = None,
     formalism: Formalism = Formalism.KET,