#include <rmm/cuda_device.hpp>
#include <rmm/cuda_stream_view.hpp>
#include <rmm/detail/error.hpp>
#include <rmm/detail/export.hpp>
#include <rmm/mr/device/per_device_resource.hpp>
#include <rmm/resource_ref.hpp>
#include <cuda_runtime_api.h>
#include <cassert>
#include <cstddef>
#include <utility>
类 | |
class | rmm::device_buffer |
用于设备内存分配的 RAII 结构。更多... | |