cubecl_core/codegen/
compiler.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
use crate::ir::{Elem, KernelDefinition, LocalAllocator};
use cubecl_runtime::ExecutionMode;
use std::fmt::Display;

/// Trait for compiled code representation
pub trait CompilerRepresentation: Display {
    /// Computes and returns the shared memory size
    fn shared_memory_size(&self) -> usize;
}

/// Compiles the representation into its own representation that can be formatted into tokens.
pub trait Compiler: Sync + Send + 'static + Clone + Default + core::fmt::Debug {
    /// The representation for the compiled code.
    type Representation: CompilerRepresentation;

    /// Compiles the [kernel definition](KernelDefinition) into the compiler's representation.
    fn compile(kernel: KernelDefinition, mode: ExecutionMode) -> Self::Representation;
    /// The size of the given element in bytes.
    fn elem_size(elem: Elem) -> usize;
    fn local_allocator() -> impl LocalAllocator;
    /// The maximal size of a shared memory, in bytes
    fn max_shared_memory_size() -> usize;
}