cubecl-ir 0.11.0-pre.3

Intermediate representation 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

use alloc::vec::Vec;

use cubecl_macros_internal::TypeHash;
use pliron::derive::{format, pliron_type};

use crate::aligned;

#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[derive(Debug, Clone, Copy, TypeHash, PartialEq, Eq, Hash, PartialOrd, Ord)]
#[format]
pub enum ClampMode {
    Undefined,
    #[format("` ` $0")]
    Constant(u32),
    ClampToEdge,
    Repeat,
    RepeatMirrored,
}

#[allow(missing_docs)]
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
#[pliron_type(
    name = "spirv.tensor_layout",
    format = "`<` $rank `d, clamp: ` $clamp_mode `>`",
    generate_get = true,
    verifier = "succ"
)]
pub struct TensorLayoutType {
    pub rank: usize,
    pub clamp_mode: ClampMode,
}
aligned!(TensorLayoutType, align_of::<u64>()); //Dummy align, ignored in SPIR-V

#[allow(missing_docs)]
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
#[pliron_type(
    name = "spirv.tensor_view",
    format = "`<` $rank `d, has_dims: ` $has_dims `[` vec($permutation, Char(`,`)) `]`",
    generate_get = true,
    verifier = "succ"
)]
pub struct TensorViewType {
    pub rank: usize,
    pub has_dims: bool,
    pub permutation: Vec<usize>,
}
aligned!(TensorViewType, align_of::<u64>()); //Dummy align, ignored in SPIR-V