From ed739815cceab0bce720166878c828dbbdcf8feb Mon Sep 17 00:00:00 2001
From: Ali Asadi <10773383+maliasadi@users.noreply.github.com>
Date: Tue, 28 May 2024 11:05:16 -0400
Subject: [PATCH 1/7] Update Measurements.probs

---
 .github/CHANGELOG.md                          |  4 +
 .../measurements/MeasurementsKokkos.hpp       | 94 ++++++++++---------
 .../measurements/MeasurementsLQubit.hpp       | 82 ++++++++--------
 3 files changed, 99 insertions(+), 81 deletions(-)

diff --git a/.github/CHANGELOG.md b/.github/CHANGELOG.md
index 7f3a70fc96..e680dc23c2 100644
--- a/.github/CHANGELOG.md
+++ b/.github/CHANGELOG.md
@@ -45,6 +45,10 @@
 * Changed the name of `lightning.tensor` to `default.tensor` with the `quimb` backend.
   [(#719)](https://github.com/PennyLaneAI/pennylane-lightning/pull/719)
 
+* Dispatch the C++ `Measurements.probs(wires)` method in Lightning-Qubit and Lighnting-Kokkos to `Measurements.probs()` when requesting the probabilities for all wires.
+  This will trigger a more optimized implementation for calculating the probabilities of the whole system.
+  [(#)]()
+
 ### Documentation
 
 ### Bug fixes
diff --git a/pennylane_lightning/core/src/simulators/lightning_kokkos/measurements/MeasurementsKokkos.hpp b/pennylane_lightning/core/src/simulators/lightning_kokkos/measurements/MeasurementsKokkos.hpp
index 20f538df9a..f7b6132e95 100644
--- a/pennylane_lightning/core/src/simulators/lightning_kokkos/measurements/MeasurementsKokkos.hpp
+++ b/pennylane_lightning/core/src/simulators/lightning_kokkos/measurements/MeasurementsKokkos.hpp
@@ -80,11 +80,12 @@ class Measurements final
      * observables.
      *
      * @tparam functor_t Expectation value functor class for Kokkos dispatcher.
-     * @tparam nqubits Number of wires.
+     * @tparam num_wires Number of wires.
      * @param wires Wires to apply the observable to.
      */
     template <template <class> class functor_t, int num_wires>
-    PrecisionT applyExpValNamedFunctor(const std::vector<std::size_t> &wires) {
+    auto applyExpValNamedFunctor(const std::vector<std::size_t> &wires)
+        -> PrecisionT {
         if constexpr (num_wires > 0)
             PL_ASSERT(wires.size() == num_wires);
 
@@ -101,13 +102,14 @@ class Measurements final
      * matrix-valued operator.
      *
      * @tparam functor_t Expectation value functor class for Kokkos dispatcher.
-     * @tparam nqubits Number of wires.
+     * @tparam num_wires Number of wires.
      * @param matrix Matrix (linearized into a KokkosVector).
      * @param wires Wires to apply the observable to.
      */
     template <template <class> class functor_t, int num_wires>
-    PrecisionT applyExpValFunctor(const KokkosVector matrix,
-                                  const std::vector<std::size_t> &wires) {
+    auto applyExpValFunctor(const KokkosVector matrix,
+                            const std::vector<std::size_t> &wires)
+        -> PrecisionT {
         PL_ASSERT(wires.size() == num_wires);
         const std::size_t num_qubits = this->_statevector.getNumQubits();
         Kokkos::View<ComplexT *> arr_data = this->_statevector.getView();
@@ -178,12 +180,12 @@ class Measurements final
     /**
      * @brief Calculate expectation value for a general Observable.
      *
-     * @param ob Observable.
+     * @param obs An Observable object.
      * @return Expectation value with respect to the given observable.
      */
-    PrecisionT expval(const Observable<StateVectorT> &ob) {
+    auto expval(const Observable<StateVectorT> &obs) -> PrecisionT {
         StateVectorT ob_sv{this->_statevector};
-        ob.applyInPlace(ob_sv);
+        obs.applyInPlace(ob_sv);
         return getRealOfComplexInnerProduct(this->_statevector.getView(),
                                             ob_sv.getView());
     }
@@ -191,13 +193,12 @@ class Measurements final
     /**
      * @brief Calculate expectation value for a HermitianObs.
      *
-     * @param ob HermitianObs.
+     * @param obs A HermitianObs object.
      * @return Expectation value with respect to the given observable.
      */
-    PrecisionT
-    expval(const Pennylane::LightningKokkos::Observables::HermitianObs<
-           StateVectorT> &ob) {
-        return expval(ob.getMatrix(), ob.getWires());
+    auto expval(const Pennylane::LightningKokkos::Observables::HermitianObs<
+                StateVectorT> &obs) -> PrecisionT {
+        return expval(obs.getMatrix(), obs.getWires());
     }
 
     /**
@@ -207,8 +208,8 @@ class Measurements final
      * @param wires Wires where to apply the operator.
      * @return Floating point expected value of the observable.
      */
-    PrecisionT expval(const std::vector<ComplexT> &matrix_,
-                      const std::vector<std::size_t> &wires) {
+    auto expval(const std::vector<ComplexT> &matrix_,
+                const std::vector<std::size_t> &wires) -> PrecisionT {
         PL_ABORT_IF(matrix_.size() != exp2(2 * wires.size()),
                     "The size of matrix does not match with the given "
                     "number of wires");
@@ -225,8 +226,8 @@ class Measurements final
      * @param wires Wires where to apply the operator.
      * @return Floating point expected value of the observable.
      */
-    PrecisionT expval(const std::string &operation,
-                      const std::vector<std::size_t> &wires) {
+    auto expval(const std::string &operation,
+                const std::vector<std::size_t> &wires) -> PrecisionT {
         switch (expval_funcs_[operation]) {
         case ExpValFunc::Identity:
             return applyExpValNamedFunctor<getExpectationValueIdentityFunctor,
@@ -260,9 +261,9 @@ class Measurements final
      * observables.
      */
     template <typename op_type>
-    std::vector<PrecisionT>
-    expval(const std::vector<op_type> &operations_list,
-           const std::vector<std::vector<std::size_t>> &wires_list) {
+    auto expval(const std::vector<op_type> &operations_list,
+                const std::vector<std::vector<std::size_t>> &wires_list)
+        -> std::vector<PrecisionT> {
         PL_ABORT_IF(
             (operations_list.size() != wires_list.size()),
             "The lengths of the list of operations and wires do not match.");
@@ -279,9 +280,9 @@ class Measurements final
     /**
      * @brief Expectation value for a Observable with shots
      *
-     * @param obs Observable.
+     * @param obs An Observable object.
      * @param num_shots Number of shots.
-     * @param shots_range Vector of shot number to measurement.
+     * @param shot_range Vector of shot number to measurement.
      * @return Floating point expected value of the observable.
      */
 
@@ -335,12 +336,12 @@ class Measurements final
     /**
      * @brief Calculate variance of a general Observable.
      *
-     * @param ob Observable.
+     * @param obs An Observable object.
      * @return Variance with respect to the given observable.
      */
-    auto var(const Observable<StateVectorT> &ob) -> PrecisionT {
+    auto var(const Observable<StateVectorT> &obs) -> PrecisionT {
         StateVectorT ob_sv{this->_statevector};
-        ob.applyInPlace(ob_sv);
+        obs.applyInPlace(ob_sv);
 
         const PrecisionT mean_square =
             getRealOfComplexInnerProduct(ob_sv.getView(), ob_sv.getView());
@@ -358,8 +359,8 @@ class Measurements final
      * @param wires Wires where to apply the operator.
      * @return Floating point with the variance of the observable.
      */
-    PrecisionT var(const std::string &operation,
-                   const std::vector<std::size_t> &wires) {
+    auto var(const std::string &operation,
+             const std::vector<std::size_t> &wires) -> PrecisionT {
         StateVectorT ob_sv{this->_statevector};
         ob_sv.applyOperation(operation, wires);
 
@@ -373,14 +374,14 @@ class Measurements final
     };
 
     /**
-     * @brief Variance of an observable.
+     * @brief Variance of a Hermitian matrix.
      *
      * @param matrix Square matrix in row-major order.
      * @param wires Wires where to apply the operator.
      * @return Floating point with the variance of the observable.
      */
-    PrecisionT var(const std::vector<ComplexT> &matrix,
-                   const std::vector<std::size_t> &wires) {
+    auto var(const std::vector<ComplexT> &matrix,
+             const std::vector<std::size_t> &wires) -> PrecisionT {
         StateVectorT ob_sv{this->_statevector};
         ob_sv.applyMatrix(matrix, wires);
 
@@ -404,9 +405,9 @@ class Measurements final
      observables.
      */
     template <typename op_type>
-    std::vector<PrecisionT>
-    var(const std::vector<op_type> &operations_list,
-        const std::vector<std::vector<std::size_t>> &wires_list) {
+    auto var(const std::vector<op_type> &operations_list,
+             const std::vector<std::vector<std::size_t>> &wires_list)
+        -> std::vector<PrecisionT> {
         PL_ABORT_IF(
             (operations_list.size() != wires_list.size()),
             "The lengths of the list of operations and wires do not match.");
@@ -437,9 +438,9 @@ class Measurements final
      * @return Floating point with the variance of the sparse Hamiltonian.
      */
     template <class index_type>
-    PrecisionT var(const index_type *row_map_ptr, const index_type row_map_size,
-                   const index_type *entries_ptr, const ComplexT *values_ptr,
-                   const index_type numNNZ) {
+    auto var(const index_type *row_map_ptr, const index_type row_map_size,
+             const index_type *entries_ptr, const ComplexT *values_ptr,
+             const index_type numNNZ) -> PrecisionT {
         PL_ABORT_IF(
             (this->_statevector.getLength() != (std::size_t(row_map_size) - 1)),
             "Statevector and Hamiltonian have incompatible sizes.");
@@ -506,11 +507,18 @@ class Measurements final
      * @return Floating point std::vector with probabilities.
      * The basis columns are rearranged according to wires.
      */
-    std::vector<PrecisionT> probs(const std::vector<std::size_t> &wires) {
+    auto probs(const std::vector<std::size_t> &wires)
+        -> std::vector<PrecisionT> {
         PL_ABORT_IF_NOT(
             std::is_sorted(wires.cbegin(), wires.cend()),
             "LightningKokkos does not currently support out-of-order wire "
             "indices with probability calculations");
+
+        // If all wires are requested, dispatch to `this->probs()`
+        if (wires.size() == this->_statevector.getNumQubits()) {
+            return this->probs();
+        }
+
         using MDPolicyType_2D =
             Kokkos::MDRangePolicy<Kokkos::Rank<2, Kokkos::Iterate::Left>>;
 
@@ -624,8 +632,8 @@ class Measurements final
      * @return Floating point std::vector with probabilities
      * in lexicographic order.
      */
-    std::vector<PrecisionT> probs(const Observable<StateVectorT> &obs,
-                                  std::size_t num_shots = 0) {
+    auto probs(const Observable<StateVectorT> &obs, std::size_t num_shots = 0)
+        -> std::vector<PrecisionT> {
         return BaseType::probs(obs, num_shots);
     }
 
@@ -636,22 +644,22 @@ class Measurements final
      *
      * @return Floating point std::vector with probabilities.
      */
-    std::vector<PrecisionT> probs(std::size_t num_shots) {
+    auto probs(std::size_t num_shots) -> std::vector<PrecisionT> {
         return BaseType::probs(num_shots);
     }
 
     /**
      * @brief Probabilities with shot-noise for a subset of the full system.
      *
-     * @param num_shots Number of shots.
      * @param wires Wires will restrict probabilities to a subset
      * of the full system.
+     * @param num_shots Number of shots.
      *
      * @return Floating point std::vector with probabilities.
      */
 
-    std::vector<PrecisionT> probs(const std::vector<std::size_t> &wires,
-                                  std::size_t num_shots) {
+    auto probs(const std::vector<std::size_t> &wires, std::size_t num_shots)
+        -> std::vector<PrecisionT> {
         PL_ABORT_IF_NOT(
             std::is_sorted(wires.cbegin(), wires.cend()),
             "LightningKokkos does not currently support out-of-order wire "
diff --git a/pennylane_lightning/core/src/simulators/lightning_qubit/measurements/MeasurementsLQubit.hpp b/pennylane_lightning/core/src/simulators/lightning_qubit/measurements/MeasurementsLQubit.hpp
index 61d450fa48..1ddcc224b9 100644
--- a/pennylane_lightning/core/src/simulators/lightning_qubit/measurements/MeasurementsLQubit.hpp
+++ b/pennylane_lightning/core/src/simulators/lightning_qubit/measurements/MeasurementsLQubit.hpp
@@ -77,7 +77,7 @@ class Measurements final
      * @return Floating point std::vector with probabilities
      * in lexicographic order.
      */
-    std::vector<PrecisionT> probs() {
+    auto probs() -> std::vector<PrecisionT> {
         const ComplexT *arr_data = this->_statevector.getData();
         std::vector<PrecisionT> basis_probs(this->_statevector.getLength(), 0);
 
@@ -96,9 +96,10 @@ class Measurements final
      * @return Floating point std::vector with probabilities.
      * The basis columns are rearranged according to wires.
      */
-    std::vector<PrecisionT>
+    auto
     probs(const std::vector<std::size_t> &wires,
-          [[maybe_unused]] const std::vector<std::size_t> &device_wires = {}) {
+          [[maybe_unused]] const std::vector<std::size_t> &device_wires = {})
+        -> std::vector<PrecisionT> {
         // Determining index that would sort the vector.
         // This information is needed later.
         const auto sorted_ind_wires = Pennylane::Util::sorting_indices(wires);
@@ -109,6 +110,12 @@ class Measurements final
             sorted_wires[pos] = wires[sorted_ind_wires[pos]];
         }
 
+        // If all wires are requested, dispatch to `this->probs()`
+        if (wires.size() == this->_statevector.getNumQubits() &&
+            wires == sorted_wires) {
+            return this->probs();
+        }
+
         // Determining probabilities for the sorted wires.
         const ComplexT *arr_data = this->_statevector.getData();
 
@@ -164,8 +171,8 @@ class Measurements final
      * @return Floating point std::vector with probabilities
      * in lexicographic order.
      */
-    std::vector<PrecisionT> probs(const Observable<StateVectorT> &obs,
-                                  std::size_t num_shots = 0) {
+    auto probs(const Observable<StateVectorT> &obs, std::size_t num_shots = 0)
+        -> std::vector<PrecisionT> {
         return BaseType::probs(obs, num_shots);
     }
 
@@ -176,22 +183,22 @@ class Measurements final
      *
      * @return Floating point std::vector with probabilities.
      */
-    std::vector<PrecisionT> probs(size_t num_shots) {
+    auto probs(size_t num_shots) -> std::vector<PrecisionT> {
         return BaseType::probs(num_shots);
     }
 
     /**
      * @brief Probabilities with shot-noise for a subset of the full system.
      *
-     * @param num_shots Number of shots.
      * @param wires Wires will restrict probabilities to a subset
+     * @param num_shots Number of shots.
      * of the full system.
      *
      * @return Floating point std::vector with probabilities.
      */
 
-    std::vector<PrecisionT> probs(const std::vector<std::size_t> &wires,
-                                  std::size_t num_shots) {
+    auto probs(const std::vector<std::size_t> &wires, std::size_t num_shots)
+        -> std::vector<PrecisionT> {
         return BaseType::probs(wires, num_shots);
     }
 
@@ -202,8 +209,8 @@ class Measurements final
      * @param wires Wires where to apply the operator.
      * @return Floating point expected value of the observable.
      */
-    PrecisionT expval(const std::vector<ComplexT> &matrix,
-                      const std::vector<std::size_t> &wires) {
+    auto expval(const std::vector<ComplexT> &matrix,
+                const std::vector<std::size_t> &wires) -> PrecisionT {
         // Copying the original state vector, for the application of the
         // observable operator.
         StateVectorLQubitManaged<PrecisionT> operator_statevector(
@@ -224,8 +231,8 @@ class Measurements final
      * @param wires Wires where to apply the operator.
      * @return Floating point expected value of the observable.
      */
-    PrecisionT expval(const std::string &operation,
-                      const std::vector<std::size_t> &wires) {
+    auto expval(const std::string &operation,
+                const std::vector<std::size_t> &wires) -> PrecisionT {
         // Copying the original state vector, for the application of the
         // observable operator.
         StateVectorLQubitManaged<PrecisionT> operator_statevector(
@@ -255,10 +262,9 @@ class Measurements final
      * @return Floating point expected value of the observable.
      */
     template <class index_type>
-    PrecisionT expval(const index_type *row_map_ptr,
-                      const index_type row_map_size,
-                      const index_type *entries_ptr, const ComplexT *values_ptr,
-                      const index_type numNNZ) {
+    auto expval(const index_type *row_map_ptr, const index_type row_map_size,
+                const index_type *entries_ptr, const ComplexT *values_ptr,
+                const index_type numNNZ) -> PrecisionT {
         PL_ABORT_IF(
             (this->_statevector.getLength() != (size_t(row_map_size) - 1)),
             "Statevector and Hamiltonian have incompatible sizes.");
@@ -283,9 +289,9 @@ class Measurements final
      * observables.
      */
     template <typename op_type>
-    std::vector<PrecisionT>
-    expval(const std::vector<op_type> &operations_list,
-           const std::vector<std::vector<std::size_t>> &wires_list) {
+    auto expval(const std::vector<op_type> &operations_list,
+                const std::vector<std::vector<std::size_t>> &wires_list)
+        -> std::vector<PrecisionT> {
         PL_ABORT_IF(
             (operations_list.size() != wires_list.size()),
             "The lengths of the list of operations and wires do not match.");
@@ -302,16 +308,16 @@ class Measurements final
     /**
      * @brief Expectation value for a general Observable
      *
-     * @param ob Observable
+     * @param obs An observable object.
      * @return Floating point expected value of the observable.
      */
-    auto expval(const Observable<StateVectorT> &ob) -> PrecisionT {
+    auto expval(const Observable<StateVectorT> &obs) -> PrecisionT {
         PrecisionT result{};
 
         if constexpr (std::is_same_v<typename StateVectorT::MemoryStorageT,
                                      MemoryStorageLocation::Internal>) {
             StateVectorT sv(this->_statevector);
-            result = calculateObsExpval(sv, ob, this->_statevector);
+            result = calculateObsExpval(sv, obs, this->_statevector);
         } else if constexpr (std::is_same_v<
                                  typename StateVectorT::MemoryStorageT,
                                  MemoryStorageLocation::External>) {
@@ -319,7 +325,7 @@ class Measurements final
                 this->_statevector.getData(),
                 this->_statevector.getData() + this->_statevector.getLength());
             StateVectorT sv(data_storage.data(), data_storage.size());
-            result = calculateObsExpval(sv, ob, this->_statevector);
+            result = calculateObsExpval(sv, obs, this->_statevector);
         } else {
             /// LCOV_EXCL_START
             PL_ABORT("Undefined memory storage location for StateVectorT.");
@@ -332,9 +338,9 @@ class Measurements final
     /**
      * @brief Expectation value for a Observable with shots
      *
-     * @param obs Observable
+     * @param obs An observable object.
      * @param num_shots Number of shots.
-     * @param shots_range Vector of shot number to measurement
+     * @param shot_range Vector of shot number to measurement
      * @return Floating point expected value of the observable.
      */
 
@@ -361,15 +367,15 @@ class Measurements final
     /**
      * @brief Variance value for a general Observable
      *
-     * @param ob Observable
+     * @param obs An observable object.
      * @return Floating point with the variance of the observable.
      */
-    auto var(const Observable<StateVectorT> &ob) -> PrecisionT {
+    auto var(const Observable<StateVectorT> &obs) -> PrecisionT {
         PrecisionT result{};
         if constexpr (std::is_same_v<typename StateVectorT::MemoryStorageT,
                                      MemoryStorageLocation::Internal>) {
             StateVectorT sv(this->_statevector);
-            result = calculateObsVar(sv, ob, this->_statevector);
+            result = calculateObsVar(sv, obs, this->_statevector);
 
         } else if constexpr (std::is_same_v<
                                  typename StateVectorT::MemoryStorageT,
@@ -378,7 +384,7 @@ class Measurements final
                 this->_statevector.getData(),
                 this->_statevector.getData() + this->_statevector.getLength());
             StateVectorT sv(data_storage.data(), data_storage.size());
-            result = calculateObsVar(sv, ob, this->_statevector);
+            result = calculateObsVar(sv, obs, this->_statevector);
         } else {
             /// LCOV_EXCL_START
             PL_ABORT("Undefined memory storage location for StateVectorT.");
@@ -394,8 +400,8 @@ class Measurements final
      * @param wires Wires where to apply the operator.
      * @return Floating point with the variance of the observable.
      */
-    PrecisionT var(const std::string &operation,
-                   const std::vector<std::size_t> &wires) {
+    auto var(const std::string &operation,
+             const std::vector<std::size_t> &wires) -> PrecisionT {
         // Copying the original state vector, for the application of the
         // observable operator.
         StateVectorLQubitManaged<PrecisionT> operator_statevector(
@@ -416,14 +422,14 @@ class Measurements final
     };
 
     /**
-     * @brief Variance of an observable.
+     * @brief Variance of a Hermitian matrix.
      *
      * @param matrix Square matrix in row-major order.
      * @param wires Wires where to apply the operator.
      * @return Floating point with the variance of the observable.
      */
-    PrecisionT var(const std::vector<ComplexT> &matrix,
-                   const std::vector<std::size_t> &wires) {
+    auto var(const std::vector<ComplexT> &matrix,
+             const std::vector<std::size_t> &wires) -> PrecisionT {
         // Copying the original state vector, for the application of the
         // observable operator.
         StateVectorLQubitManaged<PrecisionT> operator_statevector(
@@ -454,9 +460,9 @@ class Measurements final
      observables.
      */
     template <typename op_type>
-    std::vector<PrecisionT>
-    var(const std::vector<op_type> &operations_list,
-        const std::vector<std::vector<std::size_t>> &wires_list) {
+    auto var(const std::vector<op_type> &operations_list,
+             const std::vector<std::vector<std::size_t>> &wires_list)
+        -> std::vector<PrecisionT> {
         PL_ABORT_IF(
             (operations_list.size() != wires_list.size()),
             "The lengths of the list of operations and wires do not match.");

From 2961df7431126d72ea1c1e51614cc48c7f2526d0 Mon Sep 17 00:00:00 2001
From: ringo-but-quantum <github-ringo-but-quantum@xanadu.ai>
Date: Tue, 28 May 2024 17:17:21 +0000
Subject: [PATCH 2/7] Auto update version from '0.37.0-dev19' to '0.37.0-dev21'

---
 pennylane_lightning/core/_version.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pennylane_lightning/core/_version.py b/pennylane_lightning/core/_version.py
index 402445d990..c29fafeccd 100644
--- a/pennylane_lightning/core/_version.py
+++ b/pennylane_lightning/core/_version.py
@@ -16,4 +16,4 @@
    Version number (major.minor.patch[-label])
 """
 
-__version__ = "0.37.0-dev19"
+__version__ = "0.37.0-dev21"

From 3b0a081843d37d0748597167e2a9248da016f0f1 Mon Sep 17 00:00:00 2001
From: Ali Asadi <10773383+maliasadi@users.noreply.github.com>
Date: Tue, 28 May 2024 15:42:39 -0400
Subject: [PATCH 3/7] Update CHANGELOG.md

---
 .github/CHANGELOG.md | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/.github/CHANGELOG.md b/.github/CHANGELOG.md
index 6c793bbbbb..eb80e04d9a 100644
--- a/.github/CHANGELOG.md
+++ b/.github/CHANGELOG.md
@@ -48,9 +48,9 @@
 * Changed the name of `lightning.tensor` to `default.tensor` with the `quimb` backend.
   [(#719)](https://github.com/PennyLaneAI/pennylane-lightning/pull/719)
 
-* Dispatch the C++ `Measurements.probs(wires)` method in Lightning-Qubit and Lighnting-Kokkos to `Measurements.probs()` when requesting the probabilities for all wires.
-  This will trigger a more optimized implementation for calculating the probabilities of the whole system.
-  [(#)]()
+* Patch the C++ `Measurements.probs(wires)` method in Lightning-Qubit and Lighnting-Kokkos to `Measurements.probs()` when called with all wires.
+  This will trigger a more optimized implementation for calculating the probabilities of the entire system.
+  [(#744)](https://github.com/PennyLaneAI/pennylane-lightning/pull/744)
 
 ### Documentation
 

From f039cf5150cc17d01dbbd8cc799b9714588dea6e Mon Sep 17 00:00:00 2001
From: Ali Asadi <10773383+maliasadi@users.noreply.github.com>
Date: Wed, 29 May 2024 10:22:56 -0400
Subject: [PATCH 4/7] trigger ci


From cb3a816fa59f59bdba287bdbf5dfa644194f04ec Mon Sep 17 00:00:00 2001
From: Ali Asadi <10773383+maliasadi@users.noreply.github.com>
Date: Wed, 29 May 2024 14:47:09 -0400
Subject: [PATCH 5/7] Update
 pennylane_lightning/core/src/simulators/lightning_kokkos/measurements/MeasurementsKokkos.hpp

Co-authored-by: Vincent Michaud-Rioux <vincentm@nanoacademic.com>
---
 .../lightning_kokkos/measurements/MeasurementsKokkos.hpp        | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pennylane_lightning/core/src/simulators/lightning_kokkos/measurements/MeasurementsKokkos.hpp b/pennylane_lightning/core/src/simulators/lightning_kokkos/measurements/MeasurementsKokkos.hpp
index f7b6132e95..73f9754777 100644
--- a/pennylane_lightning/core/src/simulators/lightning_kokkos/measurements/MeasurementsKokkos.hpp
+++ b/pennylane_lightning/core/src/simulators/lightning_kokkos/measurements/MeasurementsKokkos.hpp
@@ -515,7 +515,7 @@ class Measurements final
             "indices with probability calculations");
 
         // If all wires are requested, dispatch to `this->probs()`
-        if (wires.size() == this->_statevector.getNumQubits()) {
+        if (wires.empty() || wires.size() == this->_statevector.getNumQubits()) {
             return this->probs();
         }
 

From 8ad0aad428ebdad9dd9d0d3b15503dc5cdaef569 Mon Sep 17 00:00:00 2001
From: Ali Asadi <10773383+maliasadi@users.noreply.github.com>
Date: Wed, 29 May 2024 14:53:14 -0400
Subject: [PATCH 6/7] Update format

---
 .../lightning_kokkos/measurements/MeasurementsKokkos.hpp       | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/pennylane_lightning/core/src/simulators/lightning_kokkos/measurements/MeasurementsKokkos.hpp b/pennylane_lightning/core/src/simulators/lightning_kokkos/measurements/MeasurementsKokkos.hpp
index 73f9754777..6f8c82d5e2 100644
--- a/pennylane_lightning/core/src/simulators/lightning_kokkos/measurements/MeasurementsKokkos.hpp
+++ b/pennylane_lightning/core/src/simulators/lightning_kokkos/measurements/MeasurementsKokkos.hpp
@@ -515,7 +515,8 @@ class Measurements final
             "indices with probability calculations");
 
         // If all wires are requested, dispatch to `this->probs()`
-        if (wires.empty() || wires.size() == this->_statevector.getNumQubits()) {
+        if (wires.empty() ||
+            wires.size() == this->_statevector.getNumQubits()) {
             return this->probs();
         }
 

From acd1740256bdd9e931da9f108977cf632812ddec Mon Sep 17 00:00:00 2001
From: ringo-but-quantum <github-ringo-but-quantum@xanadu.ai>
Date: Wed, 29 May 2024 20:10:12 +0000
Subject: [PATCH 7/7] Auto update version from '0.37.0-dev21' to '0.37.0-dev22'

---
 pennylane_lightning/core/_version.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pennylane_lightning/core/_version.py b/pennylane_lightning/core/_version.py
index c29fafeccd..110ca74c5f 100644
--- a/pennylane_lightning/core/_version.py
+++ b/pennylane_lightning/core/_version.py
@@ -16,4 +16,4 @@
    Version number (major.minor.patch[-label])
 """
 
-__version__ = "0.37.0-dev21"
+__version__ = "0.37.0-dev22"