tenferro-runtime 0.6.0

Concrete tensor helpers, traced tensors, graph execution, and extension runtime support for tenferro.
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

#[test]
fn signature_tensor_metadata_mapping_source_contract() {
    let source = include_str!("../signature.rs");
    let from_reads = function_body(source, "from_reads");

    for metadata_call in [".strides()", ".is_col_major_contiguous()"] {
        let call_start = from_reads
            .find(metadata_call)
            .unwrap_or_else(|| panic!("signature mapping must call {metadata_call}"));
        let statement = statement_containing(from_reads, call_start);

        assert!(
            statement.contains("map_err") && statement.contains("input"),
            "{metadata_call} must map the original input index in the metadata statement"
        );
    }

    assert!(
        from_reads.contains("PrepareError::InputSignature"),
        "metadata failures must map through PrepareError::InputSignature"
    );
    assert!(
        from_reads.contains("InputSignatureError::TensorMetadata"),
        "metadata failures must map through InputSignatureError::TensorMetadata"
    );
    assert!(
        !from_reads.contains("InputSignatureEntry::new"),
        "from_reads must not remap metadata failures through entry validation"
    );
    assert!(
        !from_reads.contains("ShapeStrideRankMismatch")
            && !from_reads.contains("InvalidAlignmentClass"),
        "from_reads must not remap metadata failures as entry-validation errors"
    );
}

fn function_body<'a>(source: &'a str, name: &str) -> &'a str {
    let signature_start = source
        .find(&format!("fn {name}"))
        .unwrap_or_else(|| panic!("missing {name} function"));
    let body_start = source[signature_start..]
        .find('{')
        .map(|offset| signature_start + offset)
        .unwrap_or_else(|| panic!("missing {name} body"));
    let mut depth = 0usize;
    for (offset, byte) in source[body_start..].bytes().enumerate() {
        match byte {
            b'{' => depth += 1,
            b'}' => {
                depth -= 1;
                if depth == 0 {
                    return &source[body_start..=body_start + offset];
                }
            }
            _ => {}
        }
    }
    panic!("unterminated {name} body");
}

fn statement_containing(source: &str, index: usize) -> &str {
    let start = source[..index]
        .rfind([';', '{'])
        .map_or(0, |position| position + 1);
    let end = source[index..]
        .find(';')
        .map_or(source.len(), |position| index + position + 1);
    &source[start..end]
}