svod-tensor 0.1.0-alpha.3

High-level lazy tensor API for the Svod ML compiler
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

use crate::*;

/// Regression test for ScatterND: rows not targeted by scatter indices
/// must preserve original data. Reproduces the ONNX test_scatternd inputs.
///
/// Root cause was a rewrite engine bug where `handle_link` resolved stale
/// results before pattern chains completed (PtrCat not eliminated).
#[test]
fn test_scatternd_debug() {
    let cfg = PrepareConfig::from(svod_schedule::OptimizerConfig::default());

    let data: Vec<f32> = vec![
        1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 8.0, 7.0, 6.0, 5.0, 4.0, 3.0, 2.0, 1.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0,
        7.0, 8.0, 8.0, 7.0, 6.0, 5.0, 4.0, 3.0, 2.0, 1.0, 8.0, 7.0, 6.0, 5.0, 4.0, 3.0, 2.0, 1.0, 1.0, 2.0, 3.0, 4.0,
        5.0, 6.0, 7.0, 8.0, 8.0, 7.0, 6.0, 5.0, 4.0, 3.0, 2.0, 1.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0,
    ];
    let indices: Vec<i64> = vec![0, 2];
    let updates: Vec<f32> = vec![
        5.0, 5.0, 5.0, 5.0, 6.0, 6.0, 6.0, 6.0, 7.0, 7.0, 7.0, 7.0, 8.0, 8.0, 8.0, 8.0, 1.0, 1.0, 1.0, 1.0, 2.0, 2.0,
        2.0, 2.0, 3.0, 3.0, 3.0, 3.0, 4.0, 4.0, 4.0, 4.0,
    ];
    let expected: Vec<f32> = vec![
        5.0, 5.0, 5.0, 5.0, 6.0, 6.0, 6.0, 6.0, 7.0, 7.0, 7.0, 7.0, 8.0, 8.0, 8.0, 8.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0,
        7.0, 8.0, 8.0, 7.0, 6.0, 5.0, 4.0, 3.0, 2.0, 1.0, 1.0, 1.0, 1.0, 1.0, 2.0, 2.0, 2.0, 2.0, 3.0, 3.0, 3.0, 3.0,
        4.0, 4.0, 4.0, 4.0, 8.0, 7.0, 6.0, 5.0, 4.0, 3.0, 2.0, 1.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0,
    ];

    let x = Tensor::from_slice(&data).try_reshape([4, 4, 4]).unwrap();
    let idx =
        Tensor::from_slice_with().source(&indices).device(svod_ir::DeviceSpec::Cpu).call().try_reshape([2, 1]).unwrap();
    let upd = Tensor::from_slice(&updates).try_reshape([2, 4, 4]).unwrap();

    let mut result = x.scatter_nd(&idx, &upd, "none").unwrap();
    result.realize_with(&cfg).unwrap();
    let actual = result.as_vec::<f32>().unwrap();
    assert_eq!(actual, expected);
}