onnx_test.cpp 37.2 KB
Newer Older
Scott Thornton's avatar
Scott Thornton committed
1
2
#include <iostream>
#include <vector>
Paul's avatar
Paul committed
3
4
5
6
#include <migraphx/literal.hpp>
#include <migraphx/operators.hpp>
#include <migraphx/program.hpp>
#include <migraphx/instruction.hpp>
Shucai Xiao's avatar
Shucai Xiao committed
7
#include <migraphx/instruction_ref.hpp>
Paul's avatar
Paul committed
8
#include <migraphx/onnx.hpp>
Scott Thornton's avatar
Scott Thornton committed
9
10
#include "test.hpp"

Khalique's avatar
Khalique committed
11
TEST_CASE(acos_test)
Scott Thornton's avatar
Scott Thornton committed
12
{
Paul's avatar
Paul committed
13
    migraphx::program p;
Khalique's avatar
Khalique committed
14
15
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10}});
    p.add_instruction(migraphx::op::acos{}, input);
Scott Thornton's avatar
Scott Thornton committed
16

Khalique's avatar
Khalique committed
17
    auto prog = migraphx::parse_onnx("acos_test.onnx");
Scott Thornton's avatar
Scott Thornton committed
18
19
20
21

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
22
TEST_CASE(add_bcast_test)
23
{
Paul's avatar
Paul committed
24
    migraphx::program p;
Khalique's avatar
Khalique committed
25
26
27
28
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {2, 3, 4, 5}});
    auto l1 = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {3, 4}});
    auto l2 = p.add_instruction(migraphx::op::broadcast{1, l0->get_shape().lens()}, l1);
    p.add_instruction(migraphx::op::add{}, l0, l2);
Paul's avatar
Paul committed
29

Khalique's avatar
Khalique committed
30
    auto prog = migraphx::parse_onnx("add_bcast_test.onnx");
Paul's avatar
Paul committed
31

32
33
34
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
35
TEST_CASE(add_fp16_test)
Scott Thornton's avatar
Scott Thornton committed
36
{
Paul's avatar
Paul committed
37
    migraphx::program p;
Khalique's avatar
Khalique committed
38
39
40
41
42
43
    auto l0 =
        p.add_literal(migraphx::literal{migraphx::shape{migraphx::shape::half_type, {1}}, {1.5}});
    auto l1 =
        p.add_literal(migraphx::literal{migraphx::shape{migraphx::shape::half_type, {1}}, {2.5}});
    p.add_instruction(migraphx::op::add{}, l0, l1);
    auto prog = migraphx::parse_onnx("add_fp16_test.onnx");
Scott Thornton's avatar
Scott Thornton committed
44
45
46
47

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
48
TEST_CASE(add_scalar_test)
49
{
Paul's avatar
Paul committed
50
    migraphx::program p;
Khalique's avatar
Khalique committed
51
52
53
54
55
56
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {2, 3, 4, 5}});
    auto l1 = p.add_literal(migraphx::literal{migraphx::shape{migraphx::shape::float_type}, {1}});
    auto m0 = p.add_instruction(migraphx::op::multibroadcast{{2, 3, 4, 5}}, l0);
    auto m1 = p.add_instruction(migraphx::op::multibroadcast{{2, 3, 4, 5}}, l1);
    p.add_instruction(migraphx::op::add{}, m0, m1);
    auto prog = migraphx::parse_onnx("add_scalar_test.onnx");
57
58
59
60

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
61
TEST_CASE(argmax_test)
Khalique's avatar
Khalique committed
62
{
Paul's avatar
Paul committed
63
    migraphx::program p;
Khalique's avatar
Khalique committed
64
65
66
67
    auto l0  = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {3, 4, 5, 6}});
    auto ins = p.add_instruction(migraphx::op::argmax{2}, l0);
    p.add_instruction(migraphx::op::squeeze{{2}}, ins);
    auto prog = migraphx::parse_onnx("argmax_test.onnx");
Khalique's avatar
Khalique committed
68
69
70
71

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
72
TEST_CASE(argmin_test)
73
{
Paul's avatar
Paul committed
74
    migraphx::program p;
Khalique's avatar
Khalique committed
75
76
77
78
    auto l0  = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {3, 4, 5, 6}});
    auto ins = p.add_instruction(migraphx::op::argmin{3}, l0);
    p.add_instruction(migraphx::op::squeeze{{3}}, ins);
    auto prog = migraphx::parse_onnx("argmin_test.onnx");
79
80
81
82

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
83
TEST_CASE(asin_test)
84
{
Paul's avatar
Paul committed
85
    migraphx::program p;
Khalique's avatar
Khalique committed
86
87
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10}});
    p.add_instruction(migraphx::op::asin{}, input);
88

Khalique's avatar
Khalique committed
89
    auto prog = migraphx::parse_onnx("asin_test.onnx");
90
91
92
93

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
94
TEST_CASE(atan_test)
Khalique's avatar
Khalique committed
95
{
Paul's avatar
Paul committed
96
    migraphx::program p;
Khalique's avatar
Khalique committed
97
98
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10}});
    p.add_instruction(migraphx::op::atan{}, input);
Khalique's avatar
Khalique committed
99

Khalique's avatar
Khalique committed
100
    auto prog = migraphx::parse_onnx("atan_test.onnx");
Khalique's avatar
Khalique committed
101
102
103
104

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
105
TEST_CASE(cast_test)
Khalique's avatar
Khalique committed
106
{
Paul's avatar
Paul committed
107
    migraphx::program p;
Khalique's avatar
Khalique committed
108
109
    auto l = p.add_parameter("x", migraphx::shape{migraphx::shape::half_type, {10}});
    p.add_instruction(migraphx::op::convert{migraphx::shape::float_type}, l);
Khalique's avatar
Khalique committed
110

Khalique's avatar
Khalique committed
111
    auto prog = migraphx::parse_onnx("cast_test.onnx");
Khalique's avatar
Khalique committed
112
113
114
    EXPECT(p == prog);
}

Shucai Xiao's avatar
Shucai Xiao committed
115
116
117
118
119
120
121
122
123
124
125
TEST_CASE(ceil_test)
{
    migraphx::program p;
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10}});
    p.add_instruction(migraphx::op::ceil{}, input);

    auto prog = migraphx::parse_onnx("ceil_test.onnx");

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
126
TEST_CASE(clip_test)
Khalique's avatar
Khalique committed
127
128
{
    migraphx::program p;
Khalique's avatar
Khalique committed
129
130
131
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {3}});
    p.add_instruction(migraphx::op::clip{6.0, 0.0}, l0);
    auto prog = migraphx::parse_onnx("clip_test.onnx");
Khalique's avatar
Khalique committed
132

