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
TEST_CASE(cos_test)
232
233
234
{
    migraphx::program p;
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10}});
Khalique's avatar
Khalique committed
235
    p.add_instruction(migraphx::op::cos{}, input);
236

Khalique's avatar
Khalique committed
237
    auto prog = migraphx::parse_onnx("cos_test.onnx");
238
239
240
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
241
TEST_CASE(cosh_test)
242
243
244
245
246
247
248
249
250
251
{
    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
252
TEST_CASE(dropout_test)
Shucai Xiao's avatar
Shucai Xiao committed
253
254
{
    migraphx::program p;
Khalique's avatar
Khalique committed
255
256
    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
257

Khalique's avatar
Khalique committed
258
    auto prog = migraphx::parse_onnx("dropout_test.onnx");
Shucai Xiao's avatar
Shucai Xiao committed
259
260
261
262

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
263
264
265
266
267
268
269
270
271
272
273
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
274
TEST_CASE(erf_test)
275
276
{
    migraphx::program p;
Khalique's avatar
Khalique committed
277
278
    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
279

Khalique's avatar
Khalique committed
280
    auto prog = migraphx::parse_onnx("erf_test.onnx");
Khalique's avatar
Khalique committed
281
282
283
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
284
TEST_CASE(exp_test)
285
286
287
{
    migraphx::program p;
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10}});
Khalique's avatar
Khalique committed
288
    p.add_instruction(migraphx::op::exp{}, input);
Khalique's avatar
Khalique committed
289

Khalique's avatar
Khalique committed
290
    auto prog = migraphx::parse_onnx("exp_test.onnx");
Khalique's avatar
Khalique committed
291
    EXPECT(p == prog);
Khalique's avatar
Khalique committed
292
293
}

Khalique's avatar
Khalique committed
294
TEST_CASE(expand_test)
295
296
{
    migraphx::program p;
Khalique's avatar
Khalique committed
297
298
299
300
301
    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
302

Khalique's avatar
Khalique committed
303
    auto prog = migraphx::parse_onnx("expand_test.onnx");
Khalique's avatar
Khalique committed
304
    EXPECT(p == prog);
Khalique's avatar
Khalique committed
305
306
}

Khalique's avatar
Khalique committed
307
TEST_CASE(flatten_test)
308
309
310
{
    migraphx::program p;
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {2, 3, 4, 5}});
Khalique's avatar
Khalique committed
311
312
313
    p.add_instruction(migraphx::op::flatten{2}, l0);
    p.add_instruction(migraphx::op::flatten{1}, l0);
    auto prog = migraphx::parse_onnx("flatten_test.onnx");
