vyre-spec 0.4.1

Frozen data contracts for vyre — OpDef, AlgebraicLaw, Category, IntrinsicTable
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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119

//! Surface tests for `DataType` enum.
//!
//! Covers variant distinctness, size calculations, and float-family detection.

use vyre_spec::data_type::DataType;

#[test]
fn all_scalar_variants_are_distinct() {
    assert_ne!(DataType::U8, DataType::U16);
    assert_ne!(DataType::U16, DataType::U32);
    assert_ne!(DataType::U32, DataType::U64);
    assert_ne!(DataType::I8, DataType::I16);
    assert_ne!(DataType::I16, DataType::I32);
    assert_ne!(DataType::I32, DataType::I64);
    assert_ne!(DataType::F16, DataType::F32);
    assert_ne!(DataType::F32, DataType::F64);
    assert_ne!(DataType::Bool, DataType::U32);
}

#[test]
fn u32_size_is_4_bytes() {
    assert_eq!(DataType::U32.size_bytes(), Some(4));
}

#[test]
fn f32_size_is_4_bytes() {
    assert_eq!(DataType::F32.size_bytes(), Some(4));
}

#[test]
fn f64_size_is_8_bytes() {
    assert_eq!(DataType::F64.size_bytes(), Some(8));
}

#[test]
fn f32_is_float_family() {
    assert!(DataType::F32.is_float_family());
}

#[test]
fn f16_is_float_family() {
    assert!(DataType::F16.is_float_family());
}

#[test]
fn bf16_is_float_family() {
    assert!(DataType::BF16.is_float_family());
}

#[test]
fn u32_is_not_float_family() {
    assert!(!DataType::U32.is_float_family());
}

#[test]
fn bool_is_not_float_family() {
    assert!(!DataType::Bool.is_float_family());
}

#[test]
fn u32_element_size_is_none() {
    assert_eq!(DataType::U32.element_size(), None);
}

#[test]
fn vec2_u32_size_is_8() {
    assert_eq!(DataType::Vec2U32.size_bytes(), Some(8));
}

#[test]
fn vec4_u32_size_is_16() {
    assert_eq!(DataType::Vec4U32.size_bytes(), Some(16));
}

#[test]
fn min_bytes_for_u32_is_4() {
    assert_eq!(DataType::U32.min_bytes(), 4);
}

#[test]
fn max_bytes_for_scalar_is_some() {
    assert_eq!(DataType::U32.max_bytes(), Some(4));
}

#[test]
fn i64_size_is_8() {
    assert_eq!(DataType::I64.size_bytes(), Some(8));
}

#[test]
fn f8e4m3_is_float_family() {
    assert!(DataType::F8E4M3.is_float_family());
}

#[test]
fn f8e5m2_is_float_family() {
    assert!(DataType::F8E5M2.is_float_family());
}

#[test]
fn bytes_has_no_element_size() {
    assert_eq!(DataType::Bytes.element_size(), None);
}

#[test]
fn bytes_size_is_1() {
    assert_eq!(DataType::Bytes.size_bytes(), Some(1));
}

#[test]
fn tensor_has_no_size_bytes() {
    assert_eq!(DataType::Tensor.size_bytes(), None);
}

#[test]
fn array_size_roundtrips() {
    let dt = DataType::Array { element_size: 16 };
    assert_eq!(dt.element_size(), Some(16));
}