missed some sync in the last commit
This commit is contained in:
parent
e1b9a56405
commit
7443a2ac6c
@ -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
|
||||
|
||||
@ -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
|
||||
|
||||
Loading…
Reference in New Issue
Block a user