Check CUresult for mpi_data to device
This commit is contained in:
parent
1cd7bac187
commit
f1b2f37fe2
@ -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
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user