update C++ layer

PennyLaneAI · Oct 15, 2024 · ab021b7 · ab021b7
1 parent 168b436
commit ab021b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pennylane_lightning/core/src/simulators/lightning_gpu/StateVectorCudaManaged.hpp b/pennylane_lightning/core/src/simulators/lightning_gpu/StateVectorCudaManaged.hpp
@@ -520,7 +520,7 @@ class StateVectorCudaManaged
             /* const int32_t * */ basisBits.data(),
             /* const uint32_t nBasisBits */ basisBits.size()));
 
-        double norm = (branch == 0) ? abs2sum0 : abs2sum1;
+        double norm = branch ? abs2sum1 : abs2sum0;
 
         const int parity = static_cast<int>(branch);