runmat-vm 0.4.4

RunMat virtual machine and bytecode interpreter
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
70
71
72
73
74
75
76
77
78
79
80
81

#[path = "support/mod.rs"]
mod test_helpers;

use runmat_builtins::Value;
use runmat_parser::parse;
use test_helpers::execute;
use test_helpers::lower;

#[test]
fn linear_index_and_end_keyword() {
    // v = A(end) and A(1) linear indexing
    let ast = parse("A=[1,2;3,4]; v1=A(1); v2=A(end);").unwrap();
    let hir = lower(&ast).unwrap();
    let vars = execute(&hir).unwrap();
    if let Value::Num(v1) = &vars[1] {
        assert_eq!(*v1, 1.0);
    } else {
        panic!("v1");
    }
    if let Value::Num(v2) = &vars[2] {
        assert_eq!(*v2, 4.0);
    } else {
        panic!("v2");
    }
}

#[test]
fn logical_mask_indexing() {
    // A(logical([1 0 1 0])) over linearized A
    let ast = parse("A=[1,2;3,4]; idx=logical([1,0,1,0]); v=A(idx);").unwrap();
    let hir = lower(&ast).unwrap();
    let vars = execute(&hir).unwrap();
    // MATLAB uses column-major linearization: A(:) = [1;3;2;4], mask [1 0 1 0] selects [1,2]
    if let Value::Tensor(v) = &vars[2] {
        // MATLAB logical linear indexing returns a column vector.
        assert_eq!(v.shape, vec![2, 1]);
        assert_eq!(v.data, vec![1.0, 2.0]);
    } else {
        panic!("v");
    }
}

#[test]
fn logical_mask_linear_indexing_row_vector_returns_column() {
    let ast = parse("A=[10,20,30,40]; idx=logical([1,0,1,0]); v=A(idx);").unwrap();
    let hir = lower(&ast).unwrap();
    let vars = execute(&hir).unwrap();
    if let Value::Tensor(v) = &vars[2] {
        assert_eq!(v.shape, vec![2, 1]);
        assert_eq!(v.data, vec![10.0, 30.0]);
    } else {
        panic!("v");
    }
}

#[test]
fn logical_mask_linear_indexing_matrix_mask_returns_column() {
    let ast = parse("A=[1,2;3,4]; idx=logical([1,0;0,1]); v=A(idx);").unwrap();
    let hir = lower(&ast).unwrap();
    let vars = execute(&hir).unwrap();
    // mask selects A(1,1) and A(2,2) in column-major linear order
    if let Value::Tensor(v) = &vars[2] {
        assert_eq!(v.shape, vec![2, 1]);
        assert_eq!(v.data, vec![1.0, 4.0]);
    } else {
        panic!("v");
    }
}

#[test]
fn host_linear_indexing_accepts_gpu_backed_range_selector() {
    let ast = parse("a=length(1:10); k=floor(a/2)+1; x=(1:10)'; y=x(1:k);").unwrap();
    let hir = lower(&ast).unwrap();
    let vars = execute(&hir).unwrap();
    if let Value::Tensor(v) = &vars[3] {
        assert_eq!(v.shape, vec![1, 6]);
        assert_eq!(v.data, vec![1.0, 2.0, 3.0, 4.0, 5.0, 6.0]);
    } else {
        panic!("y");
    }
}