Add barrier.

PennyLaneAI · Oct 26, 2023 · 9d472b3 · 9d472b3
1 parent edc8530
commit 9d472b3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pennylane_lightning/core/src/simulators/lightning_gpu/utils/MPILinearAlg.hpp b/pennylane_lightning/core/src/simulators/lightning_gpu/utils/MPILinearAlg.hpp
@@ -113,8 +113,9 @@ inline void SparseMV_cuSparseMPI(
                                           length_local, reduce_root_rank,
                                           "sum");
         }
+        PL_CUDA_IS_SUCCESS(cudaDeviceSynchronize());
+        mpi_manager.Barrier();
     }
-    mpi_manager.Barrier();
 }
 
 } // namespace Pennylane::LightningGPU::Util