133
134
135
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
136
TEST_CASE(concat_test)
Shucai Xiao's avatar
Shucai Xiao committed
137
138
{
    migraphx::program p;
Khalique's avatar
Khalique committed
139
140
141
142
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {2, 4, 3}});
    auto l1 = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {7, 4, 3}});
    p.add_instruction(migraphx::op::concat{0}, l0, l1);
    auto prog = migraphx::parse_onnx("concat_test.onnx");
Shucai Xiao's avatar
Shucai Xiao committed
143
144
145
146

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
147
TEST_CASE(constant_test)
Shucai Xiao's avatar
Shucai Xiao committed
148
149
{
    migraphx::program p;
Khalique's avatar
Khalique committed
150
151
    p.add_literal(migraphx::literal{migraphx::shape{migraphx::shape::float_type, {3}}, {0, 1, 2}});
    auto prog = migraphx::parse_onnx("constant_test.onnx");
Shucai Xiao's avatar
Shucai Xiao committed
152
153
154
155

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
156
TEST_CASE(constant_fill_test)
Shucai Xiao's avatar
Shucai Xiao committed
157
{
Khalique's avatar
Khalique committed
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180

    migraphx::program p;
    migraphx::shape s{migraphx::shape::float_type, {2, 3}};
    std::vector<float> value(s.elements(), 1.0);
    p.add_literal(migraphx::literal{s, value});
    auto prog = migraphx::parse_onnx("constant_fill_test.onnx");

    EXPECT(p == prog);
}

TEST_CASE(constant_fill_input_as_shape_test)
{
    migraphx::program p;
    auto l0 = p.add_literal(migraphx::literal{{migraphx::shape::int32_type, {2}}, {2, 3}});
    std::vector<std::size_t> dims(l0->get_shape().elements());
    migraphx::literal ls = l0->get_literal();
    ls.visit([&](auto s) { dims.assign(s.begin(), s.end()); });
    migraphx::shape s{migraphx::shape::float_type, dims};
    std::vector<float> value(s.elements(), 1.0);
    p.add_literal(migraphx::literal{s, value});
    auto prog = migraphx::parse_onnx("constant_fill_input_as_shape_test.onnx");

    EXPECT(p == prog);
Shucai Xiao's avatar
Shucai Xiao committed
181
182
}

Khalique's avatar
Khalique committed
183
TEST_CASE(constant_scalar_test)
184
185
{
    migraphx::program p;
Khalique's avatar
Khalique committed
186
187
    p.add_literal(migraphx::literal{migraphx::shape{migraphx::shape::int32_type, {1}}, {1}});
    auto prog = migraphx::parse_onnx("constant_scalar_test.onnx");
188
189
190
191

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
192
TEST_CASE(const_of_shape_empty_input_test)
Shucai Xiao's avatar
Shucai Xiao committed
193
194
{
    migraphx::program p;
Khalique's avatar
Khalique committed
195
196
197
198
    p.add_literal(migraphx::literal());
    migraphx::shape s(migraphx::shape::int64_type, {1}, {0});
    std::vector<int64_t> vec(s.elements(), 10);
    p.add_literal(migraphx::literal(s, vec));
Shucai Xiao's avatar
Shucai Xiao committed
199

Khalique's avatar
Khalique committed
200
    auto prog = migraphx::parse_onnx("const_of_shape_empty_input_test.onnx");
Shucai Xiao's avatar
Shucai Xiao committed
201
    EXPECT(p == prog);
Khalique's avatar
Khalique committed
202
203
}

Khalique's avatar
Khalique committed
204
TEST_CASE(const_of_shape_float_test)
205
206
{
    migraphx::program p;
Khalique's avatar
Khalique committed
207
208
209
210
211
    migraphx::shape ss(migraphx::shape::int32_type, {3});
    p.add_literal(migraphx::literal(ss, {2, 3, 4}));
    migraphx::shape s(migraphx::shape::float_type, {2, 3, 4});
    std::vector<float> vec(s.elements(), 10.0f);
    p.add_literal(migraphx::literal(s, vec));
212

Khalique's avatar
Khalique committed
213
    auto prog = migraphx::parse_onnx("const_of_shape_float_test.onnx");
214
215
216
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
217
TEST_CASE(const_of_shape_int64_test)
218
219
{
    migraphx::program p;
Khalique's avatar
Khalique committed
220
221
222
223
224
    migraphx::shape ss(migraphx::shape::int32_type, {3});
    p.add_literal(migraphx::literal(ss, {2, 3, 4}));
    migraphx::shape s(migraphx::shape::int64_type, {2, 3, 4});
    std::vector<int64_t> vec(s.elements(), 10);
    p.add_literal(migraphx::literal(s, vec));
225

Khalique's avatar
Khalique committed
226
    auto prog = migraphx::parse_onnx("const_of_shape_int64_test.onnx");
227
228
229
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
230
TEST_CASE(const_of_shape_no_value_attr_test)
231
232
{
    migraphx::program p;
Khalique's avatar
Khalique committed
233
234
235
236
237
    migraphx::shape ss(migraphx::shape::int32_type, {3});
    p.add_literal(migraphx::literal(ss, {2, 3, 4}));
    migraphx::shape s(migraphx::shape::float_type, {2, 3, 4});
    std::vector<float> vec(s.elements(), 0.0f);
    p.add_literal(migraphx::literal(s, vec));
238

Khalique's avatar
Khalique committed
239
    auto prog = migraphx::parse_onnx("const_of_shape_no_value_attr_test.onnx");
240
241
242
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
243
TEST_CASE(conv_autopad_fail_test)
244
{
Khalique's avatar
Khalique committed
245
246
    EXPECT(test::throws([&] { migraphx::parse_onnx("conv_autopad_fail_test.onnx"); }));
}
247

Khalique's avatar
Khalique committed
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
TEST_CASE(conv_bias_test)
{
    migraphx::program p;
    auto l0       = p.add_parameter("0", {migraphx::shape::float_type, {1, 3, 32, 32}});
    auto l1       = p.add_parameter("1", {migraphx::shape::float_type, {1, 3, 5, 5}});
    auto l2       = p.add_parameter("2", {migraphx::shape::float_type, {1}});
    uint64_t axis = 1;
    auto l3       = p.add_instruction(migraphx::op::convolution{}, l0, l1);
    auto l4       = p.add_instruction(migraphx::op::broadcast{axis, l3->get_shape().lens()}, l2);
    p.add_instruction(migraphx::op::add{}, l3, l4);

    auto prog = migraphx::parse_onnx("conv_bias_test.onnx");
    EXPECT(p == prog);
}

TEST_CASE(conv_bn_relu_maxpool_test)
{
    migraphx::program p;
    auto l0 = p.add_parameter("0", {migraphx::shape::float_type, {1, 3, 32, 32}});
    auto l1 = p.add_parameter("1", {migraphx::shape::float_type, {1, 3, 5, 5}});
    auto l2 = p.add_parameter("2", {migraphx::shape::float_type, {1}});

    auto p3       = p.add_parameter("3", {migraphx::shape::float_type, {1}});
    auto p4       = p.add_parameter("4", {migraphx::shape::float_type, {1}});
    auto p5       = p.add_parameter("5", {migraphx::shape::float_type, {1}});
    auto p6       = p.add_parameter("6", {migraphx::shape::float_type, {1}});
    uint64_t axis = 1;
    auto l3       = p.add_instruction(migraphx::op::convolution{}, l0, l1);
    auto l4       = p.add_instruction(migraphx::op::broadcast{axis, l3->get_shape().lens()}, l2);
    auto l5       = p.add_instruction(migraphx::op::add{}, l3, l4);
    auto l6 = p.add_instruction(migraphx::op::batch_norm_inference{1.0e-5f}, l5, p3, p4, p5, p6);
    auto l7 = p.add_instruction(migraphx::op::relu{}, l6);
    p.add_instruction(migraphx::op::pooling{"max", {{0, 0}}, {{2, 2}}, {{2, 2}}}, l7);

    auto prog = migraphx::parse_onnx("conv_bn_relu_maxpool_test.onnx");
    EXPECT(p == prog);
}

TEST_CASE(conv_relu_maxpool_test)
{
    migraphx::program p;
    auto l0       = p.add_parameter("0", {migraphx::shape::float_type, {1, 3, 32, 32}});
    auto l1       = p.add_parameter("1", {migraphx::shape::float_type, {1, 3, 5, 5}});
    auto l2       = p.add_parameter("2", {migraphx::shape::float_type, {1}});
    uint64_t axis = 1;
    auto l3       = p.add_instruction(migraphx::op::convolution{}, l0, l1);
    auto l4       = p.add_instruction(migraphx::op::broadcast{axis, l3->get_shape().lens()}, l2);
    auto l5       = p.add_instruction(migraphx::op::add{}, l3, l4);
    auto l6       = p.add_instruction(migraphx::op::relu{}, l5);
    p.add_instruction(migraphx::op::pooling{"max", {{0, 0}}, {{2, 2}}, {{2, 2}}}, l6);

    auto prog = migraphx::parse_onnx("conv_relu_maxpool_test.onnx");
    EXPECT(p == prog);
}

TEST_CASE(conv_relu_maxpool_x2_test)
{
    migraphx::program p;
    auto l0       = p.add_parameter("0", {migraphx::shape::float_type, {1, 3, 32, 32}});
    auto l1       = p.add_parameter("1", {migraphx::shape::float_type, {5, 3, 5, 5}});
    auto l2       = p.add_parameter("2", {migraphx::shape::float_type, {5}});
    uint64_t axis = 1;
    auto l3       = p.add_instruction(migraphx::op::convolution{}, l0, l1);
    auto l4       = p.add_instruction(migraphx::op::broadcast{axis, l3->get_shape().lens()}, l2);
    auto l5       = p.add_instruction(migraphx::op::add{}, l3, l4);
    auto l6       = p.add_instruction(migraphx::op::relu{}, l5);
    auto l7 = p.add_instruction(migraphx::op::pooling{"max", {{0, 0}}, {{2, 2}}, {{2, 2}}}, l6);

    auto l8  = p.add_parameter("3", {migraphx::shape::float_type, {1, 5, 5, 5}});
    auto l9  = p.add_parameter("4", {migraphx::shape::float_type, {1}});
    auto l10 = p.add_instruction(migraphx::op::convolution{}, l7, l8);
    auto l11 = p.add_instruction(migraphx::op::broadcast{axis, l10->get_shape().lens()}, l9);
    auto l12 = p.add_instruction(migraphx::op::add{}, l10, l11);
    auto l13 = p.add_instruction(migraphx::op::relu{}, l12);
    p.add_instruction(migraphx::op::pooling{"max", {{0, 0}}, {{2, 2}}, {{2, 2}}}, l13);

    auto prog = migraphx::parse_onnx("conv_relu_maxpool_x2_test.onnx");

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
329
TEST_CASE(cos_test)
330
331
332
{
    migraphx::program p;
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10}});
Khalique's avatar
Khalique committed
333
    p.add_instruction(migraphx::op::cos{}, input);
334

Khalique's avatar
Khalique committed
335
    auto prog = migraphx::parse_onnx("cos_test.onnx");
336
337
338
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
339
TEST_CASE(cosh_test)
340
341
342
343
344
345
346
347
348
349
{
    migraphx::program p;
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {1}});
    p.add_instruction(migraphx::op::cosh{}, input);

    auto prog = migraphx::parse_onnx("cosh_test.onnx");

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
350
TEST_CASE(dropout_test)
Shucai Xiao's avatar
Shucai Xiao committed
351
352
{
    migraphx::program p;
Khalique's avatar
Khalique committed
353
354
    auto input = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {1, 3, 2, 2}});
    p.add_instruction(migraphx::op::identity{}, input);
Shucai Xiao's avatar
Shucai Xiao committed
355

Khalique's avatar
Khalique committed
356
    auto prog = migraphx::parse_onnx("dropout_test.onnx");
Shucai Xiao's avatar
Shucai Xiao committed
357
358
359
360

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
361
362
363
364
365
366
367
368
369
370
371
TEST_CASE(elu_test)
{
    migraphx::program p;
    auto input = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {3}});
    p.add_instruction(migraphx::op::elu{0.01}, input);

    auto prog = migraphx::parse_onnx("elu_test.onnx");

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
372
TEST_CASE(erf_test)
373
374
{
    migraphx::program p;
Khalique's avatar
Khalique committed
375
376
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10, 15}});
    p.add_instruction(migraphx::op::erf{}, input);
