Struct BackendDescriptor 

pub struct BackendDescriptor { /* private fields */ }

RAII wrapper around a cudnnBackendDescriptor_t that destroys the descriptor on drop.

Implementations

impl BackendDescriptor

pub fn create(kind: cudnnBackendDescriptorType_t) -> Result<Self, GpuError>

cudnnBackendCreateDescriptor.

pub fn as_raw(&self) -> cudnnBackendDescriptor_t

Raw handle (only valid until Drop).

pub fn finalize(&mut self) -> Result<(), GpuError>

cudnnBackendFinalize.

pub fn is_finalized(&self) -> bool

True iff finalize() has succeeded.

pub unsafe fn set_attribute_raw( &mut self, name: cudnnBackendAttributeName_t, attr_type: cudnnBackendAttributeType_t, count: i64, data: *const c_void, ) -> Result<(), GpuError>

Generic cudnnBackendSetAttribute — caller supplies the element type and a *const c_void pointer to a contiguous array of count elements.

Safety

data must point to count valid elements of the type expected by attr_type. The descriptor must outlive the call.

pub fn set_i64( &mut self, name: cudnnBackendAttributeName_t, value: i64, ) -> Result<(), GpuError>

Set a single i64 attribute.

pub fn set_i64_array( &mut self, name: cudnnBackendAttributeName_t, values: &[i64], ) -> Result<(), GpuError>

Set an array of i64 attributes.

pub fn set_f32( &mut self, name: cudnnBackendAttributeName_t, value: f32, ) -> Result<(), GpuError>

Set a single f32 attribute (e.g. an alpha scaling parameter).

pub fn set_f64( &mut self, name: cudnnBackendAttributeName_t, value: f64, ) -> Result<(), GpuError>

Set a single f64 attribute.

pub fn set_dev_ptr( &mut self, name: cudnnBackendAttributeName_t, ptr: *mut c_void, ) -> Result<(), GpuError>

Set a single device-pointer attribute.

pub fn set_descriptor( &mut self, name: cudnnBackendAttributeName_t, sub: &BackendDescriptor, ) -> Result<(), GpuError>

Set a single sub-descriptor reference.

pub fn set_descriptors( &mut self, name: cudnnBackendAttributeName_t, subs: &[&BackendDescriptor], ) -> Result<(), GpuError>

Set an array of sub-descriptor references.

pub fn set_data_type( &mut self, name: cudnnBackendAttributeName_t, dt: cudnnDataType_t, ) -> Result<(), GpuError>

Set a cudnnDataType_t attribute.

Trait Implementations

impl Debug for BackendDescriptor

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

Formats the value using the given formatter.

impl Drop for BackendDescriptor

fn drop(&mut self)

Executes the destructor for this type.

impl Send for BackendDescriptor