diff --git a/include/atrip/Slice.hpp b/include/atrip/Slice.hpp index 47ff306..34a12bc 100644 --- a/include/atrip/Slice.hpp +++ b/include/atrip/Slice.hpp @@ -457,7 +457,8 @@ void unwrapAndMarkReady() { #if defined(HAVE_CUDA) // copy the retrieved mpi data to the device - cuMemcpyHtoD(data, (void*)mpi_data, sizeof(F) * size); + WITH_CHRONO("cuda:memcpy", + cuMemcpyHtoD(data, (void*)mpi_data, sizeof(F) * size);) std::free(mpi_data); #endif