Khalique's avatar
Khalique committed
377

Khalique's avatar
Khalique committed
378
    auto prog = migraphx::parse_onnx("erf_test.onnx");
Khalique's avatar
Khalique committed
379
380
381
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
382
TEST_CASE(exp_test)
383
384
385
{
    migraphx::program p;
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10}});
Khalique's avatar
Khalique committed
386
    p.add_instruction(migraphx::op::exp{}, input);
Khalique's avatar
Khalique committed
387

Khalique's avatar
Khalique committed
388
    auto prog = migraphx::parse_onnx("exp_test.onnx");
Khalique's avatar
Khalique committed
389
    EXPECT(p == prog);
Khalique's avatar
Khalique committed
390
391
}

Khalique's avatar
Khalique committed
392
TEST_CASE(expand_test)
393
394
{
    migraphx::program p;
Khalique's avatar
Khalique committed
395
396
397
398
399
    migraphx::shape s(migraphx::shape::float_type, {3, 1, 1});
    auto param = p.add_parameter("x", s);
    migraphx::shape ss(migraphx::shape::int32_type, {4});
    p.add_literal(migraphx::literal(ss, {2, 3, 4, 5}));
    p.add_instruction(migraphx::op::multibroadcast{{2, 3, 4, 5}}, param);
Khalique's avatar
Khalique committed
400

Khalique's avatar
Khalique committed
401
    auto prog = migraphx::parse_onnx("expand_test.onnx");
Khalique's avatar
Khalique committed
402
    EXPECT(p == prog);
Khalique's avatar
Khalique committed
403
404
}

