Struct cubecl_core::frontend::SharedMemory

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

Implementations

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

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

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

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

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

Trait Implementations

impl<T: Clone + CubeType> Clone for SharedMemory<T>

fn clone(&self) -> SharedMemory<T>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<T: CubePrimitive> CubeType for SharedMemory<T>

type ExpandType = ExpandElementTyped<SharedMemory<T>>

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

Wrapper around the init method, necessary to type inference.

impl<E: CubeType, I: Into<UInt>> Index<I> for SharedMemory<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 SharedMemory<E>

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

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

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

type Expand = ExpandElementTyped<SharedMemory<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.

impl<T: Copy + CubeType> Copy for SharedMemory<T>