Struct DeviceBox

Source

pub struct DeviceBox<'a> { /* private fields */ }

Expand description

An owned device-allocated buffer

Implementations§

Source §

impl<'a> DeviceBox<'a>

Source

pub fn alloc(handle: &Rc<Handle<'a>>, size: u64) -> CudaResult<Self>

Allocate an uninitialized buffer of size size on the device

Source

pub fn new(handle: &Rc<Handle<'a>>, input: &[u8]) -> CudaResult<Self>

Allocate a new initialized buffer on the device matching the size and content of input.

Source

pub fn new_stream<'b>( handle: &Rc<Handle<'a>>, input: &'b [u8], stream: &'b mut Stream<'a>, ) -> CudaResult<Self>

Allocates a new uninitialized buffer on the device, then asynchronously fills it with input. input must not be dropped or mutated until stream.sync is called. Does not allocate the memory asynchronously.

Source

pub fn new_stream_buf( handle: &Rc<Handle<'a>>, input: Vec<u8>, stream: &mut Stream<'a>, ) -> CudaResult<Self>

Allocates a new uninitialized buffer on the device, then synchronously fills it with input. input will be dropped when the stream is synced or dropped. Does not allocate the memory asynchronously.

Source

pub fn new_ffi<T>(handle: &Rc<Handle<'a>>, input: &[T]) -> CudaResult<Self>

Allocates a new initialized buffer on the device matching the size and content of input. Note that memory is directly copied, so [T] must be Sized should not contain any pointers, references, unsized types, or other non-FFI safe types.

Source

pub fn new_ffi_stream<'b, T>( handle: &Rc<Handle<'a>>, input: &'b [T], stream: &'b mut Stream<'a>, ) -> CudaResult<Self>

Allocates a new uninitialized buffer on the device, then synchronously fills it with input. Note that memory is directly copied, so [T] must be Sized should not contain any pointers, references, unsized types, or other non-FFI safe types. input must not be dropped or mutated until stream.sync is called. Does not allocate the memory asynchronously.

Source

pub fn new_ffi_stream_buf<'b, T>( handle: &Rc<Handle<'a>>, input: Vec<T>, stream: &'b mut Stream<'a>, ) -> CudaResult<Self>

Allocates a new uninitialized buffer on the device, then synchronously fills it with input. Note that memory is directly copied, so [T] must be Sized should not contain any pointers, references, unsized types, or other non-FFI safe types. input will be dropped when the stream is synced or dropped. Does not allocate the memory asynchronously.

Source

pub fn leak(self)

Leaks the DeviceBox, similar to Box::leak.

Source

pub unsafe fn from_raw(raw: DevicePtr<'a>) -> Self

Constructs a DeviceBox from a device pointer.

Methods from Deref<Target = DevicePtr<'a>>§

Source

pub fn as_raw(&self) -> u64

Source

pub fn copy_to<'b>(&self, target: &DevicePtr<'b>) -> CudaResult<()>

Synchronously copies data from self to target. Panics if length is not equal.

Source

pub fn copy_to_stream<'b, 'c: 'b + 'a>( &self, target: &DevicePtr<'b>, stream: &mut Stream<'c>, ) -> CudaResult<()>
where 'a: 'b,

Asynchronously copies data from self to target. Panics if length is not equal.

Source

pub fn copy_from<'b>(&self, source: &DevicePtr<'b>) -> CudaResult<()>

Synchronously copies data from source to self. Panics if length is not equal.

Source

pub fn copy_from_stream<'b: 'a, 'c: 'a + 'b>( &self, source: &DevicePtr<'b>, stream: &mut Stream<'c>, ) -> CudaResult<()>

Asynchronously copies data from source to self. Panics if length is not equal.

Source

pub fn subslice(&self, from: u64, to: u64) -> Self

Gets a subslice of this slice from [from:to]

Source

pub fn len(&self) -> u64

Gets the length of this slice

Source

pub fn is_empty(&self) -> bool

Check if the slice’s length is 0

Source

pub fn load(&self) -> CudaResult<Vec<u8>>

Synchronously loads the data from this slice into a local buffer

Source

pub unsafe fn load_stream(&self, stream: &mut Stream<'a>) -> CudaResult<Vec<u8>>

Asynchronously loads the data from this slice into a local buffer. The contents of the buffer are undefined until stream.sync is called. The output must not be dropped until the stream is synced.

Source