missed some sync in the last commit

This commit is contained in:
Ruipeng Li 2022-03-07 23:54:03 -08:00
parent e1b9a56405
commit 7443a2ac6c
2 changed files with 5 additions and 5 deletions

View File

@ -147,7 +147,7 @@ hypre_BoomerAMGBuildDirInterpDevice( hypre_ParCSRMatrix *A,
CF_marker,
int_buf_data );
#if defined(HYPRE_WITH_GPU_AWARE_MPI)
#if defined(HYPRE_WITH_GPU_AWARE_MPI) && THRUST_CALL_BLOCKING == 0
/* RL: make sure int_buf_data is ready before issuing GPU-GPU MPI */
hypre_ForceSyncComputeStream(hypre_handle());
#endif
@ -171,7 +171,7 @@ hypre_BoomerAMGBuildDirInterpDevice( hypre_ParCSRMatrix *A,
dof_func,
int_buf_data );
#if defined(HYPRE_WITH_GPU_AWARE_MPI)
#if defined(HYPRE_WITH_GPU_AWARE_MPI) && THRUST_CALL_BLOCKING == 0
/* RL: make sure int_buf_data is ready before issuing GPU-GPU MPI */
hypre_ForceSyncComputeStream(hypre_handle());
#endif
@ -1136,7 +1136,7 @@ hypre_BoomerAMGBuildInterpOnePntDevice( hypre_ParCSRMatrix *A,
CF_marker,
int_buf_data );
#if defined(HYPRE_WITH_GPU_AWARE_MPI)
#if defined(HYPRE_WITH_GPU_AWARE_MPI) && THRUST_CALL_BLOCKING == 0
/* RL: make sure int_buf_data is ready before issuing GPU-GPU MPI */
hypre_ForceSyncComputeStream(hypre_handle());
#endif
@ -1190,7 +1190,7 @@ hypre_BoomerAMGBuildInterpOnePntDevice( hypre_ParCSRMatrix *A,
big_int_buf_data,
thrust::plus<HYPRE_BigInt>() );
#if defined(HYPRE_WITH_GPU_AWARE_MPI)
#if defined(HYPRE_WITH_GPU_AWARE_MPI) && THRUST_CALL_BLOCKING == 0
/* RL: make sure big_int_buf_data is ready before issuing GPU-GPU MPI */
hypre_ForceSyncComputeStream(hypre_handle());
#endif

View File

@ -275,7 +275,7 @@ hypre_ParCSRMatrixMatvecOutOfPlace( HYPRE_Complex alpha,
hypre_profile_times[HYPRE_TIMER_ID_PACK_UNPACK] += hypre_MPI_Wtime();
#endif
#if defined(HYPRE_USING_GPU) && defined(HYPRE_WITH_GPU_AWARE_MPI)
#if defined(HYPRE_USING_GPU) && defined(HYPRE_WITH_GPU_AWARE_MPI) && THRUST_CALL_BLOCKING == 0
/* RL: make sure x_buf_data is ready before issuing GPU-GPU MPI */
hypre_ForceSyncComputeStream(hypre_handle());
#endif