Khalique's avatar
Khalique committed
405
TEST_CASE(flatten_test)
406
407
408
{
    migraphx::program p;
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {2, 3, 4, 5}});
Khalique's avatar
Khalique committed
409
410
411
    p.add_instruction(migraphx::op::flatten{2}, l0);
    p.add_instruction(migraphx::op::flatten{1}, l0);
    auto prog = migraphx::parse_onnx("flatten_test.onnx");
412
413
414
415

    EXPECT(p == prog);
}

Shucai Xiao's avatar
Shucai Xiao committed
416
417
418
419
420
421
422
423
424
425
426
TEST_CASE(floor_test)
{
    migraphx::program p;
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10}});
    p.add_instruction(migraphx::op::floor{}, input);

    auto prog = migraphx::parse_onnx("floor_test.onnx");

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
427
TEST_CASE(gather_test)
428
429
{
    migraphx::program p;
Khalique's avatar
Khalique committed
430
431
432
433
434
    auto l0  = p.add_parameter("data", migraphx::shape{migraphx::shape::float_type, {3, 4, 5, 6}});
    auto l1  = p.add_parameter("indices", migraphx::shape{migraphx::shape::int32_type, {2, 3}});
    int axis = 1;
    p.add_instruction(migraphx::op::gather{axis}, l0, l1);
    auto prog = migraphx::parse_onnx("gather_test.onnx");
435
436
437
438

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
439
TEST_CASE(gemm_test)
440
441
{
    migraphx::program p;
Khalique's avatar
Khalique committed
442
443
444
445
446
447
448
449
450
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {5, 7}});
    auto l1 = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {11, 5}});
    p.add_parameter("2", migraphx::shape{migraphx::shape::float_type, {}});
    auto t0    = p.add_instruction(migraphx::op::transpose{{1, 0}}, l0);
    auto t1    = p.add_instruction(migraphx::op::transpose{{1, 0}}, l1);
    auto alpha = 2.f;
    auto beta  = 2.0f;
    p.add_instruction(migraphx::op::dot{alpha, beta}, t0, t1);
    auto prog = migraphx::parse_onnx("gemm_test.onnx");
451
452
453
454

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
455
TEST_CASE(gemm_ex_test)
456
457
{
    migraphx::program p;
Khalique's avatar
Khalique committed
458
459
460
461
462
463
464
465
    auto l0    = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {1, 1, 5, 6}});
    auto l1    = p.add_parameter("2", migraphx::shape{migraphx::shape::float_type, {1, 1, 5, 7}});
    auto l2    = p.add_parameter("3", migraphx::shape{migraphx::shape::float_type, {1, 1, 6, 7}});
    auto t0    = p.add_instruction(migraphx::op::transpose{{0, 1, 3, 2}}, l0);
    auto alpha = 0.5f;
    auto beta  = 0.8f;
    p.add_instruction(migraphx::op::dot{alpha, beta}, t0, l1, l2);
    auto prog = migraphx::parse_onnx("gemm_ex_test.onnx");
466
467
468
469

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
470
TEST_CASE(gemm_ex_brcst_test)
471
472
{
    migraphx::program p;
Khalique's avatar
Khalique committed
473
474
475
476
477
478
479
480
481
    auto l0 = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {1, 1, 5, 6}});
    auto l1 = p.add_parameter("2", migraphx::shape{migraphx::shape::float_type, {1, 1, 5, 7}});
    auto l2 = p.add_parameter("3", migraphx::shape{migraphx::shape::float_type, {1, 1, 6, 1}});
    auto t0 = p.add_instruction(migraphx::op::transpose{{0, 1, 3, 2}}, l0);
    std::vector<std::size_t> out_lens{1, 1, 6, 7};
    auto t2    = p.add_instruction(migraphx::op::multibroadcast{out_lens}, l2);
    auto alpha = 0.5f;
    auto beta  = 0.8f;
    p.add_instruction(migraphx::op::dot{alpha, beta}, t0, l1, t2);
Khalique's avatar
Khalique committed
482
    auto prog = migraphx::parse_onnx("gemm_ex_brcst_test.onnx");
483
484
485
486

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
487
TEST_CASE(globalavgpool_test)
488
489
{
    migraphx::program p;
Khalique's avatar
Khalique committed
490
491
492
493
494
495
496
    auto input = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {1, 3, 16, 16}});
    auto op    = migraphx::op::pooling{"average"};
    auto lens  = input->get_shape().lens();
    op.lengths = {lens[2], lens[3]};
    p.add_instruction(op, input);

    auto prog = migraphx::parse_onnx("globalavgpool_test.onnx");
497
498
499
500

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
501
TEST_CASE(globalmaxpool_test)
Khalique's avatar
Khalique committed
502
503
{
    migraphx::program p;
Khalique's avatar
Khalique committed
504
505
506
507
508
509
510
    auto input = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {1, 3, 16, 16}});
    auto op    = migraphx::op::pooling{"max"};
    auto lens  = input->get_shape().lens();
    op.lengths = {lens[2], lens[3]};
    p.add_instruction(op, input);

    auto prog = migraphx::parse_onnx("globalmaxpool_test.onnx");
Khalique's avatar
Khalique committed
511
512
513
514

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
515
TEST_CASE(group_conv_test)
516
517
{
    migraphx::program p;
Khalique's avatar
Khalique committed
518
519
520
521
522
523
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {1, 4, 16, 16}});
    auto l1 = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {4, 1, 3, 3}});
    migraphx::op::convolution op;
    op.group = 4;
    p.add_instruction(op, l0, l1);
    auto prog = migraphx::parse_onnx("group_conv_test.onnx");
