Add a chrono for mpi memcpy in cuda

2022-09-08 15:27:51 +02:00 · 2022-09-08 15:27:51 +02:00 · 49ff3b377c
commit 49ff3b377c
parent 00a28c990c
1 changed files with 2 additions and 1 deletions
--- 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