Struct DeviceStorage 

pub struct DeviceStorage { /* private fields */ }

CUDA device memory allocated via cudaMalloc.

Implementations

impl DeviceStorage

pub fn new(len: usize, device_id: u32) -> Result<Self>

Allocate new device memory of the given size.

Arguments

• len - Size in bytes to allocate
• device_id - CUDA device on which to allocate

pub fn device_ptr(&self) -> u64

Get the device pointer value.

pub fn device_id(&self) -> u32

Get the CUDA device ID this memory is allocated on.

Trait Implementations

impl Debug for DeviceStorage

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Drop for DeviceStorage

fn drop(&mut self)

Executes the destructor for this type.

impl MemoryDescriptor for DeviceStorage

fn addr(&self) -> usize

Base address of the memory region.

fn size(&self) -> usize

Size of the memory region in bytes.

fn storage_kind(&self) -> StorageKind

Type of storage backing this region.

fn as_any(&self) -> &dyn Any

Enable downcasting to concrete type.

fn nixl_descriptor(&self) -> Option<NixlDescriptor>

Get the NIXL descriptor for this memory region.

impl NixlCompatible for DeviceStorage

fn nixl_params(&self) -> (*const u8, usize, MemType, u64)

Get parameters needed for NIXL registration.

impl Send for DeviceStorage

impl Sync for DeviceStorage