Check CUresult for mpi_data to device

This commit is contained in:
Gallo Alejandro 2022-09-12 18:41:37 +02:00
parent 1cd7bac187
commit f1b2f37fe2

View File

@ -21,6 +21,7 @@
#include <atrip/Tuples.hpp>
#include <atrip/Utils.hpp>
#include <atrip/CUDA.hpp>
namespace atrip {
@ -458,7 +459,11 @@ void unwrapAndMarkReady() {
#if defined(HAVE_CUDA)
// copy the retrieved mpi data to the device
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);
#endif