Skip to main content

KernelBackend

kapsl_hal::kernel

Trait KernelBackend 

Source 
pub trait KernelBackend:
    Send
    + Sync
    + Debug {
    // Required methods
    fn backend_type(&self) -> KernelBackendType;
    fn attention(&self) -> Box<dyn AttentionKernel>;
    fn mlp(&self) -> Box<dyn MlpKernel>;
}

Required Methods§

Source

fn backend_type(&self) -> KernelBackendType

Source

fn attention(&self) -> Box<dyn AttentionKernel>

Source

fn mlp(&self) -> Box<dyn MlpKernel>

Implementors§