Add support for BlockEncode gate to lightning devices (#599)

* Add support for BlockEncode operation A recent operation added to PennyLane, only supports application via matrix. * Add test * Add changelog * Bump version * trigger CI * Auto update version * Trigger CI --------- Co-authored-by: AmintorDusko <[email protected]> Co-authored-by: Dev version update bot <github-actions[bot]@users.noreply.github.com>
PennyLaneAI · Jan 24, 2024 · 2716864 · 2716864
1 parent ff34173
commit 2716864
Show file tree

Hide file tree

Showing 9 changed files with 14 additions and 4 deletions.
diff --git a/.github/CHANGELOG.md b/.github/CHANGELOG.md
@@ -9,6 +9,9 @@
 
 ### Improvements
 
+* The `BlockEncode` operation from PennyLane is now supported on all Lightning devices.
+  [(#599)](https://github.com/PennyLaneAI/pennylane-lightning/pull/599)
+
 * OpenMP acceleration can now be enabled at compile time for all `lightning.qubit` gate kernels using the "-DLQ_ENABLE_KERNEL_OMP=1" CMake argument.
   [(#510)](https://github.com/PennyLaneAI/pennylane-lightning/pull/510)
 
@@ -35,7 +38,7 @@
 
 This release contains contributions from (in alphabetical order):
 
-Amintor Dusko, Vincent Michaud-Rioux, Lee J. O'Riordan, Shuli Shu
+Amintor Dusko, David Ittah, Vincent Michaud-Rioux, Lee J. O'Riordan, Shuli Shu
 
 ---
 

diff --git a/pennylane_lightning/core/_version.py b/pennylane_lightning/core/_version.py
@@ -16,4 +16,4 @@
    Version number (major.minor.patch[-label])
 """
 
-__version__ = "0.35.0-dev6"
+__version__ = "0.35.0-dev7"
diff --git a/pennylane_lightning/lightning_gpu/lightning_gpu.py b/pennylane_lightning/lightning_gpu/lightning_gpu.py
@@ -185,6 +185,7 @@ def _mebibytesToBytes(mebibytes):
         "OrbitalRotation",
         "QFT",
         "ECR",
+        "BlockEncode",
     }
 
     allowed_observables = {

diff --git a/pennylane_lightning/lightning_gpu/lightning_gpu.toml b/pennylane_lightning/lightning_gpu/lightning_gpu.toml
@@ -88,6 +88,7 @@ matrix = [
         "QubitCarry",
         "QubitSum",
         "DiagonalQubitUnitary",
+        "BlockEncode",
 ]
 
 [measurement_processes]
@@ -119,4 +120,4 @@ mid_circuit_measurement = false
 
 # This field is currently unchecked but it is reserved for the purpose of
 # determining if the device supports dynamic qubit allocation/deallocation.
-dynamic_qubit_management = false 
+dynamic_qubit_management = false
diff --git a/pennylane_lightning/lightning_kokkos/lightning_kokkos.py b/pennylane_lightning/lightning_kokkos/lightning_kokkos.py
@@ -140,6 +140,7 @@ def _kokkos_configuration():
         "OrbitalRotation",
         "QFT",
         "ECR",
+        "BlockEncode",
     }
 
     allowed_observables = {

diff --git a/pennylane_lightning/lightning_kokkos/lightning_kokkos.toml b/pennylane_lightning/lightning_kokkos/lightning_kokkos.toml
@@ -88,6 +88,7 @@ matrix = [
         "QubitCarry",
         "QubitSum",
         "DiagonalQubitUnitary",
+        "BlockEncode",
 ]
 
 [measurement_processes]
@@ -119,4 +120,4 @@ mid_circuit_measurement = true
 
 # This field is currently unchecked but it is reserved for the purpose of
 # determining if the device supports dynamic qubit allocation/deallocation.
-dynamic_qubit_management = false 
+dynamic_qubit_management = false
diff --git a/pennylane_lightning/lightning_qubit/lightning_qubit.py b/pennylane_lightning/lightning_qubit/lightning_qubit.py
@@ -154,6 +154,7 @@
         "OrbitalRotation",
         "QFT",
         "ECR",
+        "BlockEncode",
     }
 
     allowed_observables = {

diff --git a/pennylane_lightning/lightning_qubit/lightning_qubit.toml b/pennylane_lightning/lightning_qubit/lightning_qubit.toml
@@ -90,6 +90,7 @@ matrix = [
         "QubitCarry",
         "QubitSum",
         "DiagonalQubitUnitary",
+        "BlockEncode",
 ]
 
 [measurement_processes]

diff --git a/tests/test_gates.py b/tests/test_gates.py
@@ -76,6 +76,7 @@ def op(op_name):
         "QubitSum": [qml.QubitSum, [], {"wires": [0, 1, 2]}],
         "QubitCarry": [qml.QubitCarry, [], {"wires": [0, 1, 2, 3]}],
         "QubitUnitary": [qml.QubitUnitary, [], {"U": np.eye(16) * 1j, "wires": [0, 1, 2, 3]}],
+        "BlockEncode": [qml.BlockEncode, [[[0.2, 0, 0.2], [-0.2, 0.2, 0]]], {"wires": [0, 1, 2]}],
     }
     return ops_list.get(op_name)
-Original file line number
+Diff line change
@@ Expand Up / @@ -185,6 +185,7 @@ def _mebibytesToBytes(mebibytes): @@
             "OrbitalRotation",
             "QFT",
             "ECR",
+            "BlockEncode",
         }
         allowed_observables = {
@@ Expand Down @@