device sync instead of stream sync

WeiqunZhang · Aug 12, 2023 · 9f42e9f · 9f42e9f
1 parent d0c613a
commit 9f42e9f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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);