Skip to content

GitLab

"src/vscode:/vscode.git/clone" did not exist on "585bb3312c9e29a2abf69cba4041be0f2662bb18"
Switch branch/tag
Find file Normal viewHistoryPermalink
common_subexpression_elimination_test.cpp 3.67 KB
Newer Older
Paul's avatar
Add initial cse  
Paul committed
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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121 
#include <migraph/common_subexpression_elimination.hpp>
#include <migraph/dead_code_elimination.hpp>
#include <migraph/operators.hpp>
#include <basic_ops.hpp>
#include <test.hpp>

struct cse_target
{
    std::string name() const { return "dce"; }
    std::vector<migraph::pass> get_passes(migraph::context&) const
    {
        return {migraph::common_subexpression_elimination{}, migraph::dead_code_elimination{}};
    }
    migraph::context get_context() const { return {}; }
};

void cse_test1()
{
    migraph::program p1;
    {
        auto one = p1.add_literal(1);
        auto two = p1.add_literal(2);
        auto sum1 = p1.add_instruction(migraph::op::add{}, one, two);
        auto sum2 = p1.add_instruction(migraph::op::add{}, one, two);
        auto sum3 = p1.add_instruction(migraph::op::add{}, sum1, sum2);
        p1.add_instruction(pass_op{}, sum3);
    }
    p1.compile(cse_target{});

    migraph::program p2;
    {
        auto one = p2.add_literal(1);
        auto two = p2.add_literal(2);
        auto sum1 = p2.add_instruction(migraph::op::add{}, one, two);
        auto sum3 = p2.add_instruction(migraph::op::add{}, sum1, sum1);
        p2.add_instruction(pass_op{}, sum3);
    }
    EXPECT(p1 == p2);
}

void cse_test2()
{
    migraph::program p1;
    {
        auto one = p1.add_literal(1);
        auto two = p1.add_literal(2);
        auto sum1 = p1.add_instruction(migraph::op::add{}, one, two);
        auto sum2 = p1.add_instruction(migraph::op::add{}, two, one);
        auto sum3 = p1.add_instruction(migraph::op::add{}, sum1, sum2);
        p1.add_instruction(pass_op{}, sum3);
    }
    p1.compile(cse_target{});

    migraph::program p2;
    {
        auto one = p2.add_literal(1);
        auto two = p2.add_literal(2);
        auto sum1 = p2.add_instruction(migraph::op::add{}, one, two);
        auto sum2 = p2.add_instruction(migraph::op::add{}, two, one);
        auto sum3 = p2.add_instruction(migraph::op::add{}, sum1, sum2);
        p2.add_instruction(pass_op{}, sum3);
    }
    EXPECT(p1 == p2);
}

void cse_test3()
{
    migraph::program p1;
    {
        auto one = p1.add_literal(1);
        auto two = p1.add_literal(1);
        auto sum1 = p1.add_instruction(migraph::op::add{}, one, two);
        auto sum2 = p1.add_instruction(migraph::op::add{}, two, one);
        auto sum3 = p1.add_instruction(migraph::op::add{}, sum1, sum2);
        p1.add_instruction(pass_op{}, sum3);
    }
    p1.compile(cse_target{});

    migraph::program p2;
    {
        auto one = p2.add_literal(1);
        auto sum1 = p2.add_instruction(migraph::op::add{}, one, one);
        auto sum3 = p2.add_instruction(migraph::op::add{}, sum1, sum1);
        p2.add_instruction(pass_op{}, sum3);
    }
    EXPECT(p1 == p2);
}

void cse_test4()
{
    migraph::program p1;
    {
        auto one = p1.add_literal(1);
        auto two = p1.add_literal(1);
        auto sum1 = p1.add_instruction(migraph::op::add{}, one, two);
        auto sum2 = p1.add_instruction(migraph::op::add{}, two, one);
        auto sum3 = p1.add_instruction(migraph::op::add{}, sum1, one);
        auto sum4 = p1.add_instruction(migraph::op::add{}, sum2, two);
        auto sum5 = p1.add_instruction(migraph::op::add{}, sum4, sum3);
        p1.add_instruction(pass_op{}, sum5);
    }
    p1.compile(cse_target{});

    migraph::program p2;
    {
        auto one = p2.add_literal(1);
        auto sum1 = p2.add_instruction(migraph::op::add{}, one, one);
        auto sum3 = p2.add_instruction(migraph::op::add{}, sum1, one);
        auto sum5 = p2.add_instruction(migraph::op::add{}, sum3, sum3);
        p2.add_instruction(pass_op{}, sum5);
    }
    EXPECT(p1 == p2);
}

int main() 
{
    cse_test1();
    cse_test2();
    cse_test3();
    cse_test4();
}