gallo/atrip
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add a chrono for mpi memcpy in cuda

Browse Source
This commit is contained in:
Alejandro Gallo 2022-09-08 15:27:51 +02:00
parent 00a28c990c
commit 49ff3b377c
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
include/atrip/Slice.hpp
View File

@ -457,7 +457,8 @@ void unwrapAndMarkReady() {
#if defined(HAVE_CUDA) #if defined(HAVE_CUDA)
// copy the retrieved mpi data to the device // 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); std::free(mpi_data);
#endif #endif