524
525
526
527

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
528
TEST_CASE(imagescaler_test)
529
530
{
    migraphx::program p;
Khalique's avatar
Khalique committed
531
532
533
534
535
536
537
538
539
540
541
    migraphx::shape s{migraphx::shape::float_type, {1, 3, 16, 16}};
    auto l0        = p.add_parameter("0", s);
    auto scale_val = p.add_literal(0.5f);
    auto bias_vals = p.add_literal(
        migraphx::literal{migraphx::shape{migraphx::shape::float_type, {3}}, {0.01, 0.02, 0.03}});
    auto scaled_tensor = p.add_instruction(migraphx::op::scalar{s.lens()}, scale_val);
    auto img_scaled    = p.add_instruction(migraphx::op::mul{}, l0, scaled_tensor);
    auto bias_bcast    = p.add_instruction(migraphx::op::broadcast{1, s.lens()}, bias_vals);
    p.add_instruction(migraphx::op::add{}, img_scaled, bias_bcast);

    auto prog = migraphx::parse_onnx("imagescaler_test.onnx");
542
543
544
545

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
546
TEST_CASE(implicit_add_bcast_test)
547
548
{
    migraphx::program p;
Khalique's avatar
Khalique committed
549
550
551
552
553
554
555
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {2, 3, 4, 5}});
    auto l1 = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {3, 4, 1}});
    auto l2 = p.add_instruction(migraphx::op::multibroadcast{{2, 3, 4, 5}}, l0);
    auto l3 = p.add_instruction(migraphx::op::multibroadcast{{2, 3, 4, 5}}, l1);
    p.add_instruction(migraphx::op::add{}, l2, l3);

    auto prog = migraphx::parse_onnx("implicit_add_bcast_test.onnx");
556
557
558
559

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
560
TEST_CASE(implicit_pow_bcast_test)
561
562
{
    migraphx::program p;
Khalique's avatar
Khalique committed
563
564
565
566
567
568
569
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {2, 3, 4, 5}});
    auto l1 = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {3, 4, 1}});
    auto l2 = p.add_instruction(migraphx::op::multibroadcast{{2, 3, 4, 5}}, l0);
    auto l3 = p.add_instruction(migraphx::op::multibroadcast{{2, 3, 4, 5}}, l1);
    p.add_instruction(migraphx::op::pow{}, l2, l3);

    auto prog = migraphx::parse_onnx("implicit_pow_bcast_test.onnx");
570
571
572
573

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
574
TEST_CASE(implicit_sub_bcast_test)
575
576
577
{
    migraphx::program p;
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {2, 3, 4, 5}});
Khalique's avatar
Khalique committed
578
579
580
581
582
583
    auto l1 = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {4, 5}});
    auto l2 = p.add_instruction(migraphx::op::multibroadcast{{2, 3, 4, 5}}, l0);
    auto l3 = p.add_instruction(migraphx::op::multibroadcast{{2, 3, 4, 5}}, l1);
    p.add_instruction(migraphx::op::sub{}, l2, l3);

    auto prog = migraphx::parse_onnx("implicit_sub_bcast_test.onnx");
584
585
586
587

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
588
TEST_CASE(leaky_relu_test)
589
590
{
    migraphx::program p;
Khalique's avatar
Khalique committed
591
592
593
594
595
    float alpha = 0.01f;
    auto l0     = p.add_parameter("0", {migraphx::shape::float_type, {3}});
    p.add_instruction(migraphx::op::leaky_relu{alpha}, l0);

    auto prog = migraphx::parse_onnx("leaky_relu_test.onnx");
596
597
598
599

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
600
TEST_CASE(log_test)
601
602
{
    migraphx::program p;
Khalique's avatar
Khalique committed
603
604
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10}});
    p.add_instruction(migraphx::op::log{}, input);
605

Khalique's avatar
Khalique committed
606
    auto prog = migraphx::parse_onnx("log_test.onnx");
607
608
609
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
610
TEST_CASE(logsoftmax_test)
611
612
{
    migraphx::program p;
Khalique's avatar
Khalique committed
613
614
615
616
    auto l0  = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {3, 4, 5, 6}});
    int axis = 1;
    p.add_instruction(migraphx::op::logsoftmax{axis}, l0);
    auto prog = migraphx::parse_onnx("logsoftmax_test.onnx");
617
618
619
620

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
621
TEST_CASE(lrn_test)
622
623
{
    migraphx::program p;
Khalique's avatar
Khalique committed
624
625
626
627
628
629
630
631
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {1, 28, 24, 24}});
    migraphx::op::lrn op;
    op.size  = 5;
    op.alpha = 0.0001;
    op.beta  = 0.75;
    op.bias  = 1.0;
    p.add_instruction(op, l0);
    auto prog = migraphx::parse_onnx("lrn_test.onnx");
632
633
634
635

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
636
TEST_CASE(matmul_bmbm_test)
637
638
{
    migraphx::program p;
Khalique's avatar
Khalique committed
639
640
641
642
643
644
645
    auto l0  = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {3, 6, 7}});
    auto l1  = p.add_parameter("2", migraphx::shape{migraphx::shape::float_type, {5, 2, 1, 7, 8}});
    auto bl0 = p.add_instruction(migraphx::op::multibroadcast{{5, 2, 3, 6, 7}}, l0);
    auto bl1 = p.add_instruction(migraphx::op::multibroadcast{{5, 2, 3, 7, 8}}, l1);
    p.add_instruction(migraphx::op::dot{1.0f, 0.0f}, bl0, bl1);

    auto prog = migraphx::parse_onnx("matmul_bmbm_test.onnx");
646
647
648
649

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
650
TEST_CASE(matmul_bmv_test)
651
{
Khalique's avatar
Khalique committed
652
653
654
655
656
657
658
    migraphx::program p;
    auto l0   = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {3, 6, 7}});
    auto l1   = p.add_parameter("2", migraphx::shape{migraphx::shape::float_type, {7}});
    auto sl1  = p.add_instruction(migraphx::op::unsqueeze{{1}}, l1);
    auto bsl1 = p.add_instruction(migraphx::op::multibroadcast{{3, 7, 1}}, sl1);
    auto res  = p.add_instruction(migraphx::op::dot{1.0f, 0.0f}, l0, bsl1);
    p.add_instruction(migraphx::op::squeeze{{2}}, res);
659

Khalique's avatar
Khalique committed
660
    auto prog = migraphx::parse_onnx("matmul_bmv_test.onnx");
661

Khalique's avatar
Khalique committed
662
    EXPECT(p == prog);
663
664
}

Khalique's avatar
Khalique committed
665
TEST_CASE(matmul_mv_test)
666
667
{
    migraphx::program p;
Khalique's avatar
Khalique committed
668
669
670
671
672
673
674
    auto l0  = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {6, 7}});
    auto l1  = p.add_parameter("2", migraphx::shape{migraphx::shape::float_type, {7}});
    auto sl1 = p.add_instruction(migraphx::op::unsqueeze{{1}}, l1);
    auto res = p.add_instruction(migraphx::op::dot{1.0f, 0.0f}, l0, sl1);
    p.add_instruction(migraphx::op::squeeze{{1}}, res);

    auto prog = migraphx::parse_onnx("matmul_mv_test.onnx");
