Update circuit.py

Author: ruan-prog

tensorcircuit/circuit.py

@@ -64,6 +64,7 @@ def __init__(
             nqubits=nqubits, inputs=inputs, mps_inputs=mps_inputs, split=split
         )
         self.calibrations = []
+        self.calibration_invokes = []
 
         self.inputs = inputs
         self.mps_inputs = mps_inputs
@@ -117,7 +118,7 @@ def __init__(
         self._qir: List[Dict[str, Any]] = []
         self._extra_qir: List[Dict[str, Any]] = []
 
-    def add_calibration(
+    def def_calibration(
         self, name: str, parameters: List[str], instructions: List[Dict]
     ) -> None:
         self.calibrations.append({
@@ -126,6 +127,15 @@ def add_calibration(
             "instructions": instructions
         })
 
+    def add_calibration(
+        self, name: str, parameters: List[str], instructions: List[Dict]
+    ) -> None:
+        self.calibration_invokes.append({
+            "name": name,
+            "parameters": parameters,
+            "instructions": instructions
+        })
+
 
     def to_tqasm(self) -> str:
         qasm_lines = []
@@ -149,6 +159,10 @@ def to_tqasm(self) -> str:
                     qasm_lines.append(f"  play({inst['frame']}, {wf_type}({args_str}));")
             qasm_lines.append("}")
 
+        for cal in getattr(self, "calibration_invokes", []):
+            pname = ", ".join(cal["parameters"])
+            qasm_lines.append(f"\n {cal['name']} {pname};")
+
         return "\n".join(qasm_lines)
 
     def calibrate(self, name: str, parameters: List["Param"]) -> "DefcalBuilder":
@@ -1078,8 +1092,8 @@ def play(self, frame_name: str, waveform: Any, start_time: int = None):
         return self
 
     def build(self):
-        self.circuit.add_calibration(
+        self.circuit.def_calibration(
             name=self.name,
             parameters=[p.name for p in self.parameters],
             instructions=self.instructions,
-        )
+        )