cubecl-cpp 0.11.0-pre.1

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

use crate::{Dialect, shared::Builtin};

use super::BuiltInAttribute;

impl<D: Dialect> Builtin<D> {
    pub fn attribute(&self) -> BuiltInAttribute {
        match self {
            Builtin::AbsolutePosBaseName => BuiltInAttribute::ThreadPositionInGrid,
            Builtin::CubeCountBaseName => BuiltInAttribute::ThreadgroupsPerGrid,
            Builtin::CubeDimBaseName => BuiltInAttribute::ThreadsPerThreadgroup,
            Builtin::CubePosBaseName => BuiltInAttribute::ThreadgroupPositionInGrid,
            Builtin::PlaneDim => BuiltInAttribute::ThreadsPerSIMDgroup,
            Builtin::PlanePos => BuiltInAttribute::SIMDgroupIndexInThreadgroup,
            Builtin::UnitPosBaseName => BuiltInAttribute::ThreadPositionInThreadgroup,
            Builtin::UnitPos => BuiltInAttribute::ThreadIndexInThreadgroup,
            Builtin::UnitPosPlane => BuiltInAttribute::ThreadIndexInSIMDgroup,
            _ => BuiltInAttribute::None,
        }
    }
}