675
676
677
678

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
679
TEST_CASE(matmul_vbm_test)
680
681
{
    migraphx::program p;
Khalique's avatar
Khalique committed
682
683
684
685
686
687
688
689
690
691
    auto l0   = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {7}});
    auto l1   = p.add_parameter("2", migraphx::shape{migraphx::shape::float_type, {5, 7, 8}});
    auto sl0  = p.add_instruction(migraphx::op::unsqueeze{{0}}, l0);
    auto bsl0 = p.add_instruction(migraphx::op::multibroadcast{{5, 1, 7}}, sl0);
    std::cout << "ONNX_TEST" << std::endl;
    auto res = p.add_instruction(migraphx::op::dot{1.0f, 0.0f}, bsl0, l1);
    std::cout << "After Dot" << std::endl;
    p.add_instruction(migraphx::op::squeeze{{1}}, res);

    auto prog = migraphx::parse_onnx("matmul_vbm_test.onnx");
692
693
694
695

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
696
TEST_CASE(matmul_vm_test)
697
698
{
    migraphx::program p;
Khalique's avatar
Khalique committed
699
700
701
702
703
704
705
    auto l0  = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {7}});
    auto l1  = p.add_parameter("2", migraphx::shape{migraphx::shape::float_type, {7, 8}});
    auto sl0 = p.add_instruction(migraphx::op::unsqueeze{{0}}, l0);
    auto res = p.add_instruction(migraphx::op::dot{1.0f, 0.0f}, sl0, l1);
    p.add_instruction(migraphx::op::squeeze{{0}}, res);

    auto prog = migraphx::parse_onnx("matmul_vm_test.onnx");
706
707
708
709

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
710
TEST_CASE(matmul_vv_test)
711
712
{
    migraphx::program p;
Shucai Xiao's avatar
Shucai Xiao committed
713
714
    auto l0  = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {7}});
    auto l1  = p.add_parameter("2", migraphx::shape{migraphx::shape::float_type, {7}});
715
716
717
718
719
720
    auto sl0 = p.add_instruction(migraphx::op::unsqueeze{{0}}, l0);
    auto sl1 = p.add_instruction(migraphx::op::unsqueeze{{1}}, l1);
    auto res = p.add_instruction(migraphx::op::dot{1.0f, 0.0f}, sl0, sl1);
    auto sr0 = p.add_instruction(migraphx::op::squeeze{{0}}, res);
    p.add_instruction(migraphx::op::squeeze{{0}}, sr0);

Khalique's avatar
Khalique committed
721
    auto prog = migraphx::parse_onnx("matmul_vv_test.onnx");
722
723
724
725

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
726
TEST_CASE(max_test)
727
728
{
    migraphx::program p;
Khalique's avatar
Khalique committed
729
730
731
732
733
    auto input0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {3}});
    auto input1 = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {3}});
    auto input2 = p.add_parameter("2", migraphx::shape{migraphx::shape::float_type, {3}});
    auto l0     = p.add_instruction(migraphx::op::max{}, input0, input1);
    p.add_instruction(migraphx::op::max{}, l0, input2);
734

Khalique's avatar
Khalique committed
735
736
    migraphx::parse_onnx("max_test.onnx");
}
737

Khalique's avatar
Khalique committed
738
739
740
741
742
743
744
745
746
747
TEST_CASE(min_test)
{
    migraphx::program p;
    auto input0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {3}});
    auto input1 = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {3}});
    auto input2 = p.add_parameter("2", migraphx::shape{migraphx::shape::float_type, {3}});
    auto l0     = p.add_instruction(migraphx::op::min{}, input0, input1);
    p.add_instruction(migraphx::op::min{}, l0, input2);

    migraphx::parse_onnx("min_test.onnx");
748
749
}

Khalique's avatar
Khalique committed
750
TEST_CASE(no_pad_test)
751
752
{
    migraphx::program p;
Khalique's avatar
Khalique committed
753
754
755
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {2, 2}});
    p.add_instruction(migraphx::op::identity{}, l0);
    auto prog = migraphx::parse_onnx("no_pad_test.onnx");
756

Khalique's avatar
Khalique committed
757
758
759
760
761
762
763
764
765
    EXPECT(p == prog);
}

TEST_CASE(pad_test)
{
    migraphx::program p;
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {2, 2}});
    p.add_instruction(migraphx::op::pad{{1, 1, 1, 1}}, l0);
    auto prog = migraphx::parse_onnx("pad_test.onnx");
766
767
768
769

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
770
TEST_CASE(pow_test)
771
772
{
    migraphx::program p;
Khalique's avatar
Khalique committed
773
774
775
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {2, 3, 4, 5}});
    auto l1 = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {2, 3, 4, 5}});
    p.add_instruction(migraphx::op::pow{}, l0, l1);
776

Khalique's avatar
Khalique committed
777
    auto prog = migraphx::parse_onnx("pow_test.onnx");
778
779
780
781

    EXPECT(p == prog);
}

Shucai Xiao's avatar
Shucai Xiao committed
782
783
784
785
786
787
788
789
790
791
TEST_CASE(reducemax_test)
{
    migraphx::program p;
    auto l0 = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {3, 4, 5, 6}});
    p.add_instruction(migraphx::op::reduce_max{{2}}, l0);
    auto prog = migraphx::parse_onnx("reducemax_test.onnx");

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
792
TEST_CASE(reducemean_test)
793
794
{
    migraphx::program p;
Khalique's avatar
Khalique committed
795
796
797
798
    auto l0 = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {3, 4, 5, 6}});
    auto l1 = p.add_instruction(migraphx::op::reduce_mean{{2, 3}}, l0);
    p.add_instruction(migraphx::op::squeeze{{2, 3}}, l1);
    auto prog = migraphx::parse_onnx("reducemean_test.onnx");
799
800
801
802

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
803
TEST_CASE(reducemean_keepdims_test)
804
805
{
    migraphx::program p;
Khalique's avatar
Khalique committed
806
807
808
    auto l0 = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {3, 4, 5, 6}});
    p.add_instruction(migraphx::op::reduce_mean{{2}}, l0);
    auto prog = migraphx::parse_onnx("reducemean_keepdims_test.onnx");
809
810
811
812

    EXPECT(p == prog);
}

Shucai Xiao's avatar
Shucai Xiao committed
813
814
815
816
817
818
819
820
821
822
823
TEST_CASE(reducemin_test)
{
    migraphx::program p;
    auto l0 = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {3, 4, 5, 6}});
    auto l1 = p.add_instruction(migraphx::op::reduce_min{{2, 3}}, l0);
    p.add_instruction(migraphx::op::squeeze{{2, 3}}, l1);
    auto prog = migraphx::parse_onnx("reducemin_test.onnx");

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
824
TEST_CASE(reducesum_test)
Khalique's avatar
Khalique committed
825
826
{
    migraphx::program p;
Khalique's avatar
Khalique committed
827
828
829
    auto l0 = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {3, 4, 5, 6}});
    auto l1 = p.add_instruction(migraphx::op::reduce_sum{{2}}, l0);
    p.add_instruction(migraphx::op::squeeze{{2}}, l1);
