Trait cubecl_core::compute::CubeTask

pub trait CubeTask: Send + Sync {
    // Required methods
    fn id(&self) -> KernelId;
    fn compile(&self, mode: ExecutionMode) -> CompiledKernel;
}

Kernel trait with the ComputeShader that will be compiled and cached based on the provided id.

Required Methods

fn id(&self) -> KernelId

Identifier for the kernel, used for caching kernel compilation.

fn compile(&self, mode: ExecutionMode) -> CompiledKernel

Compile the kernel into source

Trait Implementations

impl CubeTask for Box<dyn CubeTask>

fn compile(&self, mode: ExecutionMode) -> CompiledKernel

Compile the kernel into source

fn id(&self) -> KernelId

Identifier for the kernel, used for caching kernel compilation.

Implementations on Foreign Types

impl CubeTask for Box<dyn CubeTask>

fn compile(&self, mode: ExecutionMode) -> CompiledKernel

fn id(&self) -> KernelId

impl CubeTask for Arc<dyn CubeTask>

fn compile(&self, mode: ExecutionMode) -> CompiledKernel

fn id(&self) -> KernelId

Implementors

impl<C: Compiler, K: Kernel> CubeTask for KernelTask<C, K>