Class DevMemInfo

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>

© Copyright 2023, NVIDIA. Last updated on Feb 3, 2023.