Struct nncombinator::device::CublasContext

pub struct CublasContext { /* private fields */ }

cublas context

Implementations

impl CublasContext

pub fn new(pointer_mode: PointerMode) -> Result<CublasContext, Error>

Create an instance of CublasContext

Arguments

• pointer_mode - Host or Device

Errors

This function may return the following errors

• rcublas::error::Error

pub fn id_c(&self) -> &cublasHandle_t

Returns a reference to the raw handle (pointer) of the cublas context

pub fn pointer_mode(&self) -> Result<PointerMode, Error>

Returns the PointerMode that has been set.

Errors

This function may return the following errors

• rcublas::error::Error

Trait Implementations

impl Clone for CublasContext

fn clone(&self) -> Self

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.