tenferro-einsum 0.2.0

Subscripts, contraction planning, concrete/traced/eager einsum APIs, extension runtime, and AD rule 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

use super::{dot_general_config, validate_traced_contract_dims, TensorDotAxes};
use tenferro_runtime::TracedTensor;

#[test]
fn dot_general_config_rejects_invalid_count_and_explicit_axes() {
    let count_err = dot_general_config(TensorDotAxes::Count(3), 2, 2).unwrap_err();
    assert!(count_err.to_string().contains("Count(3)"));

    let length_err = dot_general_config(
        TensorDotAxes::Axes {
            lhs: &[0, 1],
            rhs: &[0],
        },
        2,
        2,
    )
    .unwrap_err();
    assert!(length_err.to_string().contains("matching lengths"));

    let duplicate_err = dot_general_config(
        TensorDotAxes::Axes {
            lhs: &[0, -2],
            rhs: &[0, 1],
        },
        2,
        2,
    )
    .unwrap_err();
    assert!(duplicate_err.to_string().contains("duplicate lhs axis"));
}

#[test]
fn validate_traced_contract_dims_allows_symbolic_and_rejects_concrete_mismatch() {
    let lhs = TracedTensor::input_symbolic_shape(tenferro_runtime::DType::F64, 2).unwrap();
    let rhs = TracedTensor::input_symbolic_shape(tenferro_runtime::DType::F64, 2).unwrap();
    let config = dot_general_config(
        TensorDotAxes::Axes {
            lhs: &[1],
            rhs: &[0],
        },
        lhs.rank,
        rhs.rank,
    )
    .unwrap();
    validate_traced_contract_dims(&lhs, &rhs, &config).unwrap();

    let lhs = TracedTensor::from_vec_col_major(vec![2, 3], vec![1.0_f64; 6]).unwrap();
    let rhs = TracedTensor::from_vec_col_major(vec![4, 2], vec![1.0_f64; 8]).unwrap();
    let err = validate_traced_contract_dims(&lhs, &rhs, &config).unwrap_err();
    assert!(err.to_string().contains("contracted dimensions differ"));
}