Struct CudaTensorLinalgBackend 

pub struct CudaTensorLinalgBackend;

Marker type for the CUDA tensor linalg backend.

Examples

let _backend = tenferro_linalg_prims::backend::CudaTensorLinalgBackend;

Trait Implementations

impl Clone for CudaTensorLinalgBackend

fn clone(&self) -> CudaTensorLinalgBackend

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for CudaTensorLinalgBackend

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

Formats the value using the given formatter.

impl Default for CudaTensorLinalgBackend

fn default() -> CudaTensorLinalgBackend

Returns the “default value” for a type.

impl<T: CudaLinalgScalar> TensorLinalgPrims<T> for CudaTensorLinalgBackend

type Context = CudaContext

fn has_linalg_support(op: LinalgCapabilityOp) -> bool

fn solve_ex( ctx: &mut Self::Context, a: &Tensor<T>, b: &Tensor<T>, ) -> Result<SolveTensorExResult<T>>

Solve a square linear system while returning per-batch numerical status.

fn solve( ctx: &mut Self::Context, a: &Tensor<T>, b: &Tensor<T>, ) -> Result<Tensor<T>>

fn lu_solve( ctx: &mut Self::Context, factors: &Tensor<T>, pivots: &Tensor<i32>, b: &Tensor<T>, ) -> Result<Tensor<T>>

fn solve_triangular( ctx: &mut Self::Context, a: &Tensor<T>, b: &Tensor<T>, upper: bool, ) -> Result<Tensor<T>>

fn qr(_ctx: &mut Self::Context, _a: &Tensor<T>) -> Result<QrTensorResult<T>>

fn thin_svd( ctx: &mut Self::Context, a: &Tensor<T>, ) -> Result<SvdTensorResult<T>>

fn svdvals(_ctx: &mut Self::Context, _a: &Tensor<T>) -> Result<Tensor<T::Real>>

fn lu_factor_ex( _ctx: &mut Self::Context, _a: &Tensor<T>, ) -> Result<LuTensorExResult<T>>

Compute an LU factorization while returning per-batch numerical status.

fn lu_factor( _ctx: &mut Self::Context, _a: &Tensor<T>, ) -> Result<LuTensorResult<T>>

fn lu_factor_no_pivot( _ctx: &mut Self::Context, _a: &Tensor<T>, ) -> Result<LuTensorResult<T>>

Compute an LU factorization without pivoting.

fn cholesky_ex( _ctx: &mut Self::Context, _a: &Tensor<T>, ) -> Result<CholeskyTensorExResult<T>>

Compute a Cholesky factorization while returning per-batch numerical status.

fn cholesky(ctx: &mut Self::Context, a: &Tensor<T>) -> Result<Tensor<T>>

fn eigen_sym( _ctx: &mut Self::Context, _a: &Tensor<T>, ) -> Result<EigenTensorResult<T>>

fn eig(_ctx: &mut Self::Context, _a: &Tensor<T>) -> Result<EigTensorResult<T>>

impl Copy for CudaTensorLinalgBackend