From 9f42e9f6d40b9a0ae3447c823081ca8fe807e3d0 Mon Sep 17 00:00:00 2001
From: Weiqun Zhang <weiqunzhang@lbl.gov>
Date: Sat, 12 Aug 2023 17:13:24 -0400
Subject: [PATCH] device sync instead of stream sync

---
 Src/Base/AMReX_Random.cpp | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/Src/Base/AMReX_Random.cpp b/Src/Base/AMReX_Random.cpp
index f9a15f34239..7dda1c036bc 100644
--- a/Src/Base/AMReX_Random.cpp
+++ b/Src/Base/AMReX_Random.cpp
@@ -247,6 +247,8 @@ void FillRandomNormal (Real* p, Long N, Real mean, Real stddev)
     AMREX_CURAND_SAFE_CALL(curandGenerateNormalDouble(cuda_rand_gen, p, N, mean, stddev));
 #endif
 
+    Gpu::synchronize();
+
 #elif defined(AMREX_USE_HIP)
 
     if (! generator_initialized) {
@@ -262,9 +264,9 @@ void FillRandomNormal (Real* p, Long N, Real mean, Real stddev)
     AMREX_HIPRAND_SAFE_CALL(hiprandGenerateNormalDouble(hip_rand_gen, p, N, mean, stddev));
 #endif
 
-#endif
+    Gpu::synchronize();
 
-    Gpu::streamSynchronize();
+#endif
 
 #else
     std::normal_distribution<Real> distribution(mean, stddev);