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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50

use crate::prelude::*;

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

impl<'a, I, T> Iterable for CopiedExpand<I>
where
    I: Iterable<Item = &'a T>,
    T: DerefExpand<Target = T> + 'a,
{
    type Item = T;

    fn expand(self, scope: &Scope, body: &mut dyn FnMut(&Scope, T)) {
        self.iter.expand(scope, &mut |scope, item| {
            let value = item.__expand_deref_method(scope);
            body(scope, value);
        });
    }

    fn expand_unroll(self, scope: &Scope, body: &mut dyn FnMut(&Scope, T)) {
        self.iter.expand_unroll(scope, &mut |scope, item| {
            let value = item.__expand_deref_method(scope);
            body(scope, value);
        });
    }
}

impl<'a, I, T> IteratorExpand for CopiedExpand<I>
where
    I: IteratorExpand<Item = &'a T>,
    T: DerefExpand<Target = T> + 'a,
{
}

impl<'a, I, T> IndexedIteratorExpand for CopiedExpand<I>
where
    I: IndexedIteratorExpand<Item = &'a T>,
    T: DerefExpand<Target = T> + 'a,
{
    fn __expand_len_method(&self, scope: &Scope) -> NativeExpand<usize> {
        self.iter.__expand_len_method(scope)
    }

    fn get(&mut self, scope: &Scope, index: NativeExpand<usize>) -> T {
        self.iter.get(scope, index).__expand_deref_method(scope)
    }
}