Struct RangeExpand

pub struct RangeExpand<I: Int> {
    pub start: ExpandElementTyped<I>,
    pub end: ExpandElementTyped<I>,
    pub inclusive: bool,
}

Fields

start: ExpandElementTyped<I>

end: ExpandElementTyped<I>

inclusive: bool

Implementations

impl<I: Int> RangeExpand<I>

pub fn new( start: ExpandElementTyped<I>, end: ExpandElementTyped<I>, inclusive: bool, ) -> Self

pub fn __expand_step_by_method( self, n: impl Into<ExpandElementTyped<u32>>, ) -> SteppedRangeExpand<I>

Trait Implementations

impl<I: Int> Iterable<I> for RangeExpand<I>

fn expand_unroll( self, scope: &mut Scope, body: impl FnMut(&mut Scope, <I as CubeType>::ExpandType), )

Expand an unrolled loop. The body should be invoced n times, where n is the number of iterations.

fn expand( self, scope: &mut Scope, body: impl FnMut(&mut Scope, <I as CubeType>::ExpandType), )

Expand a runtime loop without unrolling