Khalique's avatar
Khalique committed
830
    auto prog = migraphx::parse_onnx("reducesum_test.onnx");
831
832

    EXPECT(p == prog);
Khalique's avatar
Khalique committed
833
834
}

Khalique's avatar
Khalique committed
835
TEST_CASE(reducesum_multiaxis_test)
Khalique's avatar
Khalique committed
836
837
{
    migraphx::program p;
Khalique's avatar
Khalique committed
838
839
840
    auto l0 = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {3, 4, 5, 6}});
    auto l1 = p.add_instruction(migraphx::op::reduce_sum{{2, 3}}, l0);
    p.add_instruction(migraphx::op::squeeze{{2, 3}}, l1);
Khalique's avatar
Khalique committed
841
    auto prog = migraphx::parse_onnx("reducesum_multiaxis_test.onnx");
842
843

    EXPECT(p == prog);
Khalique's avatar
Khalique committed
844
845
}

Khalique's avatar
Khalique committed
846
TEST_CASE(reducesum_keepdims_test)
Khalique's avatar
Khalique committed
847
848
{
    migraphx::program p;
Khalique's avatar
Khalique committed
849
850
    auto l0 = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {3, 4, 5, 6}});
    p.add_instruction(migraphx::op::reduce_sum{{2, 3}}, l0);
Khalique's avatar
Khalique committed
851
    auto prog = migraphx::parse_onnx("reducesum_keepdims_test.onnx");
852
853
854
855

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
856
TEST_CASE(reshape_test)
857
{
Khalique's avatar
Khalique committed
858
    migraphx::program p;
Khalique's avatar
Khalique committed
859
860
861
862
863
864
865
866
867
    migraphx::op::reshape op;
    std::vector<int64_t> reshape_dims{3, 8};
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {4, 2, 3}});
    p.add_literal(
        migraphx::literal{migraphx::shape{migraphx::shape::int64_type, {2}}, reshape_dims});
    op.dims = reshape_dims;
    p.add_instruction(op, l0);
    p.add_instruction(op, l0);
    auto prog = migraphx::parse_onnx("reshape_test.onnx");
868
869

    EXPECT(p == prog);
Khalique's avatar
Khalique committed
870
871
}

Khalique's avatar
Khalique committed
872
TEST_CASE(reshape_non_standard_test)
873
874
{
    migraphx::program p;
Khalique's avatar
Khalique committed
875
876
877
878
879
880
881
    migraphx::op::reshape op;
    std::vector<int64_t> reshape_dims{4, 3, 2};
    migraphx::shape s{migraphx::shape::float_type, {2, 3, 4}};
    auto x      = p.add_parameter("x", s);
    auto tran_x = p.add_instruction(migraphx::op::transpose{{0, 2, 1}}, x);
    auto cont_x = p.add_instruction(migraphx::op::contiguous{}, tran_x);
    p.add_instruction(migraphx::op::reshape{{4, 3, 2}}, cont_x);
Khalique's avatar
Khalique committed
882
    auto prog = migraphx::parse_onnx("reshape_non_standard_test.onnx");
883
884
885
886

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
887
888
889
890
891
892
893
894
895
896
TEST_CASE(round_test)
{
    migraphx::program p;
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::double_type, {10, 5}});
    p.add_instruction(migraphx::op::round{}, input);

    auto prog = migraphx::parse_onnx("round_test.onnx");
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
897
TEST_CASE(shape_test)
898
899
{
    migraphx::program p;
Khalique's avatar
Khalique committed
900
901
902
903
904
    migraphx::shape s{migraphx::shape::float_type, {3, 4, 5, 6}};
    auto l0 = p.add_parameter("x", s);
    migraphx::shape s_shape{migraphx::shape::int64_type, {4}};
    p.add_literal(s_shape, l0->get_shape().lens());
    auto prog = migraphx::parse_onnx("shape_test.onnx");
905
906
907
908

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
909
TEST_CASE(shape_gather_test)
910
911
{
    migraphx::program p;
Khalique's avatar
Khalique committed
912
913
914
915
916
917
918
    auto l0 = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {7, 3, 10}});
    auto l1 =
        p.add_literal(migraphx::shape{migraphx::shape::int64_type, {3}}, l0->get_shape().lens());
    migraphx::shape const_shape{migraphx::shape::int32_type, {1}};
    auto l2  = p.add_literal(migraphx::literal{const_shape, {1}});
    int axis = 0;
    p.add_instruction(migraphx::op::gather{axis}, l1, l2);
Khalique's avatar
Khalique committed
919
    auto prog = migraphx::parse_onnx("shape_gather_test.onnx");
920
921
922
923

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
924
TEST_CASE(sign_test)
Khalique's avatar
Khalique committed
925
926
{
    migraphx::program p;
Khalique's avatar
Khalique committed
927
928
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::double_type, {10, 5}});
    p.add_instruction(migraphx::op::sign{}, input);
Khalique's avatar
Khalique committed
929

Khalique's avatar
Khalique committed
930
    auto prog = migraphx::parse_onnx("sign_test.onnx");
Khalique's avatar
Khalique committed
931
932
933
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
934
TEST_CASE(sin_test)
935
936
{
    migraphx::program p;
Khalique's avatar
Khalique committed
937
938
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10}});
    p.add_instruction(migraphx::op::sin{}, input);
939

Khalique's avatar
Khalique committed
940
    auto prog = migraphx::parse_onnx("sin_test.onnx");
941
942
943
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
944
TEST_CASE(sinh_test)
945
946
{
    migraphx::program p;
Khalique's avatar
Khalique committed
947
948
949
950
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10}});
    p.add_instruction(migraphx::op::sinh{}, input);

    auto prog = migraphx::parse_onnx("sinh_test.onnx");
951
952
953
954

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
955
TEST_CASE(slice_test)
956
957
{
    migraphx::program p;
Khalique's avatar
Khalique committed
958
959
960
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {3, 2}});
    p.add_instruction(migraphx::op::slice{{0, 1}, {1, 0}, {2, 2}}, l0);
    auto prog = migraphx::parse_onnx("slice_test.onnx");
961
962
963
964

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
965
TEST_CASE(softmax_test)
Shucai Xiao's avatar
Shucai Xiao committed
966
967
{
    migraphx::program p;
Khalique's avatar
Khalique committed
968
969
970
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {1, 3}});
    p.add_instruction(migraphx::op::softmax{1}, l0);
    auto prog = migraphx::parse_onnx("softmax_test.onnx");
