Struct SharedMemory

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

Implementations

impl<E: CubePrimitive> SharedMemory<E>

pub unsafe fn index_unchecked<I: Index>(&self, _i: I) -> &E

where Self: CubeIndex<I>,

Perform an unchecked index into the array

Safety

Out of bounds indexing causes undefined behaviour and may segfault. Ensure index is always in bounds

pub unsafe fn index_assign_unchecked<I: Index>(&mut self, _i: I, _value: E)

where Self: CubeIndexMut<I>,

Perform an unchecked index assignment into the array

Safety

Out of bounds indexing causes undefined behaviour and may segfault. Ensure index is always in bounds

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

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

pub fn new_lined<S: Index>( _size: S, _vectorization_factor: u32, ) -> SharedMemory<Line<T>>

pub fn __expand_new_lined( context: &mut CubeContext, size: ExpandElementTyped<u32>, vectorization_factor: u32, ) -> <SharedMemory<Line<T>> as CubeType>::ExpandType

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

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

pub fn __expand_new( context: &mut CubeContext, size: ExpandElementTyped<u32>, ) -> <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<E: CubeType, I: Index> CubeIndex<I> for SharedMemory<E>

type Output = E

fn cube_idx(&self, _i: T) -> &Self::Output

impl<E: CubeType, I: Index> CubeIndexMut<I> for SharedMemory<E>

fn cube_idx_mut(&mut self, _i: T) -> &mut Self::Output

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<T: CubePrimitive> IntoRuntime for SharedMemory<T>

fn __expand_runtime_method( self, _context: &mut CubeContext, ) -> ExpandElementTyped<Self>

fn runtime(self) -> Self

Make sure a type is actually expanded into its runtime expand type.

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( context: &mut CubeContext, expand: Self::Expand, start: ExpandElementTyped<u32>, end: ExpandElementTyped<u32>, ) -> 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( context: &mut CubeContext, expand: Self::Expand, start: ExpandElementTyped<u32>, end: ExpandElementTyped<u32>, ) -> 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( context: &mut CubeContext, expand: Self::Expand, start: ExpandElementTyped<u32>, end: ExpandElementTyped<u32>, ) -> 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>