Skip to main content

GemmDispatchCtx

Type Alias GemmDispatchCtx 

Source
pub type GemmDispatchCtx<'a> = BlasDispatchCtx<'a>;
Expand description

GemmDispatchCtx is now an alias for BlasDispatchCtx (the cuBLAS agent unified the per-op contexts since every cuBLAS dispatcher needs the same set: cuBLAS handle, stream, completion, state).

Aliased Type§

pub struct GemmDispatchCtx<'a> {
    pub cublas: &'a Arc<CudaBlas>,
    pub stream: &'a Arc<CudaStream>,
    pub completion: &'a Arc<dyn CompletionStrategy>,
    pub state: &'a Arc<DeviceState>,
}

Fields§

§cublas: &'a Arc<CudaBlas>§stream: &'a Arc<CudaStream>§completion: &'a Arc<dyn CompletionStrategy>§state: &'a Arc<DeviceState>