tetration 0.1.8

Chunked tensor .tet format, tet CLI, and optional C ABI (tetration-ffi)
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

//! Disjoint parallel-fold preview buffers (raw pointer slices).

/// Typed preview slice backed by a disjoint parallel-fold buffer (or empty when capped).
pub(crate) fn disjoint_preview_mut<'a, T>(preview_addr: usize, preview_len: usize) -> &'a mut [T] {
    if preview_len == 0 {
        &mut []
    } else {
        // SAFETY: planned chunks write disjoint logical indices.
        unsafe { std::slice::from_raw_parts_mut(preview_addr as *mut T, preview_len) }
    }
}

fn write_disjoint_preview<T: Copy>(preview_addr: usize, preview_len: usize, li: usize, v: T) {
    if li < preview_len {
        disjoint_preview_mut(preview_addr, preview_len)[li] = v;
    }
}

pub(crate) fn write_disjoint_preview_f32(
    preview_addr: usize,
    preview_len: usize,
    li: usize,
    v: f32,
) {
    write_disjoint_preview(preview_addr, preview_len, li, v);
}

pub(crate) fn write_disjoint_preview_f64(
    preview_addr: usize,
    preview_len: usize,
    li: usize,
    v: f64,
) {
    write_disjoint_preview(preview_addr, preview_len, li, v);
}

pub(crate) fn write_disjoint_preview_i32(
    preview_addr: usize,
    preview_len: usize,
    li: usize,
    v: f64,
) {
    write_disjoint_preview(preview_addr, preview_len, li, v as i32);
}

pub(crate) fn write_disjoint_preview_i64(
    preview_addr: usize,
    preview_len: usize,
    li: usize,
    v: f64,
) {
    write_disjoint_preview(preview_addr, preview_len, li, v as i64);
}

pub(crate) fn write_disjoint_preview_u8(
    preview_addr: usize,
    preview_len: usize,
    li: usize,
    v: f64,
) {
    write_disjoint_preview(preview_addr, preview_len, li, v.clamp(0., 255.) as u8);
}

pub(crate) fn write_disjoint_preview_u16(
    preview_addr: usize,
    preview_len: usize,
    li: usize,
    v: f64,
) {
    write_disjoint_preview(preview_addr, preview_len, li, v.clamp(0., 65535.) as u16);
}

pub(crate) fn write_disjoint_preview_i16(
    preview_addr: usize,
    preview_len: usize,
    li: usize,
    v: f64,
) {
    write_disjoint_preview(
        preview_addr,
        preview_len,
        li,
        v.clamp(i16::MIN as f64, i16::MAX as f64) as i16,
    );
}

pub(crate) fn write_disjoint_preview_u32(
    preview_addr: usize,
    preview_len: usize,
    li: usize,
    v: f64,
) {
    write_disjoint_preview(
        preview_addr,
        preview_len,
        li,
        v.clamp(0., u32::MAX as f64) as u32,
    );
}

pub(crate) fn write_disjoint_preview_u64(
    preview_addr: usize,
    preview_len: usize,
    li: usize,
    v: f64,
) {
    write_disjoint_preview(preview_addr, preview_len, li, v.max(0.) as u64);
}

pub(crate) fn write_disjoint_preview_f16(
    preview_addr: usize,
    preview_len: usize,
    li: usize,
    v: f64,
) {
    write_disjoint_preview(preview_addr, preview_len, li, half::f16::from_f64(v));
}