Skip to main content

CudnnDispatch

atomr_accel_cuda::kernel::dispatch

Trait CudnnDispatch 

Source 
pub trait CudnnDispatch: Send + 'static {
    // Required methods
    fn dtype_name(&self) -> &'static str;
    fn op_kind(&self) -> &'static str;
    fn dispatch(self: Box<Self>, ctx: &CudnnDispatchCtx<'_>);
}
Expand description

CudnnDispatch is owned by kernel::cudnn (Phase 2 cuDNN). Re-exported here for symmetry with other actors’ dispatch traits. Dispatch trait for typed cuDNN ops.

Required Methods§

Source

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

Source

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

Source

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

Implementors§