Shucai Xiao's avatar
Shucai Xiao committed
971
972
973
974

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
975
TEST_CASE(sqrt_test)
Shucai Xiao's avatar
Shucai Xiao committed
976
977
{
    migraphx::program p;
Khalique's avatar
Khalique committed
978
979
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10, 15}});
    p.add_instruction(migraphx::op::sqrt{}, input);
Shucai Xiao's avatar
Shucai Xiao committed
980

Khalique's avatar
Khalique committed
981
    auto prog = migraphx::parse_onnx("sqrt_test.onnx");
Shucai Xiao's avatar
Shucai Xiao committed
982
983
984
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
985
TEST_CASE(squeeze_unsqueeze_test)
Khalique's avatar
Khalique committed
986
987
{
    migraphx::program p;
Khalique's avatar
Khalique committed
988
989
990
991
992
993
994
    std::vector<int64_t> squeeze_axes{0, 2, 3, 5};
    std::vector<int64_t> unsqueeze_axes{0, 1, 3, 5};
    auto l0 =
        p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {1, 3, 1, 1, 2, 1}});
    auto l1 = p.add_instruction(migraphx::op::squeeze{squeeze_axes}, l0);
    p.add_instruction(migraphx::op::unsqueeze{unsqueeze_axes}, l1);
    auto prog = migraphx::parse_onnx("squeeze_unsqueeze_test.onnx");
Khalique's avatar
Khalique committed
995
996
997
998

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
999
TEST_CASE(sub_bcast_test)
Shucai Xiao's avatar
Shucai Xiao committed
1000
1001
1002
{
    migraphx::program p;
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {2, 3, 4, 5}});
Khalique's avatar
Khalique committed
1003
1004
1005
    auto l1 = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {3, 4}});
    auto l2 = p.add_instruction(migraphx::op::broadcast{1, l0->get_shape().lens()}, l1);
    p.add_instruction(migraphx::op::sub{}, l0, l2);
Shucai Xiao's avatar
Shucai Xiao committed
1006

Khalique's avatar
Khalique committed
1007
    auto prog = migraphx::parse_onnx("sub_bcast_test.onnx");
Shucai Xiao's avatar
Shucai Xiao committed
1008
1009
1010
1011

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
1012
TEST_CASE(sub_scalar_test)
Shucai Xiao's avatar
Shucai Xiao committed
1013
1014
1015
{
    migraphx::program p;
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {2, 3, 4, 5}});
Khalique's avatar
Khalique committed
1016
1017
1018
1019
1020
1021
    auto l1 =
        p.add_literal(migraphx::literal{migraphx::shape{migraphx::shape::float_type, {1}}, {1}});
    auto m0 = p.add_instruction(migraphx::op::multibroadcast{{2, 3, 4, 5}}, l0);
    auto m1 = p.add_instruction(migraphx::op::multibroadcast{{2, 3, 4, 5}}, l1);
    p.add_instruction(migraphx::op::sub{}, m0, m1);
    auto prog = migraphx::parse_onnx("sub_scalar_test.onnx");
Shucai Xiao's avatar
Shucai Xiao committed
1022
1023
1024
1025

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
1026
TEST_CASE(sum_test)
1027
1028
{
    migraphx::program p;
Khalique's avatar
Khalique committed
1029
1030
1031
1032
1033
    auto input0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {3}});
    auto input1 = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {3}});
    auto input2 = p.add_parameter("2", migraphx::shape{migraphx::shape::float_type, {3}});
    auto l0     = p.add_instruction(migraphx::op::add{}, input0, input1);
    p.add_instruction(migraphx::op::add{}, l0, input2);
1034

Khalique's avatar
Khalique committed
1035
    auto prog = migraphx::parse_onnx("sum_test.onnx");
1036
1037
1038
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
1039
TEST_CASE(tan_test)
1040
1041
{
    migraphx::program p;
Khalique's avatar
Khalique committed
1042
1043
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10}});
    p.add_instruction(migraphx::op::tan{}, input);
1044

Khalique's avatar
Khalique committed
1045
    auto prog = migraphx::parse_onnx("tan_test.onnx");
1046
1047
1048
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
1049
TEST_CASE(tanh_test)
1050
1051
{
    migraphx::program p;
Khalique's avatar
Khalique committed
1052
1053
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {1}});
    p.add_instruction(migraphx::op::tanh{}, input);
1054

Khalique's avatar
Khalique committed
1055
    auto prog = migraphx::parse_onnx("tanh_test.onnx");
1056
1057
1058
1059

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
1060
TEST_CASE(transpose_test)
1061
1062
{
    migraphx::program p;
Khalique's avatar
Khalique committed
1063
1064
1065
1066
1067
    auto input = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {1, 2, 2, 3}});
    std::vector<int64_t> perm{0, 3, 1, 2};
    p.add_instruction(migraphx::op::transpose{perm}, input);

    auto prog = migraphx::parse_onnx("transpose_test.onnx");
1068
1069
1070
1071

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
1072
TEST_CASE(transpose_gather_test)
1073
1074
{
    migraphx::program p;
Khalique's avatar
Khalique committed
1075
1076
1077
1078
1079
1080
1081
1082
1083
1084
1085
1086
1087
1088
1089
1090
1091
1092
1093
    auto make_contiguous = [&p](migraphx::instruction_ref ins) {
        if(ins->get_shape().standard())
        {
            return ins;
        }

        return p.add_instruction(migraphx::op::contiguous{}, ins);
    };

    auto data = p.add_parameter("data", migraphx::shape{migraphx::shape::float_type, {3, 5, 4, 6}});
    auto ind =
        p.add_parameter("indices", migraphx::shape{migraphx::shape::int32_type, {2, 4, 3, 5}});
    auto tr_data = p.add_instruction(migraphx::op::transpose{{0, 2, 1, 3}}, data);
    auto tr_ind  = p.add_instruction(migraphx::op::transpose{{0, 2, 1, 3}}, ind);
    int axis     = 1;
    p.add_instruction(
        migraphx::op::gather{axis}, make_contiguous(tr_data), make_contiguous(tr_ind));

    auto prog = migraphx::parse_onnx("transpose_gather_test.onnx");
1094
1095
1096
1097

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
1098
TEST_CASE(unknown_test)
1099
1100
{
    migraphx::program p;
Khalique's avatar
Khalique committed
1101
1102
1103
1104
1105
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {2, 3, 4, 5}});
    auto l1 = p.add_parameter("1", migraphx::shape{migraphx::shape::float_type, {3, 4}});
    auto l2 = p.add_instruction(migraphx::op::unknown{"Unknown"}, l0, l1);
    p.add_instruction(migraphx::op::unknown{"Unknown"}, l2);
    auto prog = migraphx::parse_onnx("unknown_test.onnx");
1106
1107
1108
1109

    EXPECT(p == prog);
}

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