Defined in File dev_mem_info.hpp
-
class DevMemInfo
Simple object that describes a buffer in device memory.
Public Functions
-
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>
-
DevMemInfo(DevMemInfo &&other) = default