auto_contiguous_test.cpp 3.22 KB
Newer Older
Paul's avatar
Paul committed
1
2
3
#include <migraphx/auto_contiguous.hpp>
#include <migraphx/operators.hpp>
#include <migraphx/instruction.hpp>
Paul's avatar
Paul committed
4
5
6
#include <basic_ops.hpp>
#include <test.hpp>

Paul's avatar
Paul committed
7
struct contiguous_target
Paul's avatar
Paul committed
8
{
Paul's avatar
Paul committed
9
    std::string name() const { return "contiguous"; }
Paul's avatar
Paul committed
10
    std::vector<migraphx::pass> get_passes(migraphx::context&) const
Paul's avatar
Paul committed
11
    {
Paul's avatar
Paul committed
12
        return {migraphx::auto_contiguous{}};
Paul's avatar
Paul committed
13
    }
Paul's avatar
Paul committed
14
    migraphx::context get_context() const { return {}; }
Paul's avatar
Paul committed
15
16
};

Paul's avatar
Paul committed
17
// TODO: Add this test case
Paul's avatar
Paul committed
18
19
void literal_broadcast()
{
Paul's avatar
Paul committed
20
    migraphx::program p;
Paul's avatar
Paul committed
21
22
23
24
25
26
27
28
    p.add_literal(get_2_broadcasted());
    EXPECT(not p.get_shape().standard());
    EXPECT(p.get_shape().broadcasted());
    p.compile(contiguous_target{});
    EXPECT(p.get_shape().standard());
    EXPECT(not p.get_shape().broadcasted());
}

Paul's avatar
Paul committed
29
TEST_CASE(literal_transpose)
Paul's avatar
Paul committed
30
{
Paul's avatar
Paul committed
31
    migraphx::program p;
Paul's avatar
Paul committed
32
33
34
35
36
37
38
39
    p.add_literal(get_2x2_transposed());
    EXPECT(not p.get_shape().standard());
    EXPECT(p.get_shape().transposed());
    p.compile(contiguous_target{});
    EXPECT(p.get_shape().standard());
    EXPECT(not p.get_shape().transposed());
}

Paul's avatar
Paul committed
40
TEST_CASE(after_literal_transpose)
Paul's avatar
Paul committed
41
{
Paul's avatar
Paul committed
42
    migraphx::program p;
Paul's avatar
Paul committed
43
    auto l = p.add_literal(get_2x2());
Paul's avatar
Paul committed
44
45
    EXPECT(p.get_shape().standard());
    EXPECT(not p.get_shape().transposed());
Paul's avatar
Paul committed
46
    auto t = p.add_instruction(migraphx::op::transpose{{1, 0}}, l);
Paul's avatar
Paul committed
47
    p.add_instruction(pass_op{}, t);
Paul's avatar
Paul committed
48
49
    EXPECT(not p.get_shape().standard());
    EXPECT(p.get_shape().transposed());
Paul's avatar
Paul committed
50
    p.compile(contiguous_target{});
Paul's avatar
Paul committed
51
52
    EXPECT(p.get_shape().standard());
    EXPECT(not p.get_shape().transposed());
Paul's avatar
Paul committed
53
54
}

Paul's avatar
Paul committed
55
TEST_CASE(after_literal_broadcast)
Paul's avatar
Paul committed
56
{
Paul's avatar
Paul committed
57
    migraphx::program p;
Paul's avatar
Paul committed
58
59
60
61
    auto l1 = p.add_literal(get_2x2());
    auto l2 = p.add_literal(get_2());
    EXPECT(p.get_shape().standard());
    EXPECT(not p.get_shape().broadcasted());
Paul's avatar
Paul committed
62
    auto b = p.add_instruction(migraphx::op::broadcast{0, l1->get_shape()}, l2);
Paul's avatar
Paul committed
63
64
65
66
67
68
69
70
    p.add_instruction(pass_op{}, b);
    EXPECT(not p.get_shape().standard());
    EXPECT(p.get_shape().broadcasted());
    p.compile(contiguous_target{});
    EXPECT(p.get_shape().standard());
    EXPECT(not p.get_shape().broadcasted());
}

Paul's avatar
Paul committed
71
TEST_CASE(after_param_transpose)
Paul's avatar
Paul committed
72
{
Paul's avatar
Paul committed
73
74
    migraphx::program p;
    auto l = p.add_parameter("2x2", {migraphx::shape::float_type, {2, 2}});
Paul's avatar
Paul committed
75
76
    EXPECT(p.get_shape().standard());
    EXPECT(not p.get_shape().transposed());
Paul's avatar
Paul committed
77
    auto t = p.add_instruction(migraphx::op::transpose{{1, 0}}, l);
Paul's avatar
Paul committed
78
79
80
81
82
83
84
85
    p.add_instruction(pass_op{}, t);
    EXPECT(not p.get_shape().standard());
    EXPECT(p.get_shape().transposed());
    p.compile(contiguous_target{});
    EXPECT(p.get_shape().standard());
    EXPECT(not p.get_shape().transposed());
}

Paul's avatar
Paul committed
86
TEST_CASE(after_param_broadcast)
Paul's avatar
Paul committed
87
{
Paul's avatar
Paul committed
88
89
90
    migraphx::program p;
    auto l1 = p.add_parameter("2x2", {migraphx::shape::float_type, {2, 2}});
    auto l2 = p.add_parameter("2", {migraphx::shape::float_type, {2}});
Paul's avatar
Paul committed
91
92
    EXPECT(p.get_shape().standard());
    EXPECT(not p.get_shape().broadcasted());
Paul's avatar
Paul committed
93
    auto b = p.add_instruction(migraphx::op::broadcast{0, l1->get_shape()}, l2);
Paul's avatar
Paul committed
94
95
96
97
98
99
100
101
    p.add_instruction(pass_op{}, b);
    EXPECT(not p.get_shape().standard());
    EXPECT(p.get_shape().broadcasted());
    p.compile(contiguous_target{});
    EXPECT(p.get_shape().standard());
    EXPECT(not p.get_shape().broadcasted());
}

Paul's avatar
Paul committed
102
int main(int argc, const char* argv[]) { test::run(argc, argv); }