support visualization on cond measure

Author: refraction-ray

tensorcircuit/basecircuit.py

@@ -1092,7 +1092,7 @@ def cond_measurement(self, index: int) -> Tensor:
         :rtype: Tensor
         """
         return self.general_kraus(  # type: ignore
-            [np.array([[1.0, 0], [0, 0]]), np.array([[0, 0], [0, 1]])], index  # type: ignore
+            [np.array([[1.0, 0], [0, 0]]), np.array([[0, 0], [0, 1]])], index, name="measure"  # type: ignore
         )
 
     cond_measure = cond_measurement

tensorcircuit/circuit.py

@@ -317,6 +317,7 @@ def unitary_kraus2(
         *index: int,
         prob: Optional[Sequence[float]] = None,
         status: Optional[float] = None,
+        name: Optional[str] = None,
     ) -> Tensor:
         # general impl from Monte Carlo trajectory depolarizing above
         # still jittable
@@ -326,7 +327,12 @@ def index2gate2(r: Tensor, kraus: Sequence[Tensor]) -> Tensor:
             return backend.switch(r, [lambda _=k: _ for k in kraus])
 
         return self._unitary_kraus_template(
-            kraus, *index, prob=prob, status=status, get_gate_from_index=index2gate2
+            kraus,
+            *index,
+            prob=prob,
+            status=status,
+            get_gate_from_index=index2gate2,
+            name=name,
         )
 
     def unitary_kraus(
@@ -335,6 +341,7 @@ def unitary_kraus(
         *index: int,
         prob: Optional[Sequence[float]] = None,
         status: Optional[float] = None,
+        name: Optional[str] = None,
     ) -> Tensor:
         """
         Apply unitary gates in ``kraus`` randomly based on corresponding ``prob``.
@@ -360,7 +367,12 @@ def index2gate(r: Tensor, kraus: Sequence[Tensor]) -> Tensor:
             return reduce(add, [r[i] * kraus[i] for i in range(l)])
 
         return self._unitary_kraus_template(
-            kraus, *index, prob=prob, status=status, get_gate_from_index=index2gate
+            kraus,
+            *index,
+            prob=prob,
+            status=status,
+            get_gate_from_index=index2gate,
+            name=name,
         )
 
     def _unitary_kraus_template(
@@ -372,6 +384,7 @@ def _unitary_kraus_template(
         get_gate_from_index: Optional[
             Callable[[Tensor, Sequence[Tensor]], Tensor]
         ] = None,
+        name: Optional[str] = None,
     ) -> Tensor:  # DRY
         sites = len(index)
         kraus = [k.tensor if isinstance(k, tn.Node) else k for k in kraus]
@@ -406,7 +419,7 @@ def step_function(x: Tensor) -> Tensor:
             raise ValueError("no `get_gate_from_index` implementation is provided")
         g = get_gate_from_index(r, kraus)
         g = backend.reshape(g, [2 for _ in range(sites * 2)])
-        self.any(*index, unitary=g)  # type: ignore
+        self.any(*index, unitary=g, name=name)  # type: ignore
         return r
 
     def _general_kraus_tf(
@@ -482,6 +495,7 @@ def _general_kraus_2(
         kraus: Sequence[Gate],
         *index: int,
         status: Optional[float] = None,
+        name: Optional[str] = None,
     ) -> Tensor:
         # the graph building time is frustratingly slow, several minutes
         # though running time is in terms of ms
@@ -528,13 +542,16 @@ def calculate_kraus_p(i: int) -> Tensor:
             for w, k in zip(prob, kraus_tensor)
         ]
 
-        return self.unitary_kraus2(new_kraus, *index, prob=prob, status=status)
+        return self.unitary_kraus2(
+            new_kraus, *index, prob=prob, status=status, name=name
+        )
 
     def general_kraus(
         self,
         kraus: Sequence[Gate],
         *index: int,
         status: Optional[float] = None,
+        name: Optional[str] = None,
     ) -> Tensor:
         """
         Monte Carlo trajectory simulation of general Kraus channel whose Kraus operators cannot be
@@ -553,7 +570,7 @@ def general_kraus(
             when the random number will be generated automatically
         :type status: Optional[float], optional
         """
-        return self._general_kraus_2(kraus, *index, status=status)
+        return self._general_kraus_2(kraus, *index, status=status, name=name)
 
     apply_general_kraus = general_kraus
 

tensorcircuit/densitymatrix.py

@@ -175,7 +175,7 @@ def check_kraus(kraus: Sequence[Gate]) -> bool:  # TODO(@refraction-ray)
         return True
 
     def apply_general_kraus(
-        self, kraus: Sequence[Gate], index: Sequence[Tuple[int, ...]]
+        self, kraus: Sequence[Gate], index: Sequence[Tuple[int, ...]], **kws: Any
     ) -> None:
         # note the API difference for index arg between DM and DM2
         self.check_kraus(kraus)
@@ -302,7 +302,7 @@ def to_circuit(self, circuit_params: Optional[Dict[str, Any]] = None) -> Circuit
 
 
 class DMCircuit2(DMCircuit):
-    def apply_general_kraus(self, kraus: Sequence[Gate], *index: int) -> None:  # type: ignore
+    def apply_general_kraus(self, kraus: Sequence[Gate], *index: int, **kws: Any) -> None:  # type: ignore
         # incompatible API for now
         kraus = [
             k

tensorcircuit/translation.py

@@ -3,21 +3,27 @@
 """
 
 from typing import Any, Dict, List, Optional
-import warnings
-
+import logging
 import numpy as np
 
+logger = logging.getLogger(__name__)
+
+
 try:
     from qiskit import QuantumCircuit
     import qiskit.quantum_info as qi
+    from qiskit.extensions.exceptions import ExtensionError
 except ImportError:
-    warnings.warn("Please first ``pip install qiskit`` to enable related functionality")
+    logger.warning(
+        "Please first ``pip install qiskit`` to enable related functionality"
+    )
 
 from . import gates
 from .circuit import Circuit
 from .densitymatrix import DMCircuit2
 from .cons import backend
 
+
 Tensor = Any
 
 
@@ -150,7 +156,16 @@ def qir2qiskit(qir: List[Dict[str, Any]], n: int) -> Any:
                     [2 ** len(index), 2 ** len(index)],
                 )
             )
-            qiskit_circ.unitary(qop, index[::-1], label=qis_name)
+            try:
+                qiskit_circ.unitary(qop, index[::-1], label=qis_name)
+            except ExtensionError:
+                logger.warning(
+                    "omit non unitary gate in tensorcircuit when transforming to qiskit"
+                )
+                qiskit_circ.unitary(
+                    np.eye(2 ** len(index)), index[::-1], label=qis_name
+                )
+
     return qiskit_circ
 
 

tests/test_circuit.py

@@ -1073,3 +1073,12 @@ def amp(s):
     np.testing.assert_allclose(
         amp(tc.array_to_tensor([0, 0, 0], dtype="float32")), 1 / np.sqrt(2), atol=1e-5
     )
+
+
+def test_draw_cond_measure():
+    c = tc.Circuit(2)
+    c.H(0)
+    c.cond_measure(0)
+    c.cnot(0, 1)
+    print("")
+    print(c.draw())