Struct cubecl_core::codegen::KernelSettings

pub struct KernelSettings {
    pub mappings: Vec<InplaceMapping>,
    pub reading_strategy: Vec<(u16, ReadingStrategy)>,
    /* private fields */
}

Fields

mappings: Vec<InplaceMapping>

reading_strategy: Vec<(u16, ReadingStrategy)>

Implementations

impl KernelSettings

pub fn vectorize_global(self, vectorization: Vectorization) -> Self

Compile the shader with vectorization enabled for all inputs and outputs.

pub fn vectorize_input( self, position: usize, vectorization: Vectorization, ) -> Self

Compile the shader with vectorization enabled for an input.

pub fn vectorize_output( self, position: usize, vectorization: Vectorization, ) -> Self

Compile the shader with vectorization enabled for an output.

pub fn vectorization_input(&self, position: usize) -> Vectorization

Fetch the vectorization for the provided input position.

pub fn vectorization_output(&self, position: usize) -> Vectorization

Fetch the vectorization for the provided output position.

pub fn inplace(self, mappings: Vec<InplaceMapping>) -> Self

Compile the shader with inplace enabled by the given mapping.

Notes:

You should favor using dynamic_settings when using fusion, since the mapping is going to be created from the runtime information.

pub fn cube_dim(self, cube_dim: CubeDim) -> Self

Set cube dimension.

Trait Implementations

impl Clone for KernelSettings

fn clone(&self) -> KernelSettings

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Default for KernelSettings

fn default() -> KernelSettings

Returns the “default value” for a type.

impl Display for KernelSettings

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.