Trait cubecl_core::codegen::Compiler

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

    // Required methods
    fn compile(
        kernel: KernelDefinition,
        mode: ExecutionMode,
    ) -> 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, mode: ExecutionMode, ) -> 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