#include <rmm/cuda_device.hpp>
#include <rmm/cuda_stream_view.hpp>
#include <rmm/detail/export.hpp>
#include <rmm/error.hpp>
#include <cuda/std/span>
函数 | |
void | rmm::prefetch (void const *ptr, std::size_t size, rmm::cuda_device_id device, rmm::cuda_stream_view stream) |
在指定的流上将内存预取到指定的设备。 更多... | |
template<typename T > | |
void | rmm::prefetch (cuda::std::span< T const > data, rmm::cuda_device_id device, rmm::cuda_stream_view stream) |
在指定的流上将一段跨度内存预取到指定的设备。 更多... | |