Struct StridedSimd

pub struct StridedSimd<T: TypeCommon> { /* private fields */ }

A single thread SIMD-optimized strided iterator

Implementations

impl<T: CommonBounds> StridedSimd<T>

pub fn shape(&self) -> &Shape

Retrieves the shape of the tensor.

Returns

A reference to the Shape struct representing the tensor’s dimensions.

pub fn strides(&self) -> &Strides

Retrieves the strides of the tensor.

Returns

A reference to the Strides struct representing the tensor’s stride information.

pub fn new<U: TensorInfo<T>>(tensor: U) -> Self

Creates a new StridedSimd instance from a given tensor.

Arguments

• tensor - The tensor implementing the TensorInfo<T> trait to iterate over mutably.

Returns

A new instance of StridedSimd initialized with the provided tensor.

Trait Implementations

impl<T: Clone + TypeCommon> Clone for StridedSimd<T>

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

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<T: CommonBounds> IterGetSetSimd for StridedSimd<T>

type Item = T

The type of the iterator’s elements.

type SimdItem = <T as TypeCommon>::Vec

The type of the iterator’s simd elements.

fn set_end_index(&mut self, _: usize)

set the end index of the iterator, this is used when rayon perform data splitting

fn set_intervals(&mut self, _: Arc<Vec<(usize, usize)>>)

set the chunk intervals of the iterator, we chunk the outer loop

fn set_strides(&mut self, strides: Strides)

set the strides for the iterator, we call this method normally when we do broadcasting

fn set_shape(&mut self, shape: Shape)

set the shape for the iterator, we call this method normally when we do broadcasting

fn set_prg(&mut self, prg: Vec<i64>)

set the loop progress for the iterator

fn intervals(&self) -> &Arc<Vec<(usize, usize)>>

get the intervals of the iterator

fn strides(&self) -> &Strides

get the strides of the iterator

fn shape(&self) -> &Shape

get the shape of the iterator

fn layout(&self) -> &Layout

get the layout of the iterator

fn broadcast_set_strides(&mut self, shape: &Shape)

set the strides for all the iterators

fn outer_loop_size(&self) -> usize

get the outer loop size

fn inner_loop_size(&self) -> usize

get the inner loop size

fn next(&mut self)

update the loop progress, this is called when we don’t do simd iteration

fn next_simd(&mut self)

update the loop progress, this is called when we do simd iteration

fn inner_loop_next(&mut self, index: usize) -> Self::Item

get the next element of the inner loop

fn inner_loop_next_simd(&mut self, index: usize) -> Self::SimdItem

get the next vector of the inner loop, this is called when we do simd iteration

fn all_last_stride_one(&self) -> bool

check if all iterators’ last stride is one, only when all iterators’ last stride is one, we can do simd iteration

fn lanes(&self) -> Option<usize>

get the simd vector size, if any of the iterator returned different vector size, it will return None

impl<T: CommonBounds> ShapeManipulator for StridedSimd<T>

fn reshape<S: Into<Shape>>(self, shape: S) -> Self

reshape the iterator, we can change the iteration behavior by changing the shape

fn transpose<AXIS: Into<Axis>>(self, axes: AXIS) -> Self

transpose the iterator, we can change the iteration behavior by changing the axes

fn expand<S: Into<Shape>>(self, shape: S) -> Self

expand the iterator, we can change the iteration behavior by changing the shape

impl<T> StridedIteratorSimd for StridedSimd<T>

where T: CommonBounds,

fn for_each<F, F2>(self, op: F, vec_op: F2)

where F: Fn(Self::Item), F2: Fn(Self::SimdItem),

perform simd iteration, this method is for single thread simd iterator

fn for_each_init<F, INIT, T>(self, init: INIT, func: F)

where F: Fn(&mut T, Self::Item), INIT: Fn() -> T,

perform simd iteration with init, this method is for single thread simd iterator

impl<T: CommonBounds> StridedSimdIteratorZip for StridedSimd<T>

fn zip<'a, C>(self, other: C) -> StridedZipSimd<'a, Self, C>

where C: 'a + IterGetSetSimd, <C as IterGetSetSimd>::Item: Send, Self: IterGetSetSimd, <Self as IterGetSetSimd>::Item: Send,

Combines this iterator with another SIMD-optimized iterator, enabling simultaneous iteration.

impl<T: CommonBounds> StridedIteratorMap for StridedSimd<T>