cubecl-core 0.12.0-pre.2

CubeCL core create
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26

use crate::frontend::iter::impl_iterable_indexed;
use crate::prelude::*;

/// The expand half of [`Rev`](core::iter::Rev).
#[derive(new)]
pub struct RevExpand<I> {
    iter: I,
}

impl_iterable_indexed!([I: IndexedIteratorExpand] RevExpand<I>, I::Item);

impl<I: IndexedIteratorExpand> IndexedIteratorExpand for RevExpand<I> {
    fn __expand_len_method(&self, scope: &Scope) -> NativeExpand<usize> {
        self.iter.__expand_len_method(scope)
    }

    fn get(&mut self, scope: &Scope, index: NativeExpand<usize>) -> Self::Item {
        // `len - index` is at least one, so this can't underflow.
        let index = self
            .iter
            .__expand_len_method(scope)
            .__expand_sub_method(scope, index)
            .__expand_sub_method(scope, NativeExpand::from_lit(scope, 1));
        self.iter.get(scope, index)
    }
}