Function for_expand

pub fn for_expand<I: Numeric>(
    context: &mut CubeContext,
    range: impl Iterable<I>,
    unroll: bool,
    body: impl FnMut(&mut CubeContext, ExpandElementTyped<I>),
)