Fix MultiRZ

unitaryfund · Jun 5, 2024 · 1d614d7 · 1d614d7
1 parent d939bfd
commit 1d614d7
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/pennylane_qrack/_version.py b/pennylane_qrack/_version.py
@@ -16,4 +16,4 @@
    Version number (major.minor.patch[-label])
 """
 
-__version__ = "0.6.5"
+__version__ = "0.6.6"
diff --git a/pennylane_qrack/qrack_device.py b/pennylane_qrack/qrack_device.py
@@ -258,21 +258,25 @@ def _apply_gate(self, op):
             op = op.base
             opname = op.name + ".inv"
 
+        par = op.parameters
+
         if opname == "MultiRZ":
             device_wires = self.map_wires(op.wires)
             for q in device_wires:
                 self._state.r(Pauli.PauliZ, par[0], q)
-        elif opname == "C(MultiRZ)":
+            return
+
+        if opname == "C(MultiRZ)":
             device_wires = self.map_wires(op.wires)
             control_wires = self.map_wires(op.control_wires)
             for q in device_wires:
                 self._state.mcr(Pauli.PauliZ, par[0], control_wires, q)
+            return
 
         # translate op wire labels to consecutive wire labels used by the device
         device_wires = self.map_wires(
             (op.control_wires + op.wires) if op.control_wires else op.wires
         )
-        par = op.parameters
 
         if opname in [
             "".join(p)