Struct cubecl_core::frontend::Sequence

pub struct Sequence<T: CubeType> { /* private fields */ }

A sequence of cube types that is inlined during compilation.

In other words, it allows you to group a dynamic amount of variables at compile time.

All methods push, index and into_iter are executed during compilation and don’t add any overhead on the generated kernel.

Implementations

impl<T: CubeType> Sequence<T>

pub fn new() -> Self

Create a new empty sequence.

pub fn push(&mut self, value: T)

Push a new value into the sequence.

pub fn index<I: Index>(&self, index: I) -> &T

Get the variable at the given position in the sequence.

pub fn __expand_new(_context: &mut CubeContext) -> SequenceExpand<T>

Expand function of new.

pub fn __expand_push( context: &mut CubeContext, expand: &mut SequenceExpand<T>, value: T::ExpandType, )

Expand function of push.

pub fn __expand_index<I: Index>( context: &mut CubeContext, expand: SequenceExpand<T>, index: I, ) -> T::ExpandType

Expand function of index.

Trait Implementations

impl<T: CubeType> CubeType for Sequence<T>

type ExpandType = SequenceExpand<T>

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

Wrapper around the init method, necessary to type inference.

impl<T: CubeType> Default for Sequence<T>

fn default() -> Self

Returns the “default value” for a type.

impl<T: CubeType> IntoIterator for Sequence<T>

type Item = T

The type of the elements being iterated over.

type IntoIter = <Vec<T> as IntoIterator>::IntoIter

Which kind of iterator are we turning this into?

fn into_iter(self) -> Self::IntoIter

Creates an iterator from a value.