Trait cubecl_core::codegen::Compiler

pub trait Compiler: Sync + Send + 'static + Clone + Default + Debug {
    type Representation: CompilerRepresentation;

    // Required methods
    fn compile(kernel: KernelDefinition) -> Self::Representation;
    fn elem_size(elem: Elem) -> usize;
    fn max_shared_memory_size() -> usize;
}

Compiles the representation into its own representation that can be formatted into tokens.

Required Associated Types

type Representation: CompilerRepresentation

The representation for the compiled code.

Required Methods

fn compile(kernel: KernelDefinition) -> Self::Representation

Compiles the kernel definition into the compiler’s representation.

fn elem_size(elem: Elem) -> usize

The size of the given element in bytes.

fn max_shared_memory_size() -> usize

The maximal size of a shared memory

Object Safety

This trait is not object safe.

Implementors