Fix setBasisState.

PennyLaneAI · Apr 2, 2024 · cc15f3d · cc15f3d
1 parent 2359274
commit cc15f3d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pennylane_lightning/core/src/simulators/lightning_kokkos/StateVectorKokkosMPI.hpp b/pennylane_lightning/core/src/simulators/lightning_kokkos/StateVectorKokkosMPI.hpp
@@ -537,7 +537,7 @@ class StateVectorKokkosMPI final
     void setBasisState(const std::size_t global_index) {
         const auto index = global_2_local_index(global_index);
         const auto rank = static_cast<std::size_t>(get_mpi_rank());
-        if (index.first == 0) {
+        if (index.first == rank) {
             (*sv_).setBasisState(index.second);
         } else {
             (*sv_).initZeros();