Enum cubecl_core::frontend::ExpandElement

pub enum ExpandElement {
    Managed(Rc<Variable>),
    Plain(Variable),
}

Reference to a JIT variable

Variants

Managed(Rc<Variable>)

Variable kept in the variable pool.

Plain(Variable)

Variable not kept in the variable pool.

Implementations

impl ExpandElement

pub fn can_mut(&self) -> bool

If the element can be mutated inplace, potentially reusing the register.

Methods from Deref<Target = Variable>

pub fn index(&self) -> Option<u16>

pub fn item(&self) -> Item

Fetch the item of the variable.

Trait Implementations

impl Clone for ExpandElement

fn clone(&self) -> ExpandElement

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for ExpandElement

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

Formats the value using the given formatter.

impl Deref for ExpandElement

type Target = Variable

The resulting type after dereferencing.

fn deref(&self) -> &Self::Target

Dereferences the value.

impl From<BF16> for ExpandElement

fn from(value: BF16) -> Self

Converts to this type from the input type.

impl<T: CubeType> From<ExpandElement> for ExpandElementTyped<T>

fn from(expand: ExpandElement) -> Self

Converts to this type from the input type.

impl From<ExpandElement> for Variable

fn from(value: ExpandElement) -> Self

Converts to this type from the input type.

impl<T: CubeType> From<ExpandElementTyped<T>> for ExpandElement

fn from(value: ExpandElementTyped<T>) -> Self

Converts to this type from the input type.

impl From<F16> for ExpandElement

fn from(value: F16) -> Self

Converts to this type from the input type.

impl From<F32> for ExpandElement

fn from(value: F32) -> Self

Converts to this type from the input type.

impl From<F64> for ExpandElement

fn from(value: F64) -> Self

Converts to this type from the input type.

impl From<I32> for ExpandElement

fn from(value: I32) -> Self

Converts to this type from the input type.

impl From<I64> for ExpandElement

fn from(value: I64) -> Self

Converts to this type from the input type.

impl From<UInt> for ExpandElement

Useful for Comptime

fn from(value: UInt) -> Self

Converts to this type from the input type.

impl From<bool> for ExpandElement

fn from(value: bool) -> Self

Converts to this type from the input type.

impl From<f32> for ExpandElement

fn from(value: f32) -> Self

Converts to this type from the input type.

impl From<i32> for ExpandElement

fn from(value: i32) -> Self

Converts to this type from the input type.

impl From<i64> for ExpandElement

fn from(value: i64) -> Self

Converts to this type from the input type.

impl From<u32> for ExpandElement

fn from(value: u32) -> Self

Converts to this type from the input type.

impl From<usize> for ExpandElement

fn from(value: usize) -> Self

Converts to this type from the input type.

impl Init for ExpandElement

fn init(self, context: &mut CubeContext) -> Self

Initialize a type within a context.

impl Vectorized for &ExpandElement

fn vectorization_factor(&self) -> UInt

fn vectorize(self, _factor: UInt) -> Self

impl Vectorized for ExpandElement

fn vectorization_factor(&self) -> UInt

fn vectorize(self, _factor: UInt) -> Self