diff --git a/pennylane_lightning/lightning_kokkos/_state_vector.py b/pennylane_lightning/lightning_kokkos/_state_vector.py index 3888fc972..3f6ee8a79 100644 --- 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