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

Check CUresult for mpi_data to device

Browse Source
This commit is contained in:
Gallo Alejandro 2022-09-12 18:41:37 +02:00
parent 1cd7bac187
commit f1b2f37fe2
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
include/atrip/Slice.hpp
View File

@ -21,6 +21,7 @@
#include <atrip/Tuples.hpp> #include <atrip/Tuples.hpp>
#include <atrip/Utils.hpp> #include <atrip/Utils.hpp>
#include <atrip/CUDA.hpp>
namespace atrip { namespace atrip {
@ -458,7 +459,11 @@ 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
WITH_CHRONO("cuda:memcpy", WITH_CHRONO("cuda:memcpy",
cuMemcpyHtoD(data, (void*)mpi_data, sizeof(F) * size);) _CHECK_CUDA_SUCCESS("copying mpi data to device",
cuMemcpyHtoD(data,
(void*)mpi_data,
sizeof(F) * size));
)
std::free(mpi_data); std::free(mpi_data);
#endif #endif