fix disable multiRZ

PennyLaneAI · Oct 21, 2024 · 692c2cf · 692c2cf
1 parent 2dd657c
commit 692c2cf
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