cuda_memcpy_async.hpp

 
#pragma once
 
#include <cstddef>
 
#include <cudf/detail/utilities/cuda_memcpy.hpp>
#include <rmm/cuda_stream_view.hpp>
 
namespace rapidsmpf {
 
[[nodiscard]] inline cudaError_t cuda_memcpy_async(
    void* dst, void const* src, std::size_t count, rmm::cuda_stream_view stream
) {
    return cudf::detail::memcpy_async(dst, src, count, stream);
}
 
}  // namespace rapidsmpf