cubecl-cpp 0.10.0-pre.3

CPP transpiler for CubeCL
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
51
52
53
54
55
56
57
58
59
60
61
62
63

use std::fmt::Display;

pub enum BufferAttribute {
    Buffer,
    ThreadGroup,
    None,
}

impl BufferAttribute {
    pub fn indexed_fmt(&self, index: usize, f: &mut std::fmt::Formatter) -> std::fmt::Result {
        write!(f, " [[{self}({index})]]")
    }
}

impl Display for BufferAttribute {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            Self::Buffer => f.write_str("buffer"),
            Self::ThreadGroup => f.write_str("threadgroup"),
            Self::None => Ok(()),
        }
    }
}

pub enum BuiltInAttribute {
    SIMDgroupIndexInThreadgroup,
    ThreadIndexInSIMDgroup,
    ThreadIndexInThreadgroup,
    ThreadPositionInGrid,
    ThreadPositionInThreadgroup,
    ThreadgroupPositionInGrid,
    ThreadgroupsPerGrid,
    ThreadsPerSIMDgroup,
    ThreadsPerThreadgroup,
    None,
}

impl Display for BuiltInAttribute {
    fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
        match self {
            BuiltInAttribute::SIMDgroupIndexInThreadgroup => {
                f.write_str("[[simdgroup_index_in_threadgroup]]")
            }
            BuiltInAttribute::ThreadIndexInSIMDgroup => {
                f.write_str("[[thread_index_in_simdgroup]]")
            }
            BuiltInAttribute::ThreadIndexInThreadgroup => {
                f.write_str("[[thread_index_in_threadgroup]]")
            }
            BuiltInAttribute::ThreadPositionInGrid => f.write_str("[[thread_position_in_grid]]"),
            BuiltInAttribute::ThreadPositionInThreadgroup => {
                f.write_str("[[thread_position_in_threadgroup]]")
            }
            BuiltInAttribute::ThreadgroupPositionInGrid => {
                f.write_str("[[threadgroup_position_in_grid]]")
            }
            BuiltInAttribute::ThreadgroupsPerGrid => f.write_str("[[threadgroups_per_grid]]"),
            BuiltInAttribute::ThreadsPerSIMDgroup => f.write_str("[[threads_per_simdgroup]]"),
            BuiltInAttribute::ThreadsPerThreadgroup => f.write_str("[[threads_per_threadgroup]]"),
            BuiltInAttribute::None => Ok(()),
        }
    }
}