cubecl-core 0.11.0-pre.1

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

use super::CubeType;
use crate as cubecl;
use crate::{prelude::*, unexpanded};
use cubecl_ir::{Scope, VectorSize};

/// Type from which we can read/to which we can write values in cube functions.
#[allow(clippy::len_without_is_empty)]
#[cube(expand_base_traits = "SliceOperatorExpand<T>
    + IndexExpand<NativeExpand<usize>, Output = NativeExpand<T>>
    + IndexMutExpand<NativeExpand<usize>, Output = NativeExpand<T>>")]
pub trait List<T: CubePrimitive>:
    SliceOperator<T> + CubeIndex<usize, Output = T> + CubeIndexMut<usize, Output = T> + Vectorized
{
    fn len(&self) -> usize {
        unexpanded!();
    }
}

pub trait Vectorized: CubeType<ExpandType: VectorizedExpand> {
    fn vector_size(&self) -> VectorSize {
        unexpanded!()
    }
    fn __expand_vector_size(_scope: &Scope, this: &Self::ExpandType) -> VectorSize {
        this.vector_size()
    }
}

impl<T: Vectorized + ?Sized> Vectorized for &T {}
impl<T: Vectorized + ?Sized> Vectorized for &mut T {}

pub trait VectorizedExpand {
    fn vector_size(&self) -> VectorSize;
    fn __expand_vector_size_method(&self, _scope: &Scope) -> VectorSize {
        self.vector_size()
    }
}

impl<T: VectorizedExpand + ?Sized> VectorizedExpand for &T {
    fn vector_size(&self) -> VectorSize {
        (**self).vector_size()
    }
}
impl<T: VectorizedExpand + ?Sized> VectorizedExpand for &mut T {
    fn vector_size(&self) -> VectorSize {
        (**self).vector_size()
    }
}