tenferro-runtime 0.9.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

use computegraph::types::ValueKey;

use super::*;
use crate::traced::next_input_key;

fn scope() -> Arc<GlobalMetadataScope> {
    Arc::new(
        register_scoped_value_metadata(
            ValueKey::Input(next_input_key()),
            TensorMeta::exact(DType::F64, vec![]),
        )
        .unwrap(),
    )
}

#[test]
fn metadata_scope_merge_reuses_single_parent_chain() {
    let parent = MetadataScopeChain::with_scope(scope(), []);
    let merged = MetadataScopeChain::merge([&parent]);

    assert!(merged.shares_root(&parent));
}

#[test]
fn metadata_scope_materialization_deduplicates_shared_nodes_and_scopes() {
    let shared = MetadataScopeChain::with_scope(scope(), []);
    let repeated_scope = scope();
    let left = MetadataScopeChain::with_scope(Arc::clone(&repeated_scope), [&shared]);
    let right = MetadataScopeChain::with_scope(repeated_scope, [&shared]);
    let merged = MetadataScopeChain::merge([&left, &right]);

    let (scopes, visited_nodes) = merged.materialize_with_visit_count();
    assert_eq!(scopes.len(), 2);
    assert_eq!(visited_nodes, 4);
}

#[test]
fn metadata_scope_shared_history_materializes_in_linear_node_visits() {
    let depth = 512;
    let mut chain = MetadataScopeChain::with_scope(scope(), []);
    for _ in 0..depth {
        chain = MetadataScopeChain::merge([&chain, &chain]);
    }

    let (scopes, visited_nodes) = chain.materialize_with_visit_count();
    assert_eq!(scopes.len(), 1);
    assert_eq!(visited_nodes, depth + 1);
}