shape_test.cpp 4.42 KB
Newer Older
Paul's avatar
Paul committed
1

Paul's avatar
Paul committed
2
#include <migraph/shape.hpp>
Paul's avatar
Paul committed
3
4
5
#include <array>
#include <algorithm>
#include <numeric>
Paul's avatar
Paul committed
6
7
#include "test.hpp"

8
9
10
11
12
13
14
void test_shape_default()
{
    migraph::shape s{};
    EXPECT(s.elements() == 0);
    EXPECT(s.bytes() == 0);
}

Paul's avatar
Paul committed
15
16
void test_shape_assign()
{
Paul's avatar
Paul committed
17
18
    migraph::shape s1{migraph::shape::float_type, {100, 32, 8, 8}};
    migraph::shape s2 = s1; // NOLINT
Paul's avatar
Paul committed
19
20
21
22
    EXPECT(s1 == s2);
    EXPECT(!(s1 != s2));
}

Paul's avatar
Paul committed
23
24
25
void test_shape_packed_default()
{
    migraph::shape s{migraph::shape::float_type, {2, 2}};
Paul's avatar
Paul committed
26
    EXPECT(s.standard());
Paul's avatar
Paul committed
27
    EXPECT(s.packed());
Paul's avatar
Paul committed
28
29
    EXPECT(not s.transposed());
    EXPECT(not s.broadcasted());
Paul's avatar
Paul committed
30
31
32
33
34
}

void test_shape_packed()
{
    migraph::shape s{migraph::shape::float_type, {2, 2}, {2, 1}};
Paul's avatar
Paul committed
35
    EXPECT(s.standard());
Paul's avatar
Paul committed
36
    EXPECT(s.packed());
Paul's avatar
Paul committed
37
38
    EXPECT(not s.transposed());
    EXPECT(not s.broadcasted());
Paul's avatar
Paul committed
39
40
41
42
43
}

void test_shape_transposed()
{
    migraph::shape s{migraph::shape::float_type, {2, 2}, {1, 2}};
Paul's avatar
Paul committed
44
45
46
47
48
49
50
51
52
53
    EXPECT(not s.standard());
    EXPECT(s.packed());
    EXPECT(s.transposed());
    EXPECT(not s.broadcasted());
}

void test_shape_broadcasted()
{
    migraph::shape s{migraph::shape::float_type, {2, 2}, {1, 0}};
    EXPECT(not s.standard());
Paul's avatar
Paul committed
54
    EXPECT(not s.packed());
Paul's avatar
Paul committed
55
56
    EXPECT(not s.transposed());
    EXPECT(s.broadcasted());
Paul's avatar
Paul committed
57
58
}

59
void test_shape_default_copy()
Paul's avatar
Paul committed
60
{
Paul's avatar
Paul committed
61
62
    migraph::shape s1{};
    migraph::shape s2{};
Paul's avatar
Paul committed
63
64
65
66
    EXPECT(s1 == s2);
    EXPECT(!(s1 != s2));
}

Paul's avatar
Paul committed
67
68
void test_shape4()
{
Paul's avatar
Paul committed
69
    migraph::shape s{migraph::shape::float_type, {100, 32, 8, 8}};
Paul's avatar
Paul committed
70
    EXPECT(s.standard());
Paul's avatar
Paul committed
71
    EXPECT(s.packed());
Paul's avatar
Paul committed
72
73
    EXPECT(not s.transposed());
    EXPECT(not s.broadcasted());
Paul's avatar
Paul committed
74
    EXPECT(s.type() == migraph::shape::float_type);
Paul's avatar
Paul committed
75
76
77
78
79
80
81
82
    EXPECT(s.lens()[0] == 100);
    EXPECT(s.lens()[1] == 32);
    EXPECT(s.lens()[2] == 8);
    EXPECT(s.lens()[3] == 8);
    EXPECT(s.strides()[0] == s.lens()[1] * s.strides()[1]);
    EXPECT(s.strides()[1] == s.lens()[2] * s.strides()[2]);
    EXPECT(s.strides()[2] == s.lens()[3] * s.strides()[3]);
    EXPECT(s.strides()[3] == 1);
Paul's avatar
Paul committed
83
84
    EXPECT(s.elements() == 100 * 32 * 8 * 8);
    EXPECT(s.bytes() == 100 * 32 * 8 * 8 * sizeof(float));
Paul's avatar
Paul committed
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
    EXPECT(s.index({0, 0, 0, 0}) == 0);
    EXPECT(s.index({0, 0, 0, 1}) == 1);
    EXPECT(s.index({0, 0, 0, 0}) == s.index(0));
    EXPECT(s.index({0, 0, 0, 1}) == s.index(1));
    EXPECT(s.index({0, 0, 1, 0}) == s.index(8));
    EXPECT(s.index({0, 1, 0, 0}) == s.index(8 * 8));
    EXPECT(s.index({1, 0, 0, 0}) == s.index(8 * 8 * 32));
    EXPECT(s.index(0) == 0);
    EXPECT(s.index(1) == 1);
    EXPECT(s.index(8) == 8);
    EXPECT(s.index(8 * 8) == 8 * 8);
    EXPECT(s.index(8 * 8 * 32) == 8 * 8 * 32);
    EXPECT(s.index(s.elements() - 1) == s.elements() - 1);
}

