quick fix

PennyLaneAI · Oct 22, 2024 · 454e4e0 · 454e4e0
1 parent 4e69bb3
commit 454e4e0
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/pennylane_lightning/lightning_gpu/_state_vector.py b/pennylane_lightning/lightning_gpu/_state_vector.py
@@ -238,9 +238,6 @@ def _apply_lightning_controlled(self, operation):
         Returns:
             None
         """
-        if self._mpi_handler.use_mpi:
-            raise DeviceError("Lightning-GPU-MPI does not support Controlled GlobalPhase gates.")
-
         state = self.state_vector
 
         basename = operation.base.name
@@ -336,6 +333,14 @@ def _apply_lightning(
                 isinstance(operation, qml.ops.Controlled) and not self._mpi_handler.use_mpi
             ):  # MPI backend does not have native controlled gates support
                 self._apply_lightning_controlled(operation)
+            elif (
+                self._mpi_handler.use_mpi
+                and isinstance(operation, qml.ops.Controlled)
+                and isinstance(operation.base, qml.GlobalPhase)
+            ):
+                raise DeviceError(
+                    "Lightning-GPU-MPI does not support Controlled GlobalPhase gates."
+                )
             else:  # apply gate as a matrix
                 try:
                     mat = qml.matrix(operation)