tidy up code

PennyLaneAI · Oct 10, 2024 · 0e9f1db · 0e9f1db
1 parent 85dc575
commit 0e9f1db
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/pennylane_lightning/core/src/simulators/lightning_gpu/StateVectorCudaMPI.hpp b/pennylane_lightning/core/src/simulators/lightning_gpu/StateVectorCudaMPI.hpp
@@ -1685,7 +1685,7 @@ class StateVectorCudaMPI final
 
         double norm = (branch == 0) ? abs2sum0 : abs2sum1;
 
-        int parity = branch;
+        const int parity = static_cast<int>(branch);
 
         PL_CUSTATEVEC_IS_SUCCESS(custatevecCollapseOnZBasis(
             /* custatevecHandle_t */ handle_.get(),

diff --git a/pennylane_lightning/core/src/simulators/lightning_gpu/StateVectorCudaManaged.hpp b/pennylane_lightning/core/src/simulators/lightning_gpu/StateVectorCudaManaged.hpp
@@ -522,7 +522,7 @@ class StateVectorCudaManaged
 
         double norm = (branch == 0) ? abs2sum0 : abs2sum1;
 
-        int parity = branch;
+        const int parity = static_cast<int>(branch);
 
         PL_CUSTATEVEC_IS_SUCCESS(custatevecCollapseOnZBasis(
             /* custatevecHandle_t */ handle_.get(),

diff --git a/pennylane_lightning/lightning_gpu/lightning_gpu.py b/pennylane_lightning/lightning_gpu/lightning_gpu.py
@@ -514,7 +514,7 @@ def simulate(
                 raise qml.DeviceError("LightningGPU-MPI does not support Mid-circuit measurements.")
 
             results = []
-            aux_circ = qml.tape.QuantumScript(
+            aux_circ = QuantumScript(
                 circuit.operations,
                 circuit.measurements,
                 shots=[1],