Struct DeviceBuffer 

pub struct DeviceBuffer<T: DeviceRepr> { /* private fields */ }

Owned, typed allocation of device memory (Runtime API).

Implementations

impl<T: DeviceRepr> DeviceBuffer<T>

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

Allocate an uninitialized buffer of len elements on the current device.

pub fn zeros(len: usize) -> Result<Self>

Allocate and zero-fill.

pub fn from_slice(src: &[T]) -> Result<Self>

Allocate and synchronously copy src from host memory.

pub fn copy_from_host(&self, src: &[T]) -> Result<()>

Synchronous H2D copy.

pub fn copy_to_host(&self, dst: &mut [T]) -> Result<()>

Synchronous D2H copy.

pub fn copy_from_host_async(&self, src: &[T], stream: &Stream) -> Result<()>

Asynchronous H2D copy on stream.

pub fn copy_to_host_async(&self, dst: &mut [T], stream: &Stream) -> Result<()>

Asynchronous D2H copy on stream.

pub fn len(&self) -> usize

Number of elements.

pub fn byte_size(&self) -> usize

Size in bytes.

pub fn is_empty(&self) -> bool

true if zero elements.

pub fn as_raw(&self) -> *mut c_void

Raw device pointer. Use with care.

pub fn as_device_ptr(&self) -> u64

Raw device pointer as the u64 value kernels expect. Convenience wrapper around as_raw.

impl<T: DeviceRepr> DeviceBuffer<T>

pub fn new_async(len: usize, stream: &Stream) -> Result<Self>

Asynchronously allocate len elements on stream from the device’s default memory pool (CUDA 11.2+).

pub fn free_async(self, stream: &Stream) -> Result<()>

Free this buffer asynchronously on stream. Consumes self so the sync Drop does not also free.

pub fn memset_async(&self, value: u8, stream: &Stream) -> Result<()>

Asynchronous memset of self to byte value value on stream.

Trait Implementations

impl<T: DeviceRepr> Debug for DeviceBuffer<T>

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

Formats the value using the given formatter.

impl<T: DeviceRepr> Drop for DeviceBuffer<T>

fn drop(&mut self)

Executes the destructor for this type.

impl<T: DeviceRepr + Send> Send for DeviceBuffer<T>