void test_shape4_nonpacked()
{
    std::vector<std::size_t> lens       = {100, 32, 8, 8};
Paul's avatar
Paul committed
103
104
    std::array<std::size_t, 4> offsets  = {{5, 10, 0, 6}};
    std::array<std::size_t, 4> adj_lens = {{0, 0, 0, 0}};
Paul's avatar
Paul committed
105
106
107
108
109
110

    std::transform(
        lens.begin(), lens.end(), offsets.begin(), adj_lens.begin(), std::plus<size_t>());
    // adj_lens should be: { 105, 42, 8, 14 }
    std::vector<std::size_t> strides(4);
    strides.back() = 1;
Paul's avatar
Paul committed
111
112
113
114
    std::partial_sum(adj_lens.rbegin(),
                     adj_lens.rend() - 1,
                     strides.rbegin() + 1,
                     std::multiplies<std::size_t>());
Paul's avatar
Paul committed
115

Paul's avatar
Paul committed
116
    migraph::shape s{migraph::shape::float_type, lens, strides};
Paul's avatar
Paul committed
117
118
119
120
    EXPECT(not s.standard());
    EXPECT(not s.packed());
    EXPECT(not s.transposed());
    EXPECT(not s.broadcasted());
Paul's avatar
Paul committed
121
    EXPECT(s.type() == migraph::shape::float_type);
Paul's avatar
Paul committed
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
    EXPECT(s.lens()[0] == 100);
    EXPECT(s.lens()[1] == 32);
    EXPECT(s.lens()[2] == 8);
    EXPECT(s.lens()[3] == 8);
    EXPECT(s.strides()[0] == 4704);
    EXPECT(s.strides()[1] == 112);
    EXPECT(s.strides()[2] == 14);
    EXPECT(s.strides()[3] == 1);
    EXPECT(s.elements() == 100 * 32 * 8 * 8);
    EXPECT(s.bytes() == sizeof(float) * 469274);

    EXPECT(s.index(0) == 0);
    EXPECT(s.index(1) == 1);
    EXPECT(s.index({0, 0, 0, 0}) == 0);
    EXPECT(s.index({0, 0, 0, 1}) == s.index(1));
    // TODO: Fix these tests
    // EXPECT(s.index({0, 0, 1, 0}) == s.index(8));
    // EXPECT(s.index({0, 1, 0, 0}) == s.index(8 * 8));
    // EXPECT(s.index({1, 0, 0, 0}) == s.index(8 * 8 * 32));
    // EXPECT(s.index(s.elements() - 1) == 469273);
Paul's avatar
Paul committed
142
143
}

Paul's avatar
Paul committed
144
145
int main()
{
146
    test_shape_default();
Paul's avatar
Paul committed
147
    test_shape_assign();
Paul's avatar
Paul committed
148
149
150
    test_shape_packed_default();
    test_shape_packed();
    test_shape_transposed();
Paul's avatar
Paul committed
151
    test_shape_broadcasted();
152
    test_shape_default_copy();
Paul's avatar
Paul committed
153
    test_shape4();
Paul's avatar
Paul committed
154
    test_shape4_nonpacked();
Paul's avatar
Paul committed
155
}