Enum 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.

pub fn consume(self) -> Variable

Explicitly consume the element, freeing it for reuse if no other copies exist.

Methods from Deref<Target = Variable>

pub fn is_immutable(&self) -> bool

Whether a variable is always immutable. Used for optimizations to determine whether it’s safe to inline/merge

pub fn is_array(&self) -> bool

Is this an array type that yields Items when indexed, or a scalar/vector that yields Elems when indexed?

pub fn has_length(&self) -> bool

pub fn has_buffer_length(&self) -> bool

pub fn is_constant(&self, value: i64) -> bool

Determines if the value is a constant with the specified value (converted if necessary)

pub fn is_true(&self) -> bool

Determines if the value is a boolean constant with the true value

pub fn is_false(&self) -> bool

Determines if the value is a boolean constant with the false value

pub fn vectorization_factor(&self) -> u8

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

pub fn as_const(&self) -> Option<ConstantScalarValue>

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<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<bf16> for ExpandElement

fn from(value: bf16) -> 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<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<flex32> for ExpandElement

fn from(value: flex32) -> Self

Converts to this type from the input type.

impl From<i16> for ExpandElement

fn from(value: i16) -> 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<i8> for ExpandElement

fn from(value: i8) -> Self

Converts to this type from the input type.

impl From<tf32> for ExpandElement

fn from(value: tf32) -> Self

Converts to this type from the input type.

impl From<u16> for ExpandElement

fn from(value: u16) -> 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<u64> for ExpandElement

fn from(value: u64) -> Self

Converts to this type from the input type.

impl From<u8> for ExpandElement

fn from(value: u8) -> 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 Index for ExpandElement

fn value(self) -> Variable

impl Init for ExpandElement

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

Initialize a type within a context.