multi_measure.

Author: tencentqlab

tensorcircuit/circuit.py

@@ -66,6 +66,7 @@ def __init__(
         )
         self.calibrations = []
         self.calibration_invokes = []
+        self.measz_invokes = []
 
         self.inputs = inputs
         self.mps_inputs = mps_inputs
@@ -136,6 +137,14 @@ def add_calibration(
             "parameters": parameters,
             "pos": len(self._qir)
         })
+    
+    def measz(self, *index: int) -> None:
+        self.measz_invokes.append({
+            "name": "measz",
+            "index": index,
+            "pos": len(self._qir)
+        })
+        return
 
 
     def to_tqasm(self, pragma: Optional[str]= None) -> str:
@@ -169,8 +178,18 @@ def to_tqasm(self, pragma: Optional[str]= None) -> str:
             pos = cal.get("pos", len(self._qir))
             cals_by_pos[pos].append(cal)
 
+        measz_by_pos = defaultdict(list)
+        for m in getattr(self, "measz_invokes", []):
+            # pos 记录的是加入时的 len(self._qir)
+            pos = m.get("pos", len(self._qir))
+            measz_by_pos[pos].append(m)
+
         # 交错输出：在第 i 个门之前输出所有 pos == i 的校准
         for i, gate in enumerate(self._qir):
+            for m in measz_by_pos.get(i, []):
+                targets = ", ".join(f"q[{idx}]" for idx in m["index"])
+                qasm_lines.append(f"MEASZ {targets};")
+
             for cal in cals_by_pos.get(i, []):
                 # print(cal)
                 pname = ", ".join(f"q[{x}]" for x in cal.get("parameters", []))
@@ -188,6 +207,10 @@ def to_tqasm(self, pragma: Optional[str]= None) -> str:
             else:
                 qasm_lines.append(f"{gname} {targets};")
 
+        for m in measz_by_pos.get(len(self._qir), []):
+                targets = ", ".join(f"q[{idx}]" for idx in m["index"])
+                qasm_lines.append(f"MEASZ {targets};")
+
         # 收尾：把 pos == len(self._qir) 的校准放在最后
         for cal in cals_by_pos.get(len(self._qir), []):
             # print(cal)

tests/05_test_multi_measure.py

@@ -32,12 +32,12 @@ def gen_multi_measure_circuit(t):
     # qc.rxx(0, 5, theta=1.04632)
 
 # 需添加测量指令
-    qc.measure(0, 1)
+    qc.measz(0, 1)
     qc.cz(0, 1)
-    qc.measure(0, 1)
+    qc.measz(0, 1)
 
 
-    # print(qc.to_tqasm())
+    print(qc.to_tqasm())
     return qc