Skip to content

Commit

Permalink
feat: add RAJA device synchronize options
Browse files Browse the repository at this point in the history
  • Loading branch information
pranav-sivaraman committed Jan 16, 2024
1 parent c706e05 commit a01d17e
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions src/raja/fasten.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -255,10 +255,13 @@ template <size_t PPWI> class IMPL_CLS final : public Bude<PPWI> {
static void synchronise() {
// nothing to do for host devices
#if defined(RAJA_ENABLE_CUDA)
cudaDeviceSynchronize();
RAJA::synchronize<RAJA::cuda_synchronize>();
#endif
#if defined(RAJA_ENABLE_HIP)
hipDeviceSynchronize();
RAJA::synchronize<RAJA::hip_synchronize>();
#endif
#if defined(RAJA_ENABLE_SYCL)
RAJA::synchronize<RAJA::sycl_synchronize>();
#endif
}

Expand Down

0 comments on commit a01d17e

Please sign in to comment.