Trait cuda::blas::CublasLevel3

pub trait CublasLevel3<T> {
    unsafe fn gemm(
        &mut self, 
        transpose_a: CublasTranspose, 
        transpose_b: CublasTranspose, 
        rows: i32, 
        cols: i32, 
        inner: i32, 
        alpha: *const T, 
        a: *const T, 
        lda: i32, 
        b: *const T, 
        ldb: i32, 
        beta: *const T, 
        c: *mut T, 
        ldc: i32
    ) -> CublasResult;
}

Required methods

unsafe fn gemm(
    &mut self,
    transpose_a: CublasTranspose,
    transpose_b: CublasTranspose,
    rows: i32,
    cols: i32,
    inner: i32,
    alpha: *const T,
    a: *const T,
    lda: i32,
    b: *const T,
    ldb: i32,
    beta: *const T,
    c: *mut T,
    ldc: i32
) -> CublasResult

Loading content...

Implementors

impl CublasLevel3<f32> for CublasHandle

unsafe fn gemm(
    &mut self,
    transpose_a: CublasTranspose,
    transpose_b: CublasTranspose,
    rows: i32,
    cols: i32,
    inner_dim: i32,
    alpha: *const f32,
    a: *const f32,
    lda: i32,
    b: *const f32,
    ldb: i32,
    beta: *const f32,
    c: *mut f32,
    ldc: i32
) -> CublasResult

impl CublasLevel3<f64> for CublasHandle

unsafe fn gemm(
    &mut self,
    transpose_a: CublasTranspose,
    transpose_b: CublasTranspose,
    rows: i32,
    cols: i32,
    inner_dim: i32,
    alpha: *const f64,
    a: *const f64,
    lda: i32,
    b: *const f64,
    ldb: i32,
    beta: *const f64,
    c: *mut f64,
    ldc: i32
) -> CublasResult

impl CublasLevel3<__half> for CublasHandle

unsafe fn gemm(
    &mut self,
    transpose_a: CublasTranspose,
    transpose_b: CublasTranspose,
    rows: i32,
    cols: i32,
    inner_dim: i32,
    alpha: *const __half,
    a: *const __half,
    lda: i32,
    b: *const __half,
    ldb: i32,
    beta: *const __half,
    c: *mut __half,
    ldc: i32
) -> CublasResult

Loading content...