Add U gate

apache · Oct 9, 2024 · a65168a · a65168a
1 parent 0b096f9
commit a65168a
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 0 deletions.
diff --git a/qumat/amazon_braket_backend.py b/qumat/amazon_braket_backend.py
@@ -77,3 +77,8 @@ def apply_ry_gate(circuit, qubit_index, angle):
 
 def apply_rz_gate(circuit, qubit_index, angle):
     circuit.rz(qubit_index, angle)
+
+def apply_u_gate(circuit, qubit_index, theta, phi, lambd):
+    circuit.rx(qubit_index, theta)
+    circuit.ry(qubit_index, phi)
+    circuit.rz(qubit_index, lambd)
diff --git a/qumat/cirq_backend.py b/qumat/cirq_backend.py
@@ -86,3 +86,8 @@ def apply_rz_gate(circuit, qubit_index, angle):
     qubit = cirq.LineQubit(qubit_index)
     circuit.append(cirq.rz(angle).on(qubit))
 
+def apply_u_gate(circuit, qubit_index, theta, phi, lambd):
+    qubit = cirq.LineQubit(qubit_index)
+    circuit.append(cirq.rz(lambd).on(qubit))
+    circuit.append(cirq.ry(phi).on(qubit))
+    circuit.append(cirq.rx(theta).on(qubit))
diff --git a/qumat/qiskit_backend.py b/qumat/qiskit_backend.py
@@ -99,3 +99,7 @@ def apply_ry_gate(circuit, qubit_index, angle):
 
 def apply_rz_gate(circuit, qubit_index, angle):
     circuit.rz(angle, qubit_index)
+
+def apply_u_gate(circuit, qubit_index, theta, phi, lambd):
+    # Apply the U gate directly with specified parameters
+    circuit.u(theta, phi, lambd, qubit_index)
diff --git a/qumat/qumat.py b/qumat/qumat.py
@@ -70,3 +70,6 @@ def apply_ry_gate(self, qubit_index, angle):
 
     def apply_rz_gate(self, qubit_index, angle):
         self.backend_module.apply_rz_gate(self.circuit, qubit_index, angle)
+
+    def apply_u_gate(self, qubit_index, theta, phi, lambd):
+        self.backend_module.apply_u_gate(self.circuit, qubit_index, theta, phi, lambd)