tenferro-ad 0.3.0

Eager runtime, eager tensors, and traced AD extension traits for tenferro.
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

use tenferro_cpu::CpuBackend;
use tenferro_runtime::{DType, GraphCompiler, Tensor, TracedTensor, TypedTensor};

use crate::support::{run_compiled_one, runtime_from_cpu_backend};

#[test]
fn graph_execution_with_borrowed_inputs_preserves_caller_tensors() {
    let lhs_value = TracedTensor::input_symbolic_shape(DType::F64, 0).unwrap();
    let rhs_value = TracedTensor::input_symbolic_shape(DType::F64, 0).unwrap();
    let sum = (&lhs_value + &rhs_value).unwrap();
    let mut compiler = GraphCompiler::new();
    let program = compiler
        .compile_with_input_specs(
            &sum,
            &[(&lhs_value, DType::F64, &[]), (&rhs_value, DType::F64, &[])],
        )
        .unwrap();
    let lhs = Tensor::F64(TypedTensor::from_vec_col_major(vec![], vec![2.0]).unwrap());
    let rhs = Tensor::F64(TypedTensor::from_vec_col_major(vec![], vec![3.0]).unwrap());

    let runtime = runtime_from_cpu_backend(&CpuBackend::with_threads(1).unwrap());
    let output = run_compiled_one(&runtime, &program, &[&lhs, &rhs])
        .expect("borrowed graph execution should succeed");

    assert_eq!(output.as_slice::<f64>().unwrap(), &[5.0]);
    assert_eq!(lhs.as_slice::<f64>().unwrap(), &[2.0]);
    assert_eq!(rhs.as_slice::<f64>().unwrap(), &[3.0]);
}