Struct CudaPtr

pub struct CudaPtr<T> { /* private fields */ }

Wrapper to handle cuda device memory

Implementations

impl<T> CudaPtr<T>

pub fn new(size: usize) -> Result<CudaPtr<T>, CudaError>

Create an instance of CudaPtr

Arguments

• size- Number of value elements to be allocated

Errors

This function may return the following errors

• CudaError

Trait Implementations

impl<T> AsMutPtr<T> for CudaPtr<T>

fn as_mut_ptr(&mut self) -> *mut T

impl<T> AsMutVoidPtr for CudaPtr<T>

fn as_mut_void_ptr(&mut self) -> *mut c_void

impl<T> AsPtr<T> for CudaPtr<T>

fn as_ptr(&self) -> *const T

impl<T> AsVoidPtr for CudaPtr<T>

fn as_void_ptr(&self) -> *const c_void

impl<U, P, OP, I, PI, const N: usize> BatchNormalizationLayerInstantiation<U, CudaTensor1dPtr<U, N>, P, OP, DeviceGpu<U>, I, PI, CudaPtr<U>, N> for BatchNormalizationLayer<U, CudaTensor1dPtr<U, N>, P, OP, DeviceGpu<U>, I, PI, CudaPtr<U>, N>

where P: ForwardAll<Input = I, Output = PI> + BackwardAll<U, LossInput = PI> + PreTrain<U> + Loss<U>, U: Default + Clone + Copy + Send + UnitValue<U>, I: Debug + Send + Sync, OP: Optimizer<U, DeviceGpu<U>>, DeviceGpu<U>: Device<U>,

fn with_params<B: OptimizerBuilder<U, DeviceGpu<U>, Output = OP>>( parent: P, device: &DeviceGpu<U>, scale: Arr<U, N>, bias: Arr<U, N>, momentum: U, b: &B, ) -> Result<BatchNormalizationLayer<U, CudaTensor1dPtr<U, N>, P, OP, DeviceGpu<U>, I, PI, CudaPtr<U>, N>, LayerInstantiationError>

Create and return an instance with the specified scale, bias, and momentum.

fn with_momentum<B: OptimizerBuilder<U, DeviceGpu<U>, Output = OP>>( parent: P, device: &DeviceGpu<U>, momentum: U, b: &B, ) -> Result<BatchNormalizationLayer<U, CudaTensor1dPtr<U, N>, P, OP, DeviceGpu<U>, I, PI, CudaPtr<U>, N>, LayerInstantiationError>

Create and return an instance with the momentum.

fn new<B: OptimizerBuilder<U, DeviceGpu<U>, Output = OP>>( parent: P, device: &DeviceGpu<U>, b: &B, ) -> Result<BatchNormalizationLayer<U, CudaTensor1dPtr<U, N>, P, OP, DeviceGpu<U>, I, PI, CudaPtr<U>, N>, LayerInstantiationError>

Create and return an instance.

impl<T: Debug> Debug for CudaPtr<T>

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

Formats the value using the given formatter.

impl<T> Drop for CudaPtr<T>

fn drop(&mut self)

Executes the destructor for this type.

impl<CP> MemoryMoveTo<<CP as PointerElement>::Element, CudaPtr<<CP as PointerElement>::Element>> for CP

where CP: AsCudaPtrRef + PointerElement, <CP as AsCudaPtrRef>::Pointer: MemoryMoveTo<<CP as PointerElement>::Element, CudaPtr<<CP as PointerElement>::Element>>,

fn memcpy_to( &self, dst: &mut CudaPtr<<CP as PointerElement>::Element>, len: usize, ) -> Result<usize, Error>

Memory Copy To

impl<T: Default + Debug> MemoryMoveTo<T, CudaHostPtr<T>> for CudaPtr<T>

fn memcpy_to( &self, dst: &mut CudaHostPtr<T>, len: usize, ) -> Result<usize, Error>

Memory Copy To

impl<T: Default + Debug> MemoryMoveTo<T, CudaMemoryPoolPtr<T>> for CudaPtr<T>

fn memcpy_to( &self, dst: &mut CudaMemoryPoolPtr<T>, len: usize, ) -> Result<usize, Error>

Memory Copy To

impl<T: Default + Debug> MemoryMoveTo<T, CudaPtr<T>> for CudaHostPtr<T>

fn memcpy_to(&self, dst: &mut CudaPtr<T>, len: usize) -> Result<usize, Error>

Memory Copy To

impl<T: Default + Debug> MemoryMoveTo<T, CudaPtr<T>> for CudaMemoryPoolPtr<T>

fn memcpy_to(&self, dst: &mut CudaPtr<T>, len: usize) -> Result<usize, Error>

Memory Copy To

impl<T: Default + Debug> MemoryMoveTo<T, CudaPtr<T>> for CudaPtr<T>

fn memcpy_to(&self, dst: &mut CudaPtr<T>, len: usize) -> Result<usize, Error>

Memory Copy To

impl<CP> MemoryMoveToAsync<<CP as PointerElement>::Element, CudaPtr<<CP as PointerElement>::Element>> for CP

where CP: AsCudaPtrRef + PointerElement, <CP as AsCudaPtrRef>::Pointer: MemoryMoveToAsync<<CP as PointerElement>::Element, CudaPtr<<CP as PointerElement>::Element>>,

fn memcpy_to_async( &self, dst: &mut CudaPtr<<CP as PointerElement>::Element>, len: usize, stream: cudaStream_t, ) -> Result<usize, Error>

Memory Copy To

impl<T: Default + Debug> MemoryMoveToAsync<T, CudaPtr<T>> for CudaHostPtr<T>

fn memcpy_to_async( &self, dst: &mut CudaPtr<T>, len: usize, stream: cudaStream_t, ) -> Result<usize, Error>

Memory Copy To

impl<T> PointerElement for CudaPtr<T>

where T: Default + Debug,

type Element = T

impl<T: Default + Debug> ReadMemory<T> for CudaPtr<T>

fn read_to_vec(&self) -> Result<Vec<T>, Error>

Errors

fn read_to_vec_with_size(&self, size: usize) -> Result<Vec<T>, Error>

Read memory as Vec with size specified

impl TryFrom<f32> for CudaPtr<f32>

type Error = CudaError

The type returned in the event of a conversion error.

fn try_from(value: f32) -> Result<Self, Self::Error>

Performs the conversion.

impl TryFrom<f64> for CudaPtr<f64>

type Error = CudaError

The type returned in the event of a conversion error.

fn try_from(value: f64) -> Result<Self, Self::Error>

Performs the conversion.

impl TryFrom<i32> for CudaPtr<i32>

type Error = CudaError

The type returned in the event of a conversion error.

fn try_from(value: i32) -> Result<Self, Self::Error>

Performs the conversion.

impl TryFrom<i64> for CudaPtr<i64>

type Error = CudaError

The type returned in the event of a conversion error.

fn try_from(value: i64) -> Result<Self, Self::Error>

Performs the conversion.

impl<T: Default + Debug> WriteMemory<T> for CudaPtr<T>

fn memcpy(&mut self, p: *const T, len: usize) -> Result<usize, Error>

Memory Copy

fn memcpy_repeat( &mut self, p: *const T, len: usize, count: usize, ) -> Result<usize, Error>

Repeatedly copy the contents of memory