onnx_test.cpp 34.7 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
7
#include <migraphx/literal.hpp>
#include <migraphx/operators.hpp>
#include <migraphx/program.hpp>
#include <migraphx/instruction.hpp>
#include <migraphx/onnx.hpp>
Scott Thornton's avatar
Scott Thornton committed
8
9
#include "test.hpp"

Khalique's avatar
Khalique committed
10
TEST_CASE(acos_test)
Scott Thornton's avatar
Scott Thornton committed
11
{
Paul's avatar
Paul committed
12
    migraphx::program p;
Khalique's avatar
Khalique committed
13
14
    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
15

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

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
21
TEST_CASE(add_bcast_test)
22
{
Paul's avatar
Paul committed
23
    migraphx::program p;
Khalique's avatar
Khalique committed
24
25
26
27
    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
28

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

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

Khalique's avatar
Khalique committed
34
TEST_CASE(add_fp16_test)
Scott Thornton's avatar
Scott Thornton committed
35
{
Paul's avatar
Paul committed
36
    migraphx::program p;
Khalique's avatar
Khalique committed
37
38
39
40
41
42
    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
43
44
45
46

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
47
TEST_CASE(add_scalar_test)
48
{
Paul's avatar
Paul committed
49
    migraphx::program p;
Khalique's avatar
Khalique committed
50
51
52
53
54
55
    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");
56
57
58
59

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
60
TEST_CASE(argmax)
Khalique's avatar
Khalique committed
61
{
Paul's avatar
Paul committed
62
    migraphx::program p;
Khalique's avatar
Khalique committed
63
64
65
66
    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
67
68
69
70

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
71
TEST_CASE(argmin)
72
{
Paul's avatar
Paul committed
73
    migraphx::program p;
Khalique's avatar
Khalique committed
74
75
76
77
    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");
78
79
80
81

    EXPECT(p == prog);
}

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

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

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
93
TEST_CASE(atan_test)
Khalique's avatar
Khalique committed
94
{
Paul's avatar
Paul committed
95
    migraphx::program p;
Khalique's avatar
Khalique committed
96
97
    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
98

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

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
104
TEST_CASE(cast_test)
Khalique's avatar
Khalique committed
105
{
Paul's avatar
Paul committed
106
    migraphx::program p;
Khalique's avatar
Khalique committed
107
108
    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
109

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

Khalique's avatar
Khalique committed
114
TEST_CASE(clip_test)
Khalique's avatar
Khalique committed
115
116
{
    migraphx::program p;
Khalique's avatar
Khalique committed
117
118
119
    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
120

121
122
123
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
124
TEST_CASE(concat_test)
Shucai Xiao's avatar
Shucai Xiao committed
125
126
{
    migraphx::program p;
Khalique's avatar
Khalique committed
127
128
129
130
    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
131
132
133
134

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
135
TEST_CASE(constant_test)
Shucai Xiao's avatar
Shucai Xiao committed
136
137
{
    migraphx::program p;
Khalique's avatar
Khalique committed
138
139
    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
140
141
142
143

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
144
TEST_CASE(constant_fill_test)
Shucai Xiao's avatar
Shucai Xiao committed
145
{
Khalique's avatar
Khalique committed
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168

    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
169
170
}

Khalique's avatar
Khalique committed
171
TEST_CASE(constant_scalar_test)
172
173
{
    migraphx::program p;
Khalique's avatar
Khalique committed
174
175
    p.add_literal(migraphx::literal{migraphx::shape{migraphx::shape::int32_type, {1}}, {1}});
    auto prog = migraphx::parse_onnx("constant_scalar_test.onnx");
176
177
178
179

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
180
TEST_CASE(const_of_shape_empty_input_test)
Shucai Xiao's avatar
Shucai Xiao committed
181
182
{
    migraphx::program p;
Khalique's avatar
Khalique committed
183
184
185
186
    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
187

Khalique's avatar
Khalique committed
188
    auto prog = migraphx::parse_onnx("const_of_shape_empty_input_test.onnx");
Shucai Xiao's avatar
Shucai Xiao committed
189
    EXPECT(p == prog);
Khalique's avatar
Khalique committed
190
191
}

Khalique's avatar
Khalique committed
192
TEST_CASE(const_of_shape_float_test)
193
194
{
    migraphx::program p;
Khalique's avatar
Khalique committed
195
196
197
198
199
    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));
200

Khalique's avatar
Khalique committed
201
    auto prog = migraphx::parse_onnx("const_of_shape_float_test.onnx");
202
203
204
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
205
TEST_CASE(const_of_shape_int64_test)
206
207
{
    migraphx::program p;
Khalique's avatar
Khalique committed
208
209
210
211
212
    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));
213

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

Khalique's avatar
Khalique committed
218
TEST_CASE(const_of_shape_no_value_attr_test)
219
220
{
    migraphx::program p;
Khalique's avatar
Khalique committed
221
222
223
224
225
    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));
226

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

Khalique's avatar
Khalique committed
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
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
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
312
TEST_CASE(cos_test)
313
314
315
{
    migraphx::program p;
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10}});
Khalique's avatar
Khalique committed
316
    p.add_instruction(migraphx::op::cos{}, input);
317

Khalique's avatar
Khalique committed
318
    auto prog = migraphx::parse_onnx("cos_test.onnx");
319
320
321
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
322
TEST_CASE(cosh_test)
323
324
325
326
327
328
329
330
331
332
{
    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
333
TEST_CASE(dropout_test)
Shucai Xiao's avatar
Shucai Xiao committed
334
335
{
    migraphx::program p;
Khalique's avatar
Khalique committed
336
337
    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
338

Khalique's avatar
Khalique committed
339
    auto prog = migraphx::parse_onnx("dropout_test.onnx");
Shucai Xiao's avatar
Shucai Xiao committed
340
341
342
343

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
344
345
346
347
348
349
350
351
352
353
354
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
355
TEST_CASE(erf_test)
356
357
{
    migraphx::program p;
Khalique's avatar
Khalique committed
358
359
    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
360

Khalique's avatar
Khalique committed
361
    auto prog = migraphx::parse_onnx("erf_test.onnx");
Khalique's avatar
Khalique committed
362
363
364
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
365
TEST_CASE(exp_test)
366
367
368
{
    migraphx::program p;
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10}});
Khalique's avatar
Khalique committed
369
    p.add_instruction(migraphx::op::exp{}, input);
Khalique's avatar
Khalique committed
370

Khalique's avatar
Khalique committed
371
    auto prog = migraphx::parse_onnx("exp_test.onnx");
Khalique's avatar
Khalique committed
372
    EXPECT(p == prog);
Khalique's avatar
Khalique committed
373
374
}

Khalique's avatar
Khalique committed
375
TEST_CASE(expand_test)
376
377
{
    migraphx::program p;
Khalique's avatar
Khalique committed
378
379
380
381
382
    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
383

Khalique's avatar
Khalique committed
384
    auto prog = migraphx::parse_onnx("expand_test.onnx");
Khalique's avatar
Khalique committed
385
    EXPECT(p == prog);
Khalique's avatar
Khalique committed
386
387
}

Khalique's avatar
Khalique committed
388
TEST_CASE(flatten_test)
389
390
391
{
    migraphx::program p;
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {2, 3, 4, 5}});
Khalique's avatar
Khalique committed
392
393
394
    p.add_instruction(migraphx::op::flatten{2}, l0);
    p.add_instruction(migraphx::op::flatten{1}, l0);
    auto prog = migraphx::parse_onnx("flatten_test.onnx");
395
396
397
398

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
399
TEST_CASE(gather_test)
400
401
{
    migraphx::program p;
Khalique's avatar
Khalique committed
402
403
404
405
406
    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");
407
408
409
410

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
411
TEST_CASE(gemm_test)
412
413
{
    migraphx::program p;
Khalique's avatar
Khalique committed
414
415
416
417
418
419
420
421
422
    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");
423
424
425
426

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
427
TEST_CASE(gemm_ex_test)
428
429
{
    migraphx::program p;
Khalique's avatar
Khalique committed
430
431
432
433
434
435
436
437
    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");
438
439
440
441

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
442
TEST_CASE(gemm_ex_brcst_test)
443
444
{
    migraphx::program p;
Khalique's avatar
Khalique committed
445
446
447
448
449
450
451
452
453
    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
454
    auto prog = migraphx::parse_onnx("gemm_ex_brcst_test.onnx");
455
456
457
458

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
459
TEST_CASE(globalavgpool_test)
460
461
{
    migraphx::program p;
Khalique's avatar
Khalique committed
462
463
464
465
466
467
468
    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");
469
470
471
472

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
473
TEST_CASE(globalmaxpool_test)
Khalique's avatar
Khalique committed
474
475
{
    migraphx::program p;
Khalique's avatar
Khalique committed
476
477
478
479
480
481
482
    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
483
484
485
486

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
487
TEST_CASE(group_conv_test)
488
489
{
    migraphx::program p;
Khalique's avatar
Khalique committed
490
491
492
493
494
495
    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");
496
497
498
499

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
500
TEST_CASE(imagescaler_test)
501
502
{
    migraphx::program p;
Khalique's avatar
Khalique committed
503
504
505
506
507
508
509
510
511
512
513
    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");
514
515
516
517

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
518
TEST_CASE(implicit_add_bcast_test)
519
520
{
    migraphx::program p;
Khalique's avatar
Khalique committed
521
522
523
524
525
526
527
    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");
528
529
530
531

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
532
TEST_CASE(implicit_pow_bcast_test)
533
534
{
    migraphx::program p;
Khalique's avatar
Khalique committed
535
536
537
538
539
540
541
    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");
542
543
544
545

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
546
TEST_CASE(implicit_sub_bcast_test)
547
548
549
{
    migraphx::program p;
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {2, 3, 4, 5}});
Khalique's avatar
Khalique committed
550
551
552
553
554
555
    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");
556
557
558
559

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
560
TEST_CASE(leaky_relu_test)
561
562
{
    migraphx::program p;
Khalique's avatar
Khalique committed
563
564
565
566
567
    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");
568
569
570
571

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
572
TEST_CASE(log_test)
573
574
{
    migraphx::program p;
Khalique's avatar
Khalique committed
575
576
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10}});
    p.add_instruction(migraphx::op::log{}, input);
577

Khalique's avatar
Khalique committed
578
    auto prog = migraphx::parse_onnx("log_test.onnx");
579
580
581
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
582
TEST_CASE(logsoftmax_test)
583
584
{
    migraphx::program p;
Khalique's avatar
Khalique committed
585
586
587
588
    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");
589
590
591
592

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
593
TEST_CASE(lrn_test)
594
595
{
    migraphx::program p;
Khalique's avatar
Khalique committed
596
597
598
599
600
601
602
603
    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");
604
605
606
607

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
608
TEST_CASE(matmul_bmbm_test)
609
610
{
    migraphx::program p;
Khalique's avatar
Khalique committed
611
612
613
614
615
616
617
    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");
618
619
620
621

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
622
TEST_CASE(matmul_bmv_test)
623
{
Khalique's avatar
Khalique committed
624
625
626
627
628
629
630
    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);
631

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

Khalique's avatar
Khalique committed
634
    EXPECT(p == prog);
635
636
}

Khalique's avatar
Khalique committed
637
TEST_CASE(matmul_mv_test)
638
639
{
    migraphx::program p;
Khalique's avatar
Khalique committed
640
641
642
643
644
645
646
    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");
647
648
649
650

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
651
TEST_CASE(matmul_vbm_test)
652
653
{
    migraphx::program p;
Khalique's avatar
Khalique committed
654
655
656
657
658
659
660
661
662
663
    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");
664
665
666
667

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
668
TEST_CASE(matmul_vm_test)
669
670
{
    migraphx::program p;
Khalique's avatar
Khalique committed
671
672
673
674
675
676
677
    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");
678
679
680
681

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
682
TEST_CASE(matmul_vv_test)
683
684
{
    migraphx::program p;
Shucai Xiao's avatar
Shucai Xiao committed
685
686
    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}});
687
688
689
690
691
692
    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
693
    auto prog = migraphx::parse_onnx("matmul_vv_test.onnx");
694
695
696
697

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
698
TEST_CASE(max_test)
699
700
{
    migraphx::program p;
Khalique's avatar
Khalique committed
701
702
703
704
705
    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);
706

Khalique's avatar
Khalique committed
707
708
    migraphx::parse_onnx("max_test.onnx");
}
709

Khalique's avatar
Khalique committed
710
711
712
713
714
715
716
717
718
719
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");
720
721
}

Khalique's avatar
Khalique committed
722
TEST_CASE(no_pad_test)
723
724
{
    migraphx::program p;
Khalique's avatar
Khalique committed
725
726
727
    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");
728

Khalique's avatar
Khalique committed
729
730
731
732
733
734
735
736
737
    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");
738
739
740
741

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
742
TEST_CASE(pow_test)
743
744
{
    migraphx::program p;
Khalique's avatar
Khalique committed
745
746
747
    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);
748

Khalique's avatar
Khalique committed
749
    auto prog = migraphx::parse_onnx("pow_test.onnx");
750
751
752
753

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
754
TEST_CASE(reducemean_test)
755
756
{
    migraphx::program p;
Khalique's avatar
Khalique committed
757
758
759
760
    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");
761
762
763
764

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
765
TEST_CASE(reducemean_keepdims_test)
766
767
{
    migraphx::program p;
Khalique's avatar
Khalique committed
768
769
770
    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");
771
772
773
774

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
775
TEST_CASE(reducesum_test)
Khalique's avatar
Khalique committed
776
777
{
    migraphx::program p;
Khalique's avatar
Khalique committed
778
779
780
    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
781
    auto prog = migraphx::parse_onnx("reducesum_test.onnx");
782
783

    EXPECT(p == prog);
Khalique's avatar
Khalique committed
784
785
}

Khalique's avatar
Khalique committed
786
TEST_CASE(reducesum_multiaxis_test)
Khalique's avatar
Khalique committed
787
788
{
    migraphx::program p;
Khalique's avatar
Khalique committed
789
790
791
    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
792
    auto prog = migraphx::parse_onnx("reducesum_multiaxis_test.onnx");
793
794

    EXPECT(p == prog);
Khalique's avatar
Khalique committed
795
796
}

Khalique's avatar
Khalique committed
797
TEST_CASE(reducesum_keepdims_test)
Khalique's avatar
Khalique committed
798
799
{
    migraphx::program p;
Khalique's avatar
Khalique committed
800
801
    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
802
    auto prog = migraphx::parse_onnx("reducesum_keepdims_test.onnx");
803
804
805
806

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
807
TEST_CASE(reshape_test)
808
{
Khalique's avatar
Khalique committed
809
    migraphx::program p;
Khalique's avatar
Khalique committed
810
811
812
813
814
815
816
817
818
    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");
819
820

    EXPECT(p == prog);
Khalique's avatar
Khalique committed
821
822
}

Khalique's avatar
Khalique committed
823
TEST_CASE(reshape_non_standard_test)
824
825
{
    migraphx::program p;
Khalique's avatar
Khalique committed
826
827
828
829
830
831
832
    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
833
    auto prog = migraphx::parse_onnx("reshape_non_standard_test.onnx");
834
835
836
837

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
838
TEST_CASE(shape_test)
839
840
{
    migraphx::program p;
Khalique's avatar
Khalique committed
841
842
843
844
845
    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");
846
847
848
849

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
850
TEST_CASE(shape_gather_test)
851
852
{
    migraphx::program p;
Khalique's avatar
Khalique committed
853
854
855
856
857
858
859
    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
860
    auto prog = migraphx::parse_onnx("shape_gather_test.onnx");
861
862
863
864

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
865
TEST_CASE(sign_test)
Khalique's avatar
Khalique committed
866
867
{
    migraphx::program p;
Khalique's avatar
Khalique committed
868
869
    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
870

Khalique's avatar
Khalique committed
871
    auto prog = migraphx::parse_onnx("sign_test.onnx");
Khalique's avatar
Khalique committed
872
873
874
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
875
TEST_CASE(sin_test)
876
877
{
    migraphx::program p;
Khalique's avatar
Khalique committed
878
879
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10}});
    p.add_instruction(migraphx::op::sin{}, input);
880

Khalique's avatar
Khalique committed
881
    auto prog = migraphx::parse_onnx("sin_test.onnx");
882
883
884
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
885
TEST_CASE(sinh_test)
886
887
{
    migraphx::program p;
Khalique's avatar
Khalique committed
888
889
890
891
    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");
892
893
894
895

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
896
TEST_CASE(slice_test)
897
898
{
    migraphx::program p;
Khalique's avatar
Khalique committed
899
900
901
    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");
902
903
904
905

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
906
TEST_CASE(softmax_test)
Shucai Xiao's avatar
Shucai Xiao committed
907
908
{
    migraphx::program p;
Khalique's avatar
Khalique committed
909
910
911
    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
912
913
914
915

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
916
TEST_CASE(sqrt_test)
Shucai Xiao's avatar
Shucai Xiao committed
917
918
{
    migraphx::program p;
Khalique's avatar
Khalique committed
919
920
    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
921

Khalique's avatar
Khalique committed
922
    auto prog = migraphx::parse_onnx("sqrt_test.onnx");
Shucai Xiao's avatar
Shucai Xiao committed
923
924
925
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
926
TEST_CASE(squeeze_unsqueeze_test)
Khalique's avatar
Khalique committed
927
928
{
    migraphx::program p;
Khalique's avatar
Khalique committed
929
930
931
932
933
934
935
    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
936
937
938
939

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
940
TEST_CASE(sub_bcast_test)
Shucai Xiao's avatar
Shucai Xiao committed
941
942
943
{
    migraphx::program p;
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {2, 3, 4, 5}});
Khalique's avatar
Khalique committed
944
945
946
    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
947

Khalique's avatar
Khalique committed
948
    auto prog = migraphx::parse_onnx("sub_bcast_test.onnx");
Shucai Xiao's avatar
Shucai Xiao committed
949
950
951
952

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
953
TEST_CASE(sub_scalar_test)
Shucai Xiao's avatar
Shucai Xiao committed
954
955
956
{
    migraphx::program p;
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {2, 3, 4, 5}});
Khalique's avatar
Khalique committed
957
958
959
960
961
962
    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
963
964
965
966

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
967
TEST_CASE(sum_test)
968
969
{
    migraphx::program p;
Khalique's avatar
Khalique committed
970
971
972
973
974
    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);
975

Khalique's avatar
Khalique committed
976
    auto prog = migraphx::parse_onnx("sum_test.onnx");
977
978
979
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
980
TEST_CASE(tan_test)
981
982
{
    migraphx::program p;
Khalique's avatar
Khalique committed
983
984
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10}});
    p.add_instruction(migraphx::op::tan{}, input);
985

Khalique's avatar
Khalique committed
986
    auto prog = migraphx::parse_onnx("tan_test.onnx");
987
988
989
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
990
TEST_CASE(tanh_test)
991
992
{
    migraphx::program p;
Khalique's avatar
Khalique committed
993
994
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {1}});
    p.add_instruction(migraphx::op::tanh{}, input);
995

Khalique's avatar
Khalique committed
996
    auto prog = migraphx::parse_onnx("tanh_test.onnx");
997
998
999
1000

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
1001
TEST_CASE(transpose_test)
1002
1003
{
    migraphx::program p;
Khalique's avatar
Khalique committed
1004
1005
1006
1007
1008
    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");
1009
1010
1011
1012

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
1013
TEST_CASE(unknown_test)
1014
1015
{
    migraphx::program p;
Khalique's avatar
Khalique committed
1016
1017
1018
1019
1020
    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");
1021
1022
1023
1024

    EXPECT(p == prog);
}

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