Class DevMemInfo

Simple object that describes a buffer in device memory.

Public Functions

DevMemInfo(std::shared_ptr<rmm::device_buffer> buffer, DType dtype, std::vector<std::size_t> shape, std::vector<std::size_t> stride, size_t offset_bytes = 0)

DevMemInfo(DevMemInfo &&other) = default

std::size_t bytes() const

std::size_t count() const

std::size_t offset_bytes() const

const DType &dtype() const

TypeId type_id() const

const std::vector<std::size_t> &shape() const

std::size_t shape(std::size_t idx) const

const std::vector<std::size_t> &stride() const

std::size_t stride(std::size_t idx) const

void *data() const

Returns raw pointer to underlying buffer offset by the offset

Returns

void*

std::unique_ptr<rmm::device_buffer> make_new_buffer(std::size_t bytes) const

Constructs a new rmm buffer with the same stream and memory resource as the current buffer.

Parameters

bytes –

Returns

std::unique_ptr<rmm::device_buffer>