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

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

use super::BuiltInAttribute;

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