fix disable multiRZ

PennyLaneAI · Oct 21, 2024 · 2d5b58e · 2d5b58e
1 parent f38470d
commit 2d5b58e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pennylane_lightning/lightning_kokkos/_state_vector.py b/pennylane_lightning/lightning_kokkos/_state_vector.py
@@ -285,7 +285,7 @@ def _apply_lightning(
                 param = operation.parameters
                 method(wires, invert_param, param)
             elif isinstance(operation, qml.ops.Controlled) and not isinstance(
-                operation.base, (qml.QubitUnitary, qml.BlockEncode)
+                operation.base, (qml.QubitUnitary, qml.BlockEncode, qml.MultiRZ)
             ):  # apply n-controlled gate
                 self._apply_lightning_controlled(operation)
             else:  # apply gate as a matrix