Enum cubecl_core::codegen::OutputInfo

pub enum OutputInfo {
    ArrayWrite {
        item: Item,
        local: u16,
        position: Variable,
    },
    InputArrayWrite {
        item: Item,
        input: u16,
        local: u16,
        position: Variable,
    },
    Array {
        item: Item,
    },
}

Information related to an output.

Variants

ArrayWrite

Write the local variable to a new array.

This will create a new binding in the kernel definition.

Fields

item: Item

local: u16

position: Variable

InputArrayWrite

Write the local variable to an existing input binding.

Fields

item: Item

input: u16

local: u16

position: Variable

Array

Simply register the output, but don’t automatically add a write to it.

Useful when a procedure writes to the output using operations.

Fields

item: Item

Implementations

impl OutputInfo

pub fn item(&self) -> Item

The item type of the input.

impl OutputInfo

pub fn elem_size<R: Runtime>(&self) -> usize

Trait Implementations

impl Clone for OutputInfo

fn clone(&self) -> OutputInfo

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for OutputInfo

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

Formats the value using the given formatter.