314
315
316
317

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
318
TEST_CASE(gather_test)
319
320
{
    migraphx::program p;
Khalique's avatar
Khalique committed
321
322
323
324
325
    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");
326
327
328
329

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
330
TEST_CASE(gemm_test)
331
332
{
    migraphx::program p;
Khalique's avatar
Khalique committed
333
334
335
336
337
338
339
340
341
    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");
342
343
344
345

    EXPECT(p == prog);
}

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

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
361
TEST_CASE(gemm_ex_brcst_test)
362
363
{
    migraphx::program p;
Khalique's avatar
Khalique committed
364
365
366
367
368
369
370
371
372
    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
373
    auto prog = migraphx::parse_onnx("gemm_ex_brcst_test.onnx");
374
375
376
377

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
378
TEST_CASE(globalavgpool_test)
379
380
{
    migraphx::program p;
Khalique's avatar
Khalique committed
381
382
383
384
385
386
387
    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");
388
389
390
391

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
392
TEST_CASE(globalmaxpool_test)
Khalique's avatar
Khalique committed
393
394
{
    migraphx::program p;
Khalique's avatar
Khalique committed
395
396
397
398
399
400
401
    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
402
403
404
405

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
406
TEST_CASE(group_conv_test)
407
408
{
    migraphx::program p;
Khalique's avatar
Khalique committed
409
410
411
412
413
414
    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");
415
416
417
418

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
419
TEST_CASE(imagescaler_test)
420
421
{
    migraphx::program p;
Khalique's avatar
Khalique committed
422
423
424
425
426
427
428
429
430
431
432
    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");
433
434
435
436

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
437
TEST_CASE(implicit_add_bcast_test)
438
439
{
    migraphx::program p;
Khalique's avatar
Khalique committed
440
441
442
443
444
445
446
    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");
447
448
449
450

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
451
TEST_CASE(implicit_pow_bcast_test)
452
453
{
    migraphx::program p;
Khalique's avatar
Khalique committed
454
455
456
457
458
459
460
    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");
461
462
463
464

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
465
TEST_CASE(implicit_sub_bcast_test)
466
467
468
{
    migraphx::program p;
    auto l0 = p.add_parameter("0", migraphx::shape{migraphx::shape::float_type, {2, 3, 4, 5}});
Khalique's avatar
Khalique committed
469
470
471
472
473
474
    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");
475
476
477
478

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
479
TEST_CASE(leaky_relu_test)
480
481
{
    migraphx::program p;
Khalique's avatar
Khalique committed
482
483
484
485
486
    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");
487
488
489
490

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
491
TEST_CASE(log_test)
492
493
{
    migraphx::program p;
Khalique's avatar
Khalique committed
494
495
    auto input = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {10}});
    p.add_instruction(migraphx::op::log{}, input);
496

Khalique's avatar
Khalique committed
497
    auto prog = migraphx::parse_onnx("log_test.onnx");
498
499
500
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
501
TEST_CASE(logsoftmax_test)
502
503
{
    migraphx::program p;
Khalique's avatar
Khalique committed
504
505
506
507
    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");
508
509
510
511

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
512
TEST_CASE(lrn_test)
513
514
{
    migraphx::program p;
Khalique's avatar
Khalique committed
515
516
517
518
519
520
521
522
    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");
523
524
525
526

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
527
TEST_CASE(matmul_bmbm_test)
528
529
{
    migraphx::program p;
Khalique's avatar
Khalique committed
530
531
532
533
534
535
536
    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");
537
538
539
540

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
541
TEST_CASE(matmul_bmv_test)
542
{
Khalique's avatar
Khalique committed
543
544
545
546
547
548
549
    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);
550

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

Khalique's avatar
Khalique committed
553
    EXPECT(p == prog);
554
555
}

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

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
570
TEST_CASE(matmul_vbm_test)
571
572
{
    migraphx::program p;
Khalique's avatar
Khalique committed
573
574
575
576
577
578
579
580
581
582
    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");
583
584
585
586

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
587
TEST_CASE(matmul_vm_test)
588
589
{
    migraphx::program p;
Khalique's avatar
Khalique committed
590
591
592
593
594
595
596
    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");
597
598
599
600

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
601
TEST_CASE(matmul_vv_test)
602
603
{
    migraphx::program p;
Shucai Xiao's avatar
Shucai Xiao committed
604
605
    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}});
606
607
608
609
610
611
    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
612
    auto prog = migraphx::parse_onnx("matmul_vv_test.onnx");
613
614
615
616

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
617
TEST_CASE(max_test)
618
619
{
    migraphx::program p;
Khalique's avatar
Khalique committed
620
621
622
623
624
    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);
625

Khalique's avatar
Khalique committed
626
627
    migraphx::parse_onnx("max_test.onnx");
}
628

Khalique's avatar
Khalique committed
629
630
631
632
633
634
635
636
637
638
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");
639
640
}

Khalique's avatar
Khalique committed
641
TEST_CASE(no_pad_test)
642
643
{
    migraphx::program p;
Khalique's avatar
Khalique committed
644
645
646
    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");
647

Khalique's avatar
Khalique committed
648
649
650
651
652
653
654
655
656
    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");
657
658
659
660

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
661
TEST_CASE(pow_test)
662
663
{
    migraphx::program p;
Khalique's avatar
Khalique committed
664
665
666
    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);
667

Khalique's avatar
Khalique committed
668
    auto prog = migraphx::parse_onnx("pow_test.onnx");
669
670
671
672

    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
673
TEST_CASE(pytorch_conv_bias_test)
674
675
{
    migraphx::program p;
Khalique's avatar
Khalique committed
676
677
678
679
680
681
682
    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);
683

Khalique's avatar
Khalique committed
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
    auto prog = migraphx::parse_onnx("conv.onnx");
    EXPECT(p == prog);
}

TEST_CASE(pytorch_conv_bn_relu_maxpool)
{
    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);
706

Khalique's avatar
Khalique committed
707
    auto prog = migraphx::parse_onnx("conv_bn_relu_maxpool.onnx");
708
709
710
    EXPECT(p == prog);
}

Khalique's avatar
Khalique committed
711
TEST_CASE(pytorch_conv_relu_maxpool)
712
713
{
    migraphx::program p;
Khalique's avatar
Khalique committed
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
    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.onnx");
    EXPECT(p == prog);
}

TEST_CASE(pytorch_conv_relu_maxpool_x2)
{
    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);
748

Khalique's avatar
Khalique committed
749
    auto prog = migraphx::parse_onnx("conv_relu_maxpoolX2.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); }