Struct cubecl_core::frontend::Array

pub struct Array<E> { /* private fields */ }

A contiguous array of elements.

Implementations

impl<T: CubePrimitive + Clone> Array<T>

pub fn new<S: Index>(_size: S) -> Self

pub fn vectorized<S: Index>(_size: S, _vectorization_factor: UInt) -> Self

pub fn __expand_new<S: Index>( context: &mut CubeContext, size: S, ) -> <Self as CubeType>::ExpandType

pub fn __expand_vectorized<S: Index>( context: &mut CubeContext, size: S, vectorization_factor: UInt, ) -> <Self as CubeType>::ExpandType

pub fn to_vectorized(self, _vectorization_factor: Comptime<UInt>) -> T

impl<E: CubeType> Array<E>

pub fn len(&self) -> UInt

Obtain the array length

Trait Implementations

impl<C: CubeType> CubeType for &Array<C>

type ExpandType = ExpandElementTyped<Array<C>>

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

Wrapper around the init method, necessary to type inference.

impl<C: CubeType> CubeType for Array<C>

type ExpandType = ExpandElementTyped<Array<C>>

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

Wrapper around the init method, necessary to type inference.

impl<C: CubeType> ExpandElementBaseInit for Array<C>

fn init_elem(_context: &mut CubeContext, elem: ExpandElement) -> ExpandElement

impl<E: CubeType, I: Into<UInt>> Index<I> for Array<E>

type Output = E

The returned type after indexing.

fn index(&self, _index: I) -> &Self::Output

Performs the indexing (container[index]) operation.

impl<E: CubeType, I: Into<UInt>> IndexMut<I> for Array<E>

fn index_mut(&mut self, _index: I) -> &mut Self::Output

Performs the mutable indexing (container[index]) operation.

impl<C: CubePrimitive> LaunchArg for Array<C>

type RuntimeArg<'a, R: Runtime> = ArrayArg<'a, R>

The runtime argument for the kernel.

impl<C: CubePrimitive> LaunchArgExpand for Array<C>

fn expand( builder: &mut KernelBuilder, vectorization: Vectorization, ) -> ExpandElementTyped<Array<C>>

Register an input variable during compilation that fill the KernelBuilder.

fn expand_output( builder: &mut KernelBuilder, vectorization: Vectorization, ) -> ExpandElementTyped<Array<C>>

Register an output variable during compilation that fill the KernelBuilder.

impl<E: CubePrimitive> SliceOperator<E> for Array<E>

type Expand = ExpandElementTyped<Array<E>>

fn slice<Start: Index, End: Index>( &self, start: Start, end: End, ) -> &Slice<'_, E>

Return a read-only view of all elements comprise between the start and end index.

fn __expand_slice<Start: Index, End: Index>( context: &mut CubeContext, expand: Self::Expand, start: Start, end: End, ) -> ExpandElementTyped<Slice<'static, E>>

Expand function of SliceOperator::slice.

fn slice_mut<Start: Index, End: Index>( &mut self, start: Start, end: End, ) -> &mut SliceMut<'_, E>

Return a read-write view of all elements comprise between the start and end index.

fn __expand_slice_mut<Start: Index, End: Index>( context: &mut CubeContext, expand: Self::Expand, start: Start, end: End, ) -> ExpandElementTyped<SliceMut<'static, E>>

Expand function of SliceOperator::slice_mut.

fn slice_mut_unsafe<Start: Index, End: Index>( &self, start: Start, end: End, ) -> &mut SliceMut<'_, E>

Return a read-write view of all elements comprise between the start and end index.

fn __expand_slice_mut_unsafe<Start: Index, End: Index>( context: &mut CubeContext, expand: Self::Expand, start: Start, end: End, ) -> ExpandElementTyped<SliceMut<'static, E>>

Expand function of SliceOperator::slice_mut_unsafe.

fn as_slice(&self) -> &Slice<'_, E>

Reinterprete the current type as a read-only slice.

fn __expand_as_slice( context: &mut CubeContext, expand: Self::Expand, ) -> ExpandElementTyped<Slice<'static, E>>

Expand function of SliceOperator::as_slice.

fn as_slice_mut(&mut self) -> &mut SliceMut<'_, E>

Reinterprete the current type as a read-write slice.

fn __expand_as_slice_mut( context: &mut CubeContext, expand: Self::Expand, ) -> ExpandElementTyped<SliceMut<'static, E>>

Expand function of SliceOperator::as_slice_mut.

fn as_slice_mut_unsafe(&self) -> &mut SliceMut<'_, E>

Reinterprete the current type as a read-write slice.

fn __expand_as_slice_mut_unsafe( context: &mut CubeContext, expand: Self::Expand, ) -> ExpandElementTyped<SliceMut<'static, E>>

Expand function of SliceOperator::as_slice_mut_unsafe.