Struct GemmStridedBatchedRequest 

pub struct GemmStridedBatchedRequest<T: GemmSupported> {

    pub a: GpuRef<T>,
    pub b: GpuRef<T>,
    pub c: GpuRef<T>,
    pub m: i32,
    pub n: i32,
    pub k: i32,
    pub alpha: T,
    pub beta: T,
    pub trans_a: cublasOperation_t,
    pub trans_b: cublasOperation_t,
    pub lda: i32,
    pub ldb: i32,
    pub ldc: i32,
    pub stride_a: i64,
    pub stride_b: i64,
    pub stride_c: i64,
    pub batch_size: i32,
    pub reply: Sender<Result<(), GpuError>>,
}

Typed strided-batched gemm request. Per-batch strides describe the element offset between consecutive batch entries inside a single allocation.

Fields

a: GpuRef<T>

b: GpuRef<T>

c: GpuRef<T>

m: i32

n: i32

k: i32

alpha: T

beta: T

trans_a: cublasOperation_t

trans_b: cublasOperation_t

lda: i32

ldb: i32

ldc: i32

stride_a: i64

stride_b: i64

stride_c: i64

batch_size: i32

reply: Sender<Result<(), GpuError>>

Trait Implementations

impl GemmStridedBatchedDispatch for GemmStridedBatchedRequest<f32>

fn dtype_name(&self) -> &'static str

fn op_name(&self) -> &'static str

fn dispatch(self: Box<Self>, ctx: &BlasDispatchCtx<'_>)

impl GemmStridedBatchedDispatch for GemmStridedBatchedRequest<f64>

fn dtype_name(&self) -> &'static str

fn op_name(&self) -> &'static str

fn dispatch(self: Box<Self>, ctx: &BlasDispatchCtx<'_>)