ops_test.cpp 90.9 KB
Newer Older
Paul's avatar
Paul committed
1
#include <test.hpp>
2
3
#include <migraphx/quantization.hpp>
#include "test_utils.hpp"
Paul's avatar
Paul committed
4

Paul's avatar
Paul committed
5
6
7
8
#ifdef __clang__
#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Wglobal-constructors"
#endif
Paul's avatar
Paul committed
9

Paul's avatar
Paul committed
10
struct test_literals : verify_program<test_literals>
Paul's avatar
Paul committed
11
{
Paul's avatar
Paul committed
12
    migraphx::program create_program() const
Paul's avatar
Paul committed
13
    {
Paul's avatar
Paul committed
14
        migraphx::program p;
Paul's avatar
Paul committed
15
        auto input = p.add_literal(
Paul's avatar
Paul committed
16
            generate_literal(migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}}));
Paul's avatar
Paul committed
17
        auto weights = p.add_literal(
Paul's avatar
Paul committed
18
19
20
            generate_literal(migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}}));
        auto conv = p.add_instruction(migraphx::op::convolution{}, input, weights);
        p.add_instruction(migraphx::op::relu{}, conv);
Paul's avatar
Paul committed
21
22
23
24
        return p;
    }
};

Paul's avatar
Paul committed
25
struct test_add : verify_program<test_add>
Paul's avatar
Paul committed
26
{
Paul's avatar
Paul committed
27
    migraphx::program create_program() const
Paul's avatar
Paul committed
28
    {
Paul's avatar
Paul committed
29
30
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {3}};
Paul's avatar
Paul committed
31
32
        auto x = p.add_parameter("x", s);
        auto y = p.add_parameter("y", s);
Paul's avatar
Paul committed
33
        p.add_instruction(migraphx::op::add{}, x, y);
Paul's avatar
Paul committed
34
35
36
37
        return p;
    }
};

Paul's avatar
Paul committed
38
struct test_add_half : verify_program<test_add_half>
Paul's avatar
Paul committed
39
{
Paul's avatar
Paul committed
40
    migraphx::program create_program() const
Paul's avatar
Paul committed
41
    {
Paul's avatar
Paul committed
42
43
        migraphx::program p;
        migraphx::shape s{migraphx::shape::half_type, {3}};
Paul's avatar
Paul committed
44
45
        auto x = p.add_parameter("x", s);
        auto y = p.add_parameter("y", s);
Paul's avatar
Paul committed
46
        p.add_instruction(migraphx::op::add{}, x, y);
Paul's avatar
Paul committed
47
48
49
50
        return p;
    }
};

Paul's avatar
Paul committed
51
struct test_mul : verify_program<test_mul>
Khalique's avatar
Khalique committed
52
{
Paul's avatar
Paul committed
53
    migraphx::program create_program() const
Khalique's avatar
Khalique committed
54
    {
Paul's avatar
Paul committed
55
56
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {3}};
Khalique's avatar
Khalique committed
57
58
        auto x = p.add_parameter("x", s);
        auto y = p.add_parameter("y", s);
Paul's avatar
Paul committed
59
        p.add_instruction(migraphx::op::mul{}, x, y);
Khalique's avatar
Khalique committed
60
61
62
63
        return p;
    }
};

Paul's avatar
Paul committed
64
struct test_exp : verify_program<test_exp>
Shucai Xiao's avatar
Shucai Xiao committed
65
66
67
68
69
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {6}};
70
        auto x = p.add_instruction(migraphx::op::abs{}, p.add_parameter("x", s));
Shucai Xiao's avatar
Shucai Xiao committed
71
72
73
74
75
        p.add_instruction(migraphx::op::exp{}, x);
        return p;
    }
};

Shucai Xiao's avatar
Shucai Xiao committed
76
77
78
79
80
81
82
83
84
85
86
87
struct test_erf : verify_program<test_erf>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {2, 3, 4, 6}};
        auto param = p.add_parameter("x", s);
        p.add_instruction(migraphx::op::erf{}, param);
        return p;
    }
};

Shucai Xiao's avatar
Shucai Xiao committed
88
89
90
91
92
93
struct test_sqrt : verify_program<test_sqrt>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {2, 3, 4, 6}};
Shucai Xiao's avatar
Shucai Xiao committed
94
        auto param     = p.add_parameter("x", s);
Shucai Xiao's avatar
Shucai Xiao committed
95
96
97
98
99
100
        auto param_abs = p.add_instruction(migraphx::op::abs{}, param);
        p.add_instruction(migraphx::op::sqrt{}, param_abs);
        return p;
    }
};

101
102
103
104
105
106
struct test_sign : verify_program<test_sign>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::double_type, {2, 3, 4, 6}};
Shucai Xiao's avatar
Shucai Xiao committed
107
        auto param = p.add_parameter("x", s);
108
109
110
111
112
        p.add_instruction(migraphx::op::sign{}, param);
        return p;
    }
};

Paul's avatar
Paul committed
113
struct test_log : verify_program<test_log>
Shucai Xiao's avatar
Shucai Xiao committed
114
115
116
117
118
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {6}};
119
        auto x = p.add_instruction(migraphx::op::abs{}, p.add_parameter("x", s));
Shucai Xiao's avatar
Shucai Xiao committed
120
121
122
123
124
        p.add_instruction(migraphx::op::log{}, x);
        return p;
    }
};

Shucai Xiao's avatar
Shucai Xiao committed
125
126
127
128
129
130
131
132
133
134
135
136
137
138
struct test_pow : verify_program<test_pow>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {6}};
        std::vector<float> vec_e(s.elements(), 2.0f);
        auto b = p.add_parameter("x", s);
        auto e = p.add_literal(migraphx::literal(s, vec_e));
        p.add_instruction(migraphx::op::pow{}, b, e);
        return p;
    }
};

Shucai Xiao's avatar
Shucai Xiao committed
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
struct test_prelu_brcst : verify_program<test_prelu_brcst>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {6}};
        auto x   = p.add_parameter("x", s);
        auto slp = p.add_parameter("slp", s);
        auto r   = p.add_instruction(migraphx::op::prelu{}, x, slp);
        p.add_return({r});

        return p;
    }
};

Paul's avatar
Paul committed
154
struct test_sin : verify_program<test_sin>
155
156
157
158
159
160
161
162
163
164
165
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {10}};
        auto x = p.add_parameter("x", s);
        p.add_instruction(migraphx::op::sin{}, x);
        return p;
    }
};

Paul's avatar
Paul committed
166
struct test_cos : verify_program<test_cos>
Shucai Xiao's avatar
Shucai Xiao committed
167
168
169
170
171
172
173
174
175
176
177
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::double_type, {8}};
        auto x = p.add_parameter("x", s);
        p.add_instruction(migraphx::op::cos{}, x);
        return p;
    }
};

Paul's avatar
Paul committed
178
struct test_tan : verify_program<test_tan>
Shucai Xiao's avatar
Shucai Xiao committed
179
180
181
182
183
184
185
186
187
188
189
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {16}};
        auto x = p.add_parameter("x", s);
        p.add_instruction(migraphx::op::tan{}, x);
        return p;
    }
};

Paul's avatar
Paul committed
190
struct test_sinh : verify_program<test_sinh>
191
192
193
194
195
196
197
198
199
200
201
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::double_type, {16}};
        auto x = p.add_parameter("x", s);
        p.add_instruction(migraphx::op::sinh{}, x);
        return p;
    }
};

Paul's avatar
Paul committed
202
struct test_cosh : verify_program<test_cosh>
203
204
205
206
207
208
209
210
211
212
213
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::double_type, {16}};
        auto x = p.add_parameter("x", s);
        p.add_instruction(migraphx::op::cosh{}, x);
        return p;
    }
};

Paul's avatar
Paul committed
214
struct test_tanh : verify_program<test_tanh>
215
216
217
218
219
220
221
222
223
224
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        auto x = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        p.add_instruction(migraphx::op::tanh{}, x);
        return p;
    }
};

225
226
227
228
229
struct test_trans_tanh : verify_program<test_trans_tanh>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
Shucai Xiao's avatar
Shucai Xiao committed
230
        auto x  = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
231
232
        auto tx = p.add_instruction(migraphx::op::transpose{{0, 1, 3, 2}}, x);
        auto tanhx = p.add_instruction(migraphx::op::tanh{}, tx);
Shucai Xiao's avatar
Shucai Xiao committed
233
        auto r     = p.add_instruction(migraphx::op::add{}, tanhx, tanhx);
234
235
        p.add_instruction(migraphx::op::contiguous{}, r);

236
237
238
239
        return p;
    }
};

240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
struct test_trans_tanh1 : verify_program<test_trans_tanh1>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        auto x  = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        auto tx = p.add_instruction(migraphx::op::transpose{{0, 1, 3, 2}}, x);
        auto tanhx = p.add_instruction(migraphx::op::tanh{}, tx);
        auto r     = p.add_instruction(migraphx::op::add{}, tanhx, tanhx);
        p.add_return({tx, r});

        return p;
    }
};

255
256
257
258
259
struct test_slice_sin : verify_program<test_slice_sin>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
Shucai Xiao's avatar
Shucai Xiao committed
260
261
        auto l = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {2, 2}});
        auto t = p.add_instruction(migraphx::op::slice{{1}, {1}, {2}}, l);
262
263
        p.add_instruction(migraphx::op::sin{}, t);

264
265
266
267
        return p;
    }
};

Paul's avatar
Paul committed
268
struct test_asin : verify_program<test_asin>
269
270
271
272
273
274
275
276
277
278
279
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::double_type, {16}};
        auto x = p.add_parameter("x", s);
        p.add_instruction(migraphx::op::asin{}, x);
        return p;
    }
};

Paul's avatar
Paul committed
280
struct test_acos : verify_program<test_acos>
281
282
283
284
285
286
287
288
289
290
291
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::double_type, {16}};
        auto x = p.add_parameter("x", s);
        p.add_instruction(migraphx::op::acos{}, x);
        return p;
    }
};

Paul's avatar
Paul committed
292
struct test_atan : verify_program<test_atan>
293
294
295
296
297
298
299
300
301
302
303
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::double_type, {16}};
        auto x = p.add_parameter("x", s);
        p.add_instruction(migraphx::op::atan{}, x);
        return p;
    }
};

304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
struct test_asinh : verify_program<test_asinh>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::double_type, {16}};
        auto x = p.add_parameter("x", s);
        p.add_instruction(migraphx::op::asinh{}, x);
        return p;
    }
};

struct test_acosh : verify_program<test_acosh>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {16}};
kahmed10's avatar
kahmed10 committed
322
323
324
325
326
327
        auto x       = p.add_parameter("x", s);
        auto min_val = p.add_literal(1.1f);
        auto max_val = p.add_literal(100.0f);
        min_val      = p.add_instruction(migraphx::op::multibroadcast{{16}}, min_val);
        max_val      = p.add_instruction(migraphx::op::multibroadcast{{16}}, max_val);
        auto cx      = p.add_instruction(migraphx::op::clip{}, x, min_val, max_val);
328
329
330
331
332
333
334
335
336
337
338
        p.add_instruction(migraphx::op::acosh{}, cx);
        return p;
    }
};

struct test_atanh : verify_program<test_atanh>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::double_type, {16}};
kahmed10's avatar
kahmed10 committed
339
340
341
342
343
344
        auto x       = p.add_parameter("x", s);
        auto min_val = p.add_literal(-0.95);
        auto max_val = p.add_literal(0.95);
        min_val      = p.add_instruction(migraphx::op::multibroadcast{{16}}, min_val);
        max_val      = p.add_instruction(migraphx::op::multibroadcast{{16}}, max_val);
        auto cx      = p.add_instruction(migraphx::op::clip{}, x, min_val, max_val);
345
346
347
348
349
        p.add_instruction(migraphx::op::atanh{}, cx);
        return p;
    }
};

Paul's avatar
Paul committed
350
struct test_scale : verify_program<test_scale>
Khalique's avatar
Khalique committed
351
{
Paul's avatar
Paul committed
352
    migraphx::program create_program() const
Khalique's avatar
Khalique committed
353
    {
Paul's avatar
Paul committed
354
355
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {3}};
Khalique's avatar
Khalique committed
356
        auto x     = p.add_parameter("x", s);
Paul's avatar
Paul committed
357
        auto y     = p.add_parameter("y", migraphx::shape::float_type);
358
        auto scale = p.add_instruction(migraphx::op::scalar{s.lens()}, y);
Paul's avatar
Paul committed
359
        p.add_instruction(migraphx::op::mul{}, x, scale);
Khalique's avatar
Khalique committed
360
361
362
363
        return p;
    }
};

Paul's avatar
Paul committed
364
struct test_slice : verify_program<test_slice>
365
{
Paul's avatar
Paul committed
366
    migraphx::program create_program() const
367
    {
Paul's avatar
Paul committed
368
369
        migraphx::program p;
        migraphx::shape s{migraphx::shape::int32_type, {2, 2, 4}};
370
        auto x      = p.add_parameter("x", s);
Paul's avatar
Paul committed
371
372
373
        auto y      = p.add_parameter("y", {migraphx::shape::int32_type, {2, 2, 2}});
        auto slice0 = p.add_instruction(migraphx::op::slice{{2}, {0}, {2}}, x);
        p.add_instruction(migraphx::op::add{}, y, slice0);
374
375
376
377
378

        return p;
    }
};

Paul's avatar
Paul committed
379
struct test_triadd : verify_program<test_triadd>
Paul's avatar
Paul committed
380
{
Paul's avatar
Paul committed
381
    migraphx::program create_program() const
Paul's avatar
Paul committed
382
    {
Paul's avatar
Paul committed
383
384
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {3}};
Paul's avatar
Paul committed
385
386
387
        auto x   = p.add_parameter("x", s);
        auto y   = p.add_parameter("y", s);
        auto z   = p.add_parameter("z", s);
Paul's avatar
Paul committed
388
389
        auto sum = p.add_instruction(migraphx::op::add{}, x, y);
        p.add_instruction(migraphx::op::add{}, sum, z);
Paul's avatar
Paul committed
390
391
392
393
        return p;
    }
};

Paul's avatar
Paul committed
394
struct test_triadd2 : verify_program<test_triadd2>
Paul's avatar
Paul committed
395
{
Paul's avatar
Paul committed
396
    migraphx::program create_program() const
Paul's avatar
Paul committed
397
    {
Paul's avatar
Paul committed
398
399
400
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {2, 3}};
        migraphx::shape b{migraphx::shape::float_type, {3}};
Paul's avatar
Paul committed
401
402
403
        auto x   = p.add_parameter("x", s);
        auto y   = p.add_parameter("y", s);
        auto z   = p.add_parameter("z", b);
404
        auto zb  = p.add_instruction(migraphx::op::broadcast{1, s.lens()}, z);
Paul's avatar
Paul committed
405
406
        auto sum = p.add_instruction(migraphx::op::add{}, x, y);
        p.add_instruction(migraphx::op::add{}, sum, zb);
Paul's avatar
Paul committed
407
408
409
410
        return p;
    }
};

Paul's avatar
Paul committed
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
struct test_mul_add : verify_program<test_mul_add>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {2, 3}};
        migraphx::shape bs{migraphx::shape::float_type, {3}};
        auto x   = p.add_parameter("x", s);
        auto a   = p.add_parameter("a", bs);
        auto b   = p.add_parameter("b", bs);
        auto ab  = p.add_instruction(migraphx::op::broadcast{1, s.lens()}, a);
        auto bb  = p.add_instruction(migraphx::op::broadcast{1, s.lens()}, b);
        auto mul = p.add_instruction(migraphx::op::mul{}, x, ab);
        p.add_instruction(migraphx::op::add{}, mul, bb);
        return p;
    }
};

Paul's avatar
Paul committed
429
struct test_add_broadcast : verify_program<test_add_broadcast>
Paul's avatar
Paul committed
430
{
Paul's avatar
Paul committed
431
    migraphx::program create_program() const
Paul's avatar
Paul committed
432
    {
Paul's avatar
Paul committed
433
434
435
436
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {3}};
        auto x  = p.add_parameter("x", {migraphx::shape::float_type, {2, 2, 3}});
        auto y  = p.add_parameter("y", {migraphx::shape::float_type, {2, 2}});
437
        auto by = p.add_instruction(migraphx::op::broadcast{0, x->get_shape().lens()}, y);
Paul's avatar
Paul committed
438
        p.add_instruction(migraphx::op::add{}, x, by);
Paul's avatar
Paul committed
439
440
441
442
        return p;
    }
};

Paul's avatar
Paul committed
443
struct test_add_broadcast2 : verify_program<test_add_broadcast2>
Paul's avatar
Paul committed
444
{
Paul's avatar
Paul committed
445
    migraphx::program create_program() const
Paul's avatar
Paul committed
446
    {
Paul's avatar
Paul committed
447
448
449
450
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {3}};
        auto x  = p.add_parameter("x", {migraphx::shape::float_type, {2, 3, 4}});
        auto y  = p.add_parameter("y", {migraphx::shape::float_type, {3}});
451
        auto by = p.add_instruction(migraphx::op::broadcast{1, x->get_shape().lens()}, y);
Paul's avatar
Paul committed
452
        p.add_instruction(migraphx::op::add{}, x, by);
Paul's avatar
Paul committed
453
454
455
456
        return p;
    }
};

Paul's avatar
Paul committed
457
struct test_add_broadcast3 : verify_program<test_add_broadcast3>
Paul's avatar
Latest  
Paul committed
458
{
Paul's avatar
Paul committed
459
    migraphx::program create_program() const
Paul's avatar
Latest  
Paul committed
460
    {
Paul's avatar
Paul committed
461
462
463
464
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {3}};
        auto x  = p.add_parameter("x", {migraphx::shape::float_type, {2, 4, 5}});
        auto y  = p.add_parameter("y", {migraphx::shape::float_type, {4}});
465
        auto by = p.add_instruction(migraphx::op::broadcast{1, x->get_shape().lens()}, y);
Paul's avatar
Paul committed
466
        p.add_instruction(migraphx::op::add{}, x, by);
Paul's avatar
Latest  
Paul committed
467
468
469
470
        return p;
    }
};

Paul's avatar
Paul committed
471
struct test_add_broadcast4 : verify_program<test_add_broadcast4>
Paul's avatar
Latest  
Paul committed
472
{
Paul's avatar
Paul committed
473
    migraphx::program create_program() const
Paul's avatar
Latest  
Paul committed
474
    {
Paul's avatar
Paul committed
475
476
477
478
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {3}};
        auto x  = p.add_parameter("x", {migraphx::shape::float_type, {2, 3, 5}});
        auto y  = p.add_parameter("y", {migraphx::shape::float_type, {3}});
479
        auto by = p.add_instruction(migraphx::op::broadcast{1, x->get_shape().lens()}, y);
Paul's avatar
Paul committed
480
        p.add_instruction(migraphx::op::add{}, x, by);
Paul's avatar
Latest  
Paul committed
481
482
483
484
        return p;
    }
};

Paul's avatar
Paul committed
485
struct test_add_broadcast5 : verify_program<test_add_broadcast5>
Paul's avatar
Paul committed
486
{
Paul's avatar
Paul committed
487
    migraphx::program create_program() const
Paul's avatar
Paul committed
488
    {
Paul's avatar
Paul committed
489
490
491
492
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {3}};
        auto x  = p.add_parameter("x", {migraphx::shape::float_type, {2, 4, 8}});
        auto y  = p.add_parameter("y", {migraphx::shape::float_type, {4}});
493
        auto by = p.add_instruction(migraphx::op::broadcast{1, x->get_shape().lens()}, y);
Paul's avatar
Paul committed
494
        p.add_instruction(migraphx::op::add{}, x, by);
Paul's avatar
Paul committed
495
496
497
498
        return p;
    }
};

Paul's avatar
Paul committed
499
struct test_triadd_broadcast : verify_program<test_triadd_broadcast>
Paul's avatar
Paul committed
500
{
Paul's avatar
Paul committed
501
    migraphx::program create_program() const
Paul's avatar
Paul committed
502
    {
Paul's avatar
Paul committed
503
504
505
506
507
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {3}};
        auto x   = p.add_parameter("x", {migraphx::shape::float_type, {2, 2, 3}});
        auto y   = p.add_parameter("y", {migraphx::shape::float_type, {2, 2}});
        auto z   = p.add_parameter("z", {migraphx::shape::float_type, {2, 2, 3}});
508
        auto by  = p.add_instruction(migraphx::op::broadcast{0, x->get_shape().lens()}, y);
Paul's avatar
Paul committed
509
510
        auto sum = p.add_instruction(migraphx::op::add{}, x, by);
        p.add_instruction(migraphx::op::add{}, sum, z);
Paul's avatar
Paul committed
511
512
513
514
        return p;
    }
};

kahmed10's avatar
kahmed10 committed
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
struct test_gelu : verify_program<test_gelu>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        std::vector<size_t> input_lens{1, 1, 5};
        auto x            = p.add_parameter("x", {migraphx::shape::float_type, input_lens});
        auto half         = p.add_literal(0.5f);
        auto one          = p.add_literal(1.0f);
        auto sqrt2        = p.add_literal(static_cast<float>(M_SQRT2));
        auto half_mbcast  = p.add_instruction(migraphx::op::multibroadcast{input_lens}, half);
        auto mul_half     = p.add_instruction(migraphx::op::mul{}, x, half_mbcast);
        auto sqrt2_mbcast = p.add_instruction(migraphx::op::multibroadcast{input_lens}, sqrt2);
        auto div          = p.add_instruction(migraphx::op::div{}, x, sqrt2_mbcast);
        auto erf          = p.add_instruction(migraphx::op::erf{}, div);
        auto one_mbcast   = p.add_instruction(migraphx::op::multibroadcast{input_lens}, one);
        auto add_one      = p.add_instruction(migraphx::op::add{}, erf, one_mbcast);
        p.add_instruction(migraphx::op::mul{}, mul_half, add_one);
        return p;
    }
};

struct test_add_gelu : verify_program<test_add_gelu>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        std::vector<size_t> input_lens{1, 1, 5};
        auto x            = p.add_parameter("x", {migraphx::shape::float_type, input_lens});
        auto y            = p.add_parameter("y", {migraphx::shape::float_type, input_lens});
        auto half         = p.add_literal(0.5f);
        auto one          = p.add_literal(1.0f);
        auto sqrt2        = p.add_literal(static_cast<float>(M_SQRT2));
        auto add          = p.add_instruction(migraphx::op::add{}, x, y);
        auto half_mbcast  = p.add_instruction(migraphx::op::multibroadcast{input_lens}, half);
        auto mul_half     = p.add_instruction(migraphx::op::mul{}, add, half_mbcast);
        auto sqrt2_mbcast = p.add_instruction(migraphx::op::multibroadcast{input_lens}, sqrt2);
        auto div          = p.add_instruction(migraphx::op::div{}, add, sqrt2_mbcast);
        auto erf          = p.add_instruction(migraphx::op::erf{}, div);
        auto one_mbcast   = p.add_instruction(migraphx::op::multibroadcast{input_lens}, one);
        auto add_one      = p.add_instruction(migraphx::op::add{}, erf, one_mbcast);
        p.add_instruction(migraphx::op::mul{}, mul_half, add_one);
        return p;
    }
};

Paul's avatar
Paul committed
561
struct test_sub : verify_program<test_sub>
562
563
564
565
566
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {3}};
Shucai Xiao's avatar
Shucai Xiao committed
567
568
569
        auto x    = p.add_parameter("x", s);
        auto y    = p.add_parameter("y", s);
        auto z    = p.add_parameter("z", s);
570
571
572
573
574
575
        auto diff = p.add_instruction(migraphx::op::sub{}, x, y);
        p.add_instruction(migraphx::op::sub{}, diff, z);
        return p;
    }
};

Paul's avatar
Paul committed
576
struct test_sub2 : verify_program<test_sub2>
577
578
579
580
581
582
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {2, 3}};
        migraphx::shape b{migraphx::shape::float_type, {3}};
Shucai Xiao's avatar
Shucai Xiao committed
583
584
585
        auto x    = p.add_parameter("x", s);
        auto y    = p.add_parameter("y", s);
        auto z    = p.add_parameter("z", b);
586
        auto zb   = p.add_instruction(migraphx::op::broadcast{1, s.lens()}, z);
587
588
589
590
591
592
        auto diff = p.add_instruction(migraphx::op::sub{}, x, y);
        p.add_instruction(migraphx::op::sub{}, diff, zb);
        return p;
    }
};

593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
struct test_div : verify_program<test_div>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {3}};
        auto x    = p.add_parameter("x", s);
        auto y    = p.add_parameter("y", s);
        auto z    = p.add_parameter("z", s);
        auto diff = p.add_instruction(migraphx::op::div{}, x, y);
        p.add_instruction(migraphx::op::div{}, diff, z);
        return p;
    }
};

struct test_div2 : verify_program<test_div2>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {2, 3}};
        migraphx::shape b{migraphx::shape::float_type, {3}};
        auto x    = p.add_parameter("x", s);
        auto y    = p.add_parameter("y", s);
        auto z    = p.add_parameter("z", b);
        auto zb   = p.add_instruction(migraphx::op::broadcast{1, s.lens()}, z);
        auto diff = p.add_instruction(migraphx::op::div{}, x, y);
        p.add_instruction(migraphx::op::div{}, diff, zb);
        return p;
    }
};

625
struct test_softmax1 : verify_program<test_softmax1>
Paul's avatar
Paul committed
626
{
Paul's avatar
Paul committed
627
    migraphx::program create_program() const
Paul's avatar
Paul committed
628
    {
Paul's avatar
Paul committed
629
        migraphx::program p;
630
631
        auto x = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {5, 3, 3, 4}});
        p.add_instruction(migraphx::op::softmax{0}, x);
Paul's avatar
Paul committed
632
633
634
635
        return p;
    }
};

Paul's avatar
Paul committed
636
struct test_softmax2 : verify_program<test_softmax2>
Paul's avatar
Paul committed
637
{
Paul's avatar
Paul committed
638
    migraphx::program create_program() const
Paul's avatar
Paul committed
639
    {
Paul's avatar
Paul committed
640
        migraphx::program p;
Paul's avatar
Paul committed
641
642
        auto x =
            p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {1, 1000, 1, 1}});
Paul's avatar
Paul committed
643
        p.add_instruction(migraphx::op::softmax{}, x);
Paul's avatar
Paul committed
644
645
646
647
        return p;
    }
};

Shucai Xiao's avatar
Shucai Xiao committed
648
649
template <int Axis, migraphx::shape::type_t T>
struct test_softmax : verify_program<test_softmax<Axis, T>>
650
651
652
653
{
    migraphx::program create_program() const
    {
        migraphx::program p;
Shucai Xiao's avatar
Shucai Xiao committed
654
        migraphx::shape s{T, {512, 4, 1067, 6}};
655
656
657
658
659
660
661
        auto param = p.add_parameter("0", s);
        p.add_instruction(migraphx::op::softmax{Axis}, param);

        return p;
    }
};

Shucai Xiao's avatar
Shucai Xiao committed
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
template struct test_softmax<0, migraphx::shape::float_type>;
template struct test_softmax<2, migraphx::shape::float_type>;
template struct test_softmax<1, migraphx::shape::double_type>;
template struct test_softmax<3, migraphx::shape::double_type>;
template struct test_softmax<0, migraphx::shape::half_type>;
template struct test_softmax<1, migraphx::shape::half_type>;
template struct test_softmax<2, migraphx::shape::half_type>;
template struct test_softmax<3, migraphx::shape::half_type>;

template <class T, int Axis>
struct test_arg_ops : verify_program<test_arg_ops<T, Axis>>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {2, 3, 4, 1025}};
        auto param = p.add_parameter("data", s);
        p.add_instruction(T{Axis}, param);

        return p;
    }
};

template struct test_arg_ops<migraphx::op::argmax, 0>;
template struct test_arg_ops<migraphx::op::argmax, 1>;
template struct test_arg_ops<migraphx::op::argmax, 2>;
template struct test_arg_ops<migraphx::op::argmax, 3>;
689
690
template struct test_arg_ops<migraphx::op::argmax, -1>;
template struct test_arg_ops<migraphx::op::argmax, -2>;
Shucai Xiao's avatar
Shucai Xiao committed
691
692
693
694
695

template struct test_arg_ops<migraphx::op::argmin, 0>;
template struct test_arg_ops<migraphx::op::argmin, 1>;
template struct test_arg_ops<migraphx::op::argmin, 2>;
template struct test_arg_ops<migraphx::op::argmin, 3>;
696
697
template struct test_arg_ops<migraphx::op::argmin, -3>;
template struct test_arg_ops<migraphx::op::argmin, -4>;
698

Paul's avatar
Paul committed
699
struct test_conv : verify_program<test_conv>
Paul's avatar
Paul committed
700
{
Paul's avatar
Paul committed
701
    migraphx::program create_program() const
Paul's avatar
Paul committed
702
    {
Paul's avatar
Paul committed
703
        migraphx::program p;
Paul's avatar
Paul committed
704
705
        auto input =
            p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
Paul's avatar
Paul committed
706
        auto weights =
Paul's avatar
Paul committed
707
708
            p.add_parameter("w", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        p.add_instruction(migraphx::op::convolution{}, input, weights);
Paul's avatar
Paul committed
709
710
711
712
        return p;
    }
};

Paul's avatar
Paul committed
713
struct test_conv2 : verify_program<test_conv2>
Paul's avatar
Paul committed
714
{
Paul's avatar
Paul committed
715
    migraphx::program create_program() const
Paul's avatar
Paul committed
716
    {
Paul's avatar
Paul committed
717
        migraphx::program p;
Paul's avatar
Paul committed
718
        auto input =
Paul's avatar
Paul committed
719
            p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {1, 512, 28, 28}});
Paul's avatar
Paul committed
720
        auto weights =
Paul's avatar
Paul committed
721
722
            p.add_parameter("w", migraphx::shape{migraphx::shape::float_type, {256, 512, 1, 1}});
        p.add_instruction(migraphx::op::convolution{{0, 0}, {1, 1}, {1, 1}}, input, weights);
Paul's avatar
Paul committed
723
724
725
726
        return p;
    }
};

Paul's avatar
Paul committed
727
struct test_group_conv : verify_program<test_group_conv>
Khalique's avatar
Khalique committed
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        auto input =
            p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {1, 4, 16, 16}});
        auto weights =
            p.add_parameter("w", migraphx::shape{migraphx::shape::float_type, {4, 1, 3, 3}});
        migraphx::op::convolution op;
        op.group = 4;
        p.add_instruction(op, input, weights);
        return p;
    }
};

Paul's avatar
Paul committed
743
struct test_conv_relu : verify_program<test_conv_relu>
Paul's avatar
Paul committed
744
{
Paul's avatar
Paul committed
745
    migraphx::program create_program() const
Paul's avatar
Paul committed
746
    {
Paul's avatar
Paul committed
747
        migraphx::program p;
Paul's avatar
Paul committed
748
749
        auto input =
            p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
Paul's avatar
Paul committed
750
        auto weights =
Paul's avatar
Paul committed
751
752
753
            p.add_parameter("w", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        auto conv = p.add_instruction(migraphx::op::convolution{}, input, weights);
        p.add_instruction(migraphx::op::relu{}, conv);
Paul's avatar
Paul committed
754
755
756
757
        return p;
    }
};

Paul's avatar
Paul committed
758
struct test_conv_relu_half : verify_program<test_conv_relu_half>
Paul's avatar
Paul committed
759
{
Paul's avatar
Paul committed
760
    migraphx::program create_program() const
Paul's avatar
Paul committed
761
    {
Paul's avatar
Paul committed
762
        migraphx::program p;
Paul's avatar
Paul committed
763
764
        auto input =
            p.add_parameter("x", migraphx::shape{migraphx::shape::half_type, {4, 3, 3, 3}});
Paul's avatar
Paul committed
765
        auto weights =
Paul's avatar
Paul committed
766
767
768
            p.add_parameter("w", migraphx::shape{migraphx::shape::half_type, {4, 3, 3, 3}});
        auto conv = p.add_instruction(migraphx::op::convolution{}, input, weights);
        p.add_instruction(migraphx::op::relu{}, conv);
Paul's avatar
Paul committed
769
770
771
772
        return p;
    }
};

kahmed10's avatar
kahmed10 committed
773
774
775
776
777
struct test_conv_bias_clipped_relu : verify_program<test_conv_bias_clipped_relu>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
kahmed10's avatar
kahmed10 committed
778
        std::vector<size_t> input_lens{4, 3, 3, 3};
kahmed10's avatar
kahmed10 committed
779
780
781
782
783
784
785
786
787
788
789
        auto input =
            p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        auto weights =
            p.add_parameter("w", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        auto l0   = migraphx::literal{migraphx::shape{migraphx::shape::float_type, {4}},
                                    {2.0f, 2.0f, 2.0f, 2.0f}};
        auto bias = p.add_literal(l0);
        auto conv = p.add_instruction(migraphx::op::convolution{}, input, weights);
        auto bcast_add =
            p.add_instruction(migraphx::op::broadcast{1, conv->get_shape().lens()}, bias);
        auto bias_add = p.add_instruction(migraphx::op::add{}, conv, bcast_add);
kahmed10's avatar
kahmed10 committed
790
791
792
793
794
        auto min_val  = p.add_literal(0.0f);
        auto max_val  = p.add_literal(6.0f);
        min_val       = p.add_instruction(migraphx::op::multibroadcast{input_lens}, min_val);
        max_val       = p.add_instruction(migraphx::op::multibroadcast{input_lens}, max_val);
        p.add_instruction(migraphx::op::clip{}, bias_add, min_val, max_val);
kahmed10's avatar
kahmed10 committed
795
796
797
798
        return p;
    }
};

799
800
801
802
803
804
struct test_conv_add : verify_program<test_conv_add>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        auto x = p.add_parameter("x", {migraphx::shape::float_type, {1, 8, 4, 4}});
805
806
        auto w = p.add_literal(
            migraphx::generate_literal({migraphx::shape::float_type, {2, 8, 3, 3}}, 1));
807
        auto y = p.add_parameter("y", {migraphx::shape::float_type, {1, 8, 4, 4}});
808
809
        auto v = p.add_literal(
            migraphx::generate_literal({migraphx::shape::float_type, {2, 8, 3, 3}}, 2));
810
811
812
813
814
815
816
817
818
819
820
821
822
823
        auto conv1 = p.add_instruction(migraphx::op::convolution{}, x, w);
        auto conv2 = p.add_instruction(migraphx::op::convolution{}, y, v);
        auto sum   = p.add_instruction(migraphx::op::add{}, conv1, conv2);
        p.add_instruction(migraphx::op::exp{}, sum);
        return p;
    }
};

struct test_conv_add_1x1_diff_strides : verify_program<test_conv_add_1x1_diff_strides>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        auto x = p.add_parameter("x", {migraphx::shape::float_type, {1, 8, 2, 2}});
824
825
        auto w = p.add_literal(
            migraphx::generate_literal({migraphx::shape::float_type, {2, 8, 1, 1}}, 1));
826
        auto y = p.add_parameter("y", {migraphx::shape::float_type, {1, 8, 4, 4}});
827
828
        auto v = p.add_literal(
            migraphx::generate_literal({migraphx::shape::float_type, {2, 8, 1, 1}}, 2));
829
830
831
832
833
834
835
836
        auto conv1 = p.add_instruction(migraphx::op::convolution{}, x, w);
        auto conv2 = p.add_instruction(migraphx::op::convolution{{0, 0}, {2, 2}}, y, v);
        auto sum   = p.add_instruction(migraphx::op::add{}, conv1, conv2);
        p.add_instruction(migraphx::op::exp{}, sum);
        return p;
    }
};

837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
struct test_conv_bn_add : verify_program<test_conv_bn_add>
{
    static migraphx::instruction_ref add_bn(migraphx::program& p,
                                            migraphx::instruction_ref x,
                                            std::size_t channels,
                                            std::size_t seed = 1)
    {
        migraphx::shape vars{migraphx::shape::float_type, {channels}};
        auto scale    = p.add_literal(migraphx::abs(migraphx::generate_literal(vars, 1 + seed)));
        auto bias     = p.add_literal(migraphx::abs(migraphx::generate_literal(vars, 2 + seed)));
        auto mean     = p.add_literal(migraphx::abs(migraphx::generate_literal(vars, 3 + seed)));
        auto variance = p.add_literal(migraphx::abs(migraphx::generate_literal(vars, 4 + seed)));
        return p.add_instruction(
            migraphx::op::batch_norm_inference{}, x, scale, bias, mean, variance);
    }

    migraphx::program create_program() const
    {
        migraphx::program p;
        std::size_t ichannels = 64;
        std::size_t ochannels = 256;
        auto x     = p.add_parameter("x", {migraphx::shape::float_type, {1, ichannels, 56, 56}});
        auto w     = p.add_literal(migraphx::generate_literal(
            {migraphx::shape::float_type, {ochannels, ichannels, 1, 1}}, 1));
        auto y     = p.add_parameter("y", {migraphx::shape::float_type, {1, ichannels, 56, 56}});
        auto v     = p.add_literal(migraphx::generate_literal(
            {migraphx::shape::float_type, {ochannels, ichannels, 1, 1}}, 2));
        auto relu1 = p.add_instruction(migraphx::op::relu{}, x);
        auto conv1 = p.add_instruction(migraphx::op::convolution{}, relu1, w);
        auto bn1   = add_bn(p, conv1, ochannels, 1);
        auto relu2 = p.add_instruction(migraphx::op::relu{}, y);
        auto conv2 = p.add_instruction(migraphx::op::convolution{}, relu2, v);
        auto bn2   = add_bn(p, conv2, ochannels, 1);
        auto sum   = p.add_instruction(migraphx::op::add{}, bn1, bn2);
        p.add_instruction(migraphx::op::relu{}, sum);
        return p;
    }
};

Paul's avatar
Paul committed
876
struct test_add_relu : verify_program<test_add_relu>
Paul's avatar
Paul committed
877
{
Paul's avatar
Paul committed
878
    migraphx::program create_program() const
Paul's avatar
Paul committed
879
    {
Paul's avatar
Paul committed
880
881
882
883
884
        migraphx::program p;
        auto x   = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        auto y   = p.add_parameter("y", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        auto add = p.add_instruction(migraphx::op::add{}, x, y);
        p.add_instruction(migraphx::op::relu{}, add);
Paul's avatar
Paul committed
885
886
887
888
        return p;
    }
};

889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
struct test_add_sigmoid : verify_program<test_add_sigmoid>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        auto x   = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        auto y   = p.add_parameter("y", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        auto add = p.add_instruction(migraphx::op::add{}, x, y);
        p.add_instruction(migraphx::op::sigmoid{}, add);
        return p;
    }
};

struct test_add_tanh : verify_program<test_add_tanh>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        auto x   = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        auto y   = p.add_parameter("y", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        auto add = p.add_instruction(migraphx::op::add{}, x, y);
        p.add_instruction(migraphx::op::tanh{}, add);
        return p;
    }
};

struct test_triadd_relu : verify_program<test_triadd_relu>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        auto x   = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        auto y   = p.add_parameter("y", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        auto z   = p.add_parameter("z", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        auto sum = p.add_instruction(migraphx::op::add{}, x, y);
        auto triadd = p.add_instruction(migraphx::op::add{}, sum, z);
        p.add_instruction(migraphx::op::relu{}, triadd);
        return p;
    }
};

struct test_triadd_sigmoid : verify_program<test_triadd_sigmoid>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        auto x   = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        auto y   = p.add_parameter("y", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        auto z   = p.add_parameter("z", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        auto sum = p.add_instruction(migraphx::op::add{}, x, y);
        auto triadd = p.add_instruction(migraphx::op::add{}, sum, z);
        p.add_instruction(migraphx::op::sigmoid{}, triadd);
        return p;
    }
};

struct test_triadd_tanh : verify_program<test_triadd_tanh>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        auto x   = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        auto y   = p.add_parameter("y", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        auto z   = p.add_parameter("z", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        auto sum = p.add_instruction(migraphx::op::add{}, x, y);
        auto triadd = p.add_instruction(migraphx::op::add{}, sum, z);
        p.add_instruction(migraphx::op::tanh{}, triadd);
        return p;
    }
};

Paul's avatar
Paul committed
960
struct test_sigmoid : verify_program<test_sigmoid>
Khalique's avatar
Khalique committed
961
962
963
964
{
    migraphx::program create_program() const
    {
        migraphx::program p;
Khalique's avatar
Khalique committed
965
        auto x = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
Khalique's avatar
Khalique committed
966
967
968
969
970
        p.add_instruction(migraphx::op::sigmoid{}, x);
        return p;
    }
};

Paul's avatar
Paul committed
971
struct test_abs : verify_program<test_abs>
Khalique's avatar
Khalique committed
972
973
974
975
{
    migraphx::program create_program() const
    {
        migraphx::program p;
Khalique's avatar
Khalique committed
976
        auto x = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
Khalique's avatar
Khalique committed
977
978
979
980
981
        p.add_instruction(migraphx::op::abs{}, x);
        return p;
    }
};

982
983
984
985
986
struct test_trans_abs : verify_program<test_trans_abs>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
Shucai Xiao's avatar
Shucai Xiao committed
987
        auto x  = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
988
        auto tx = p.add_instruction(migraphx::op::transpose{{0, 1, 3, 2}}, x);
989
        auto absx = p.add_instruction(migraphx::op::abs{}, tx);
Shucai Xiao's avatar
Shucai Xiao committed
990
        auto r    = p.add_instruction(migraphx::op::add{}, absx, absx);
991
        p.add_instruction(migraphx::op::contiguous{}, r);
Shucai Xiao's avatar
Shucai Xiao committed
992

993
994
995
996
        return p;
    }
};

Paul's avatar
Paul committed
997
struct test_leaky_relu : verify_program<test_leaky_relu>
998
{
Paul's avatar
Paul committed
999
    migraphx::program create_program() const
1000
    {
Paul's avatar
Paul committed
1001
1002
1003
        migraphx::program p;
        auto x = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        p.add_instruction(migraphx::op::leaky_relu{0.01}, x);
1004
1005
1006
1007
        return p;
    }
};

Paul's avatar
Paul committed
1008
struct test_elu : verify_program<test_elu>
Khalique's avatar
Khalique committed
1009
1010
1011
1012
1013
1014
1015
1016
1017
1018
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        auto x = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        p.add_instruction(migraphx::op::leaky_relu{1.0}, x);
        return p;
    }
};

Paul's avatar
Paul committed
1019
struct test_relu_lrn : verify_program<test_relu_lrn>
Khalique's avatar
Khalique committed
1020
{
Khalique's avatar
Khalique committed
1021
    migraphx::program create_program() const
Khalique's avatar
Khalique committed
1022
    {
Khalique's avatar
Khalique committed
1023
1024
1025
1026
        migraphx::program p;
        auto x = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {1, 5, 2, 2}});
        auto y = p.add_instruction(migraphx::op::relu{}, x);
        p.add_instruction(migraphx::op::lrn{0.0001, 0.75, 1.0, 5}, y);
Khalique's avatar
Khalique committed
1027
1028
1029
1030
        return p;
    }
};

Paul's avatar
Paul committed
1031
struct test_conv_pooling : verify_program<test_conv_pooling>
Paul's avatar
Paul committed
1032
{
Paul's avatar
Paul committed
1033
    migraphx::program create_program() const
Paul's avatar
Paul committed
1034
    {
Paul's avatar
Paul committed
1035
        migraphx::program p;
Paul's avatar
Paul committed
1036
        auto input =
Paul's avatar
Paul committed
1037
            p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {4, 3, 32, 32}});
Paul's avatar
Paul committed
1038
        auto weights =
Paul's avatar
Paul committed
1039
1040
1041
1042
            p.add_parameter("w", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        auto conv    = p.add_instruction(migraphx::op::convolution{}, input, weights);
        auto pooling = p.add_instruction(migraphx::op::pooling{"max"}, conv);
        p.add_instruction(migraphx::op::relu{}, pooling);
Paul's avatar
Paul committed
1043
1044
1045
1046
        return p;
    }
};

1047
1048
1049
1050
1051
1052
1053
1054
1055
1056
1057
1058
1059
1060
1061
1062
1063
1064
struct test_concat_pooling : verify_program<test_concat_pooling>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        auto input =
            p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {1, 256, 8, 8}});
        auto transpose = p.add_instruction(migraphx::op::transpose{{0, 2, 3, 1}}, input);
        auto concat    = p.add_instruction(migraphx::op::concat{3}, transpose);
        auto concat_t  = p.add_instruction(migraphx::op::transpose{{0, 3, 1, 2}}, concat);

        auto pooling =
            p.add_instruction(migraphx::op::pooling{"average", {0, 0}, {1, 1}, {8, 8}}, concat_t);
        p.add_instruction(migraphx::op::relu{}, pooling);
        return p;
    }
};

Paul's avatar
Paul committed
1065
struct test_global_avg_pooling : verify_program<test_global_avg_pooling>
1066
{
Paul's avatar
Paul committed
1067
    migraphx::program create_program() const
1068
    {
Paul's avatar
Paul committed
1069
        migraphx::program p;
1070
        auto input =
Paul's avatar
Paul committed
1071
1072
            p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {1, 3, 16, 16}});
        auto op    = migraphx::op::pooling{"average"};
1073
        auto lens  = input->get_shape().lens();
Khalique's avatar
Khalique committed
1074
        op.lengths = {lens[2], lens[3]};
1075
1076
1077
1078
1079
        p.add_instruction(op, input);
        return p;
    }
};

Paul's avatar
Paul committed
1080
struct test_global_max_pooling : verify_program<test_global_max_pooling>
1081
{
Paul's avatar
Paul committed
1082
    migraphx::program create_program() const
1083
    {
Paul's avatar
Paul committed
1084
        migraphx::program p;
1085
        auto input =
Paul's avatar
Paul committed
1086
1087
            p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {1, 3, 16, 16}});
        auto op    = migraphx::op::pooling{"max"};
1088
        auto lens  = input->get_shape().lens();
Khalique's avatar
Khalique committed
1089
        op.lengths = {lens[2], lens[3]};
1090
1091
1092
1093
1094
        p.add_instruction(op, input);
        return p;
    }
};

Paul's avatar
Paul committed
1095
struct test_gemm : verify_program<test_gemm>
Paul's avatar
Paul committed
1096
{
Paul's avatar
Paul committed
1097
    migraphx::program create_program() const
Paul's avatar
Paul committed
1098
    {
Paul's avatar
Paul committed
1099
1100
1101
1102
        migraphx::program p;
        auto a = p.add_parameter("a", migraphx::shape{migraphx::shape::float_type, {4, 5}});
        auto b = p.add_parameter("b", migraphx::shape{migraphx::shape::float_type, {5, 3}});
        p.add_instruction(migraphx::op::dot{}, a, b);
Paul's avatar
Paul committed
1103
1104
1105
1106
        return p;
    }
};

Shucai Xiao's avatar
Shucai Xiao committed
1107
1108
1109
1110
1111
1112
1113
1114
1115
1116
1117
1118
1119
1120
1121
1122
1123
1124
struct test_gemm_copy : verify_program<test_gemm_copy>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape sa{migraphx::shape::float_type, {2, 16}};
        migraphx::shape sb{migraphx::shape::float_type, {16, 8}};
        migraphx::shape sc{migraphx::shape::float_type, {2, 8}};
        auto pa = p.add_parameter("a", sa);
        auto pb = p.add_parameter("b", sb);
        auto pc = p.add_parameter("c", sc);
        auto dr = p.add_instruction(migraphx::op::dot{}, pa, pb, pc);
        p.add_instruction(migraphx::op::add{}, dr, dr);

        return p;
    }
};

Paul's avatar
Paul committed
1125
struct test_gemm_ex : verify_program<test_gemm_ex>
1126
1127
1128
1129
1130
1131
1132
1133
1134
1135
1136
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        auto a = p.add_parameter("a", migraphx::shape{migraphx::shape::float_type, {1, 1, 4, 5}});
        auto b = p.add_parameter("b", migraphx::shape{migraphx::shape::float_type, {1, 1, 5, 3}});
        p.add_instruction(migraphx::op::dot{}, a, b);
        return p;
    }
};

Paul's avatar
Paul committed
1137
struct test_gemm_half : verify_program<test_gemm_half>
Paul's avatar
Paul committed
1138
{
Paul's avatar
Paul committed
1139
    migraphx::program create_program() const
Paul's avatar
Paul committed
1140
    {
Paul's avatar
Paul committed
1141
1142
1143
1144
        migraphx::program p;
        auto a = p.add_parameter("a", migraphx::shape{migraphx::shape::half_type, {4, 5}});
        auto b = p.add_parameter("b", migraphx::shape{migraphx::shape::half_type, {5, 3}});
        p.add_instruction(migraphx::op::dot{}, a, b);
Paul's avatar
Paul committed
1145
1146
1147
1148
        return p;
    }
};

Paul's avatar
Paul committed
1149
struct test_gemm_ld //: verify_program<test_gemm_ld>
Paul's avatar
Paul committed
1150
{
Paul's avatar
Paul committed
1151
    migraphx::program create_program() const
Paul's avatar
Paul committed
1152
    {
Paul's avatar
Paul committed
1153
        migraphx::program p;
Paul's avatar
Paul committed
1154
1155
1156
1157
        auto a =
            p.add_parameter("a", migraphx::shape{migraphx::shape::float_type, {4, 5}, {10, 1}});
        auto b =
            p.add_parameter("b", migraphx::shape{migraphx::shape::float_type, {5, 3}, {20, 1}});
Paul's avatar
Paul committed
1158
        p.add_instruction(migraphx::op::dot{}, a, b);
Paul's avatar
Paul committed
1159
1160
1161
1162
        return p;
    }
};

Paul's avatar
Paul committed
1163
struct test_gemm_transposeb : verify_program<test_gemm_transposeb>
1164
{
Paul's avatar
Paul committed
1165
    migraphx::program create_program() const
1166
    {
Paul's avatar
Paul committed
1167
1168
1169
1170
1171
        migraphx::program p;
        auto a  = p.add_parameter("a", migraphx::shape{migraphx::shape::float_type, {4, 5}});
        auto b  = p.add_parameter("b", migraphx::shape{migraphx::shape::float_type, {3, 5}});
        auto bt = p.add_instruction(migraphx::op::transpose{{1, 0}}, b);
        p.add_instruction(migraphx::op::dot{}, a, bt);
1172
1173
1174
1175
        return p;
    }
};

Paul's avatar
Paul committed
1176
struct test_gemm_transposeb_ex : verify_program<test_gemm_transposeb_ex>
1177
1178
1179
1180
1181
1182
1183
1184
1185
1186
1187
1188
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        auto a  = p.add_parameter("a", migraphx::shape{migraphx::shape::float_type, {1, 4, 5}});
        auto b  = p.add_parameter("b", migraphx::shape{migraphx::shape::float_type, {1, 3, 5}});
        auto bt = p.add_instruction(migraphx::op::transpose{{0, 2, 1}}, b);
        p.add_instruction(migraphx::op::dot{}, a, bt);
        return p;
    }
};

Paul's avatar
Paul committed
1189
struct test_gemm_transposea : verify_program<test_gemm_transposea>
1190
{
Paul's avatar
Paul committed
1191
    migraphx::program create_program() const
1192
    {
Paul's avatar
Paul committed
1193
1194
1195
1196
1197
        migraphx::program p;
        auto a  = p.add_parameter("a", migraphx::shape{migraphx::shape::float_type, {5, 4}});
        auto b  = p.add_parameter("b", migraphx::shape{migraphx::shape::float_type, {5, 3}});
        auto at = p.add_instruction(migraphx::op::transpose{{1, 0}}, a);
        p.add_instruction(migraphx::op::dot{}, at, b);
1198
1199
1200
1201
        return p;
    }
};

Paul's avatar
Paul committed
1202
struct test_gemm_transposea_ex : verify_program<test_gemm_transposea_ex>
1203
1204
1205
1206
1207
1208
1209
1210
1211
1212
1213
1214
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        auto a  = p.add_parameter("a", migraphx::shape{migraphx::shape::float_type, {1, 1, 5, 4}});
        auto b  = p.add_parameter("b", migraphx::shape{migraphx::shape::float_type, {1, 1, 5, 3}});
        auto at = p.add_instruction(migraphx::op::transpose{{0, 1, 3, 2}}, a);
        p.add_instruction(migraphx::op::dot{}, at, b);
        return p;
    }
};

Paul's avatar
Paul committed
1215
struct test_gemm_transposeab : verify_program<test_gemm_transposeab>
1216
{
Paul's avatar
Paul committed
1217
    migraphx::program create_program() const
1218
    {
Paul's avatar
Paul committed
1219
1220
1221
1222
1223
1224
        migraphx::program p;
        auto a  = p.add_parameter("a", migraphx::shape{migraphx::shape::float_type, {5, 4}});
        auto b  = p.add_parameter("b", migraphx::shape{migraphx::shape::float_type, {3, 5}});
        auto at = p.add_instruction(migraphx::op::transpose{{1, 0}}, a);
        auto bt = p.add_instruction(migraphx::op::transpose{{1, 0}}, b);
        p.add_instruction(migraphx::op::dot{}, at, bt);
1225
1226
1227
1228
        return p;
    }
};

1229
struct gemm_multi_dim_2 : verify_program<gemm_multi_dim_2>
1230
1231
1232
1233
1234
1235
1236
1237
1238
1239
1240
1241
1242
1243
1244
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::float_type, {2, 2, 3}};
        migraphx::shape m2_shape{migraphx::shape::float_type, {2, 3, 4}};
        auto l1 = p.add_parameter("1", m1_shape);
        auto l2 = p.add_parameter("2", m2_shape);

        p.add_instruction(migraphx::op::dot{}, l1, l2);

        return p;
    }
};

1245
1246
1247
1248
1249
1250
1251
struct gemm_2args_mm_1 : verify_program<gemm_2args_mm_1>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::float_type, {2, 2, 3}};
        migraphx::shape m2_shape{migraphx::shape::float_type, {1, 3, 4}};
Shucai Xiao's avatar
Shucai Xiao committed
1252
1253
        auto l1  = p.add_parameter("1", m1_shape);
        auto l2  = p.add_parameter("2", m2_shape);
1254
1255
1256
1257
1258
1259
1260
1261
1262
1263
1264
1265
1266
1267
1268
        auto bl2 = p.add_instruction(migraphx::op::multibroadcast{{2, 3, 4}}, l2);

        p.add_instruction(migraphx::op::dot{}, l1, bl2);

        return p;
    }
};

struct gemm_2args_mm_2 : verify_program<gemm_2args_mm_2>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::float_type, {2, 2, 3}};
        migraphx::shape m2_shape{migraphx::shape::float_type, {3, 4}};
Shucai Xiao's avatar
Shucai Xiao committed
1269
1270
        auto l1  = p.add_parameter("1", m1_shape);
        auto l2  = p.add_parameter("2", m2_shape);
1271
1272
1273
1274
1275
1276
1277
1278
1279
1280
1281
1282
1283
1284
1285
        auto bl2 = p.add_instruction(migraphx::op::multibroadcast{{2, 3, 4}}, l2);

        p.add_instruction(migraphx::op::dot{}, l1, bl2);

        return p;
    }
};

struct gemm_2args_mm_3 : verify_program<gemm_2args_mm_3>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::float_type, {1, 2, 3}};
        migraphx::shape m2_shape{migraphx::shape::float_type, {3, 3, 4}};
Shucai Xiao's avatar
Shucai Xiao committed
1286
        auto l1  = p.add_parameter("1", m1_shape);
1287
        auto bl1 = p.add_instruction(migraphx::op::multibroadcast{{3, 2, 3}}, l1);
Shucai Xiao's avatar
Shucai Xiao committed
1288
        auto l2  = p.add_parameter("2", m2_shape);
1289
1290
1291
1292
1293
1294
1295
1296
1297
1298
1299
1300
1301
1302

        p.add_instruction(migraphx::op::dot{}, bl1, l2);

        return p;
    }
};

struct gemm_2args_mm_4 : verify_program<gemm_2args_mm_4>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::float_type, {2, 3}};
        migraphx::shape m2_shape{migraphx::shape::float_type, {3, 3, 4}};
Shucai Xiao's avatar
Shucai Xiao committed
1303
        auto l1  = p.add_parameter("1", m1_shape);
1304
        auto bl1 = p.add_instruction(migraphx::op::multibroadcast{{3, 2, 3}}, l1);
Shucai Xiao's avatar
Shucai Xiao committed
1305
        auto l2  = p.add_parameter("2", m2_shape);
1306
1307
1308
1309
1310
1311
1312
1313
1314
1315
1316
1317
1318
1319

        p.add_instruction(migraphx::op::dot{}, bl1, l2);

        return p;
    }
};

struct gemm_2args_mm_5 : verify_program<gemm_2args_mm_5>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::float_type, {2, 1, 2, 3}};
        migraphx::shape m2_shape{migraphx::shape::float_type, {2, 3, 3, 4}};
Shucai Xiao's avatar
Shucai Xiao committed
1320
        auto l1  = p.add_parameter("1", m1_shape);
1321
        auto bl1 = p.add_instruction(migraphx::op::multibroadcast{{2, 3, 2, 3}}, l1);
Shucai Xiao's avatar
Shucai Xiao committed
1322
        auto l2  = p.add_parameter("2", m2_shape);
1323
1324
1325
1326
1327
1328
1329
1330
1331
1332
1333
1334
1335
1336

        p.add_instruction(migraphx::op::dot{}, bl1, l2);

        return p;
    }
};

struct gemm_2args_mm_6 : verify_program<gemm_2args_mm_6>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::float_type, {2, 1, 2, 3}};
        migraphx::shape m2_shape{migraphx::shape::float_type, {1, 3, 3, 4}};
Shucai Xiao's avatar
Shucai Xiao committed
1337
        auto l1  = p.add_parameter("1", m1_shape);
1338
        auto bl1 = p.add_instruction(migraphx::op::multibroadcast{{2, 3, 2, 3}}, l1);
Shucai Xiao's avatar
Shucai Xiao committed
1339
        auto l2  = p.add_parameter("2", m2_shape);
1340
1341
1342
1343
1344
1345
1346
1347
1348
1349
1350
1351
1352
1353
1354
        auto bl2 = p.add_instruction(migraphx::op::multibroadcast{{2, 3, 3, 4}}, l2);

        p.add_instruction(migraphx::op::dot{}, bl1, bl2);

        return p;
    }
};

struct gemm_2args_mm_7 : verify_program<gemm_2args_mm_7>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::float_type, {2, 3}};
        migraphx::shape m2_shape{migraphx::shape::float_type, {2, 3, 3, 4}};
Shucai Xiao's avatar
Shucai Xiao committed
1355
        auto l1  = p.add_parameter("1", m1_shape);
1356
        auto bl1 = p.add_instruction(migraphx::op::multibroadcast{{2, 3, 2, 3}}, l1);
Shucai Xiao's avatar
Shucai Xiao committed
1357
        auto l2  = p.add_parameter("2", m2_shape);
1358
1359
1360
1361
1362
1363
1364
1365

        p.add_instruction(migraphx::op::dot{}, bl1, l2);

        return p;
    }
};

struct gemm_multi_dim_2_3 : verify_program<gemm_multi_dim_2_3>
1366
1367
1368
1369
1370
1371
1372
1373
1374
1375
1376
1377
1378
1379
1380
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::float_type, {2, 3, 2, 3}};
        migraphx::shape m2_shape{migraphx::shape::float_type, {2, 3, 3, 2}};
        auto l1 = p.add_parameter("1", m1_shape);
        auto l2 = p.add_parameter("2", m2_shape);

        p.add_instruction(migraphx::op::dot{}, l1, l2);

        return p;
    }
};

1381
1382
1383
1384
1385
1386
1387
1388
struct gemm_2args_vv : verify_program<gemm_2args_vv>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::float_type, {8}};
        migraphx::shape m2_shape{migraphx::shape::float_type, {8}};
        auto l1     = p.add_parameter("1", m1_shape);
Shucai Xiao's avatar
Shucai Xiao committed
1389
        auto ul1    = p.add_instruction(migraphx::op::unsqueeze{{0}}, l1);
1390
        auto l2     = p.add_parameter("2", m2_shape);
Shucai Xiao's avatar
Shucai Xiao committed
1391
        auto ul2    = p.add_instruction(migraphx::op::unsqueeze{{1}}, l2);
1392
1393
        float alpha = 0.23f;

Shucai Xiao's avatar
Shucai Xiao committed
1394
        auto res  = p.add_instruction(migraphx::op::dot{alpha}, ul1, ul2);
1395
1396
1397
1398
1399
1400
1401
1402
1403
1404
1405
1406
1407
1408
        auto sres = p.add_instruction(migraphx::op::squeeze{{0}}, res);
        p.add_instruction(migraphx::op::squeeze{{0}}, sres);

        return p;
    }
};

struct gemm_2args_mv : verify_program<gemm_2args_mv>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::float_type, {3, 5}};
        migraphx::shape m2_shape{migraphx::shape::float_type, {5}};
Shucai Xiao's avatar
Shucai Xiao committed
1409
1410
        auto l1  = p.add_parameter("1", m1_shape);
        auto l2  = p.add_parameter("2", m2_shape);
1411
1412
1413
1414
1415
1416
1417
1418
1419
1420
1421
1422
1423
1424
1425
        auto ul2 = p.add_instruction(migraphx::op::unsqueeze{{1}}, l2);

        p.add_instruction(migraphx::op::dot{}, l1, ul2);

        return p;
    }
};

struct gemm_2args_bmv : verify_program<gemm_2args_bmv>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::float_type, {2, 3, 3, 5}};
        migraphx::shape m2_shape{migraphx::shape::float_type, {5}};
Shucai Xiao's avatar
Shucai Xiao committed
1426
1427
1428
        auto l1   = p.add_parameter("1", m1_shape);
        auto l2   = p.add_parameter("2", m2_shape);
        auto ul2  = p.add_instruction(migraphx::op::unsqueeze{{1}}, l2);
1429
1430
1431
1432
1433
1434
1435
1436
1437
1438
1439
1440
1441
1442
1443
        auto bul2 = p.add_instruction(migraphx::op::multibroadcast{{2, 3, 5, 1}}, ul2);

        p.add_instruction(migraphx::op::dot{}, l1, bul2);

        return p;
    }
};

struct gemm_2args_vm : verify_program<gemm_2args_vm>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::float_type, {5}};
        migraphx::shape m2_shape{migraphx::shape::float_type, {5, 4}};
Shucai Xiao's avatar
Shucai Xiao committed
1444
        auto l1  = p.add_parameter("1", m1_shape);
1445
        auto ul1 = p.add_instruction(migraphx::op::unsqueeze{{0}}, l1);
Shucai Xiao's avatar
Shucai Xiao committed
1446
        auto l2  = p.add_parameter("2", m2_shape);
1447
1448
1449
1450
1451
1452
1453
1454
1455
1456
1457
1458
1459
1460
1461

        auto res = p.add_instruction(migraphx::op::dot{}, ul1, l2);
        p.add_instruction(migraphx::op::squeeze{{0}}, res);

        return p;
    }
};

struct gemm_2args_vbm : verify_program<gemm_2args_vbm>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::float_type, {5}};
        migraphx::shape m2_shape{migraphx::shape::float_type, {2, 2, 5, 4}};
Shucai Xiao's avatar
Shucai Xiao committed
1462
1463
        auto l1   = p.add_parameter("1", m1_shape);
        auto ul1  = p.add_instruction(migraphx::op::unsqueeze{{0}}, l1);
1464
1465
1466
1467
1468
1469
1470
1471
1472
1473
1474
1475
1476
1477
1478
1479
1480
1481
1482
1483
1484
1485
1486
1487
1488
1489
1490
1491
1492
1493
1494
1495
1496
1497
1498
1499
1500
1501
1502
1503
1504
1505
1506
1507
1508
1509
1510
1511
1512
1513
1514
1515
1516
1517
1518
1519
1520
1521
1522
1523
1524
1525
1526
1527
1528
1529
1530
1531
1532
1533
1534
1535
1536
1537
1538
1539
1540
1541
1542
1543
1544
1545
1546
1547
1548
1549
1550
1551
1552
1553
1554
        auto bul1 = p.add_instruction(migraphx::op::multibroadcast{{2, 2, 1, 5}}, ul1);

        auto l2 = p.add_parameter("2", m2_shape);

        auto res = p.add_instruction(migraphx::op::dot{}, bul1, l2);
        p.add_instruction(migraphx::op::squeeze{{2}}, res);

        return p;
    }
};

struct gemm_multi_3args : verify_program<gemm_multi_3args>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::float_type, {2, 3, 2, 3}};
        migraphx::shape m2_shape{migraphx::shape::float_type, {2, 3, 3, 2}};
        migraphx::shape m3_shape{migraphx::shape::float_type, {2, 3, 2, 2}};

        auto l1     = p.add_parameter("1", m1_shape);
        auto l2     = p.add_parameter("2", m2_shape);
        auto l3     = p.add_parameter("3", m3_shape);
        float alpha = 0.35;
        float beta  = 0.41;
        p.add_instruction(migraphx::op::dot{alpha, beta}, l1, l2, l3);

        return p;
    }
};

struct gemm_multi_3args_c25 : verify_program<gemm_multi_3args_c25>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::float_type, {2, 3}};
        migraphx::shape m2_shape{migraphx::shape::float_type, {3, 5}};
        migraphx::shape m3_shape{migraphx::shape::float_type, {2, 5}};

        auto l1     = p.add_parameter("1", m1_shape);
        auto l2     = p.add_parameter("2", m2_shape);
        auto l3     = p.add_parameter("3", m3_shape);
        float alpha = 0.35;
        float beta  = 0.41;
        p.add_instruction(migraphx::op::dot{alpha, beta}, l1, l2, l3);

        return p;
    }
};

struct gemm_multi_3args_beta0 : verify_program<gemm_multi_3args_beta0>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::float_type, {1, 2, 3}};
        migraphx::shape m2_shape{migraphx::shape::float_type, {1, 3, 4}};
        migraphx::shape m3_shape{migraphx::shape::float_type, {1, 2, 4}};
        auto l1 = p.add_parameter("1", m1_shape);
        auto l2 = p.add_parameter("2", m2_shape);
        auto l3 = p.add_parameter("3", m3_shape);

        float alpha = 1.0f;
        float beta  = 0.0f;
        p.add_instruction(migraphx::op::dot{alpha, beta}, l1, l2, l3);

        return p;
    }
};

struct gemm_multi_3args_alpha0 : verify_program<gemm_multi_3args_alpha0>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::float_type, {1, 2, 3}};
        migraphx::shape m2_shape{migraphx::shape::float_type, {1, 3, 4}};
        migraphx::shape m3_shape{migraphx::shape::float_type, {1, 2, 4}};
        auto l1 = p.add_parameter("1", m1_shape);
        auto l2 = p.add_parameter("2", m2_shape);
        auto l3 = p.add_parameter("3", m3_shape);

        float alpha = 0.0f;
        float beta  = 1.0f;
        p.add_instruction(migraphx::op::dot{alpha, beta}, l1, l2, l3);

        return p;
    }
};

Shucai Xiao's avatar
Shucai Xiao committed
1555
1556
1557
1558
1559
1560
1561
1562
1563
1564
1565
1566
1567
1568
1569
1570
1571
1572
1573
struct gemm_multi_transpose : verify_program<gemm_multi_transpose>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::float_type, {2, 2, 3}};
        migraphx::shape m2_shape{migraphx::shape::float_type, {3, 2, 4}};
        auto l1  = p.add_parameter("1", m1_shape);
        auto l2  = p.add_parameter("2", m2_shape);
        auto tl2 = p.add_instruction(migraphx::op::transpose{{1, 0, 2}}, l2);

        float alpha = 1.0f;
        float beta  = 1.0f;
        p.add_instruction(migraphx::op::dot{alpha, beta}, l1, tl2);

        return p;
    }
};

Shucai Xiao's avatar
Shucai Xiao committed
1574
1575
1576
1577
1578
1579
1580
1581
1582
1583
1584
1585
1586
1587
1588
1589
1590
1591
1592
1593
1594
1595
1596
1597
1598
1599
struct quant_dot_3args_1 : verify_program<quant_dot_3args_1>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::int8_type, {2, 8}};
        migraphx::shape m2_shape{migraphx::shape::int8_type, {8, 7}};
        migraphx::shape m3_shape{migraphx::shape::int32_type, {2, 7}};

        auto l1 = p.add_parameter("a", m1_shape);
        auto l2 = p.add_parameter("b", m2_shape);
        auto l3 = p.add_parameter("c", m3_shape);
        p.add_instruction(migraphx::op::quant_dot{}, l1, l2, l3);
        return p;
    }
};

struct quant_dot_3args_2 : verify_program<quant_dot_3args_2>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::int8_type, {8, 2}};
        migraphx::shape m2_shape{migraphx::shape::int8_type, {8, 7}};
        migraphx::shape m3_shape{migraphx::shape::int32_type, {2, 7}};

Shucai Xiao's avatar
Shucai Xiao committed
1600
        auto l1  = p.add_parameter("a", m1_shape);
Shucai Xiao's avatar
Shucai Xiao committed
1601
        auto tl1 = p.add_instruction(migraphx::op::transpose{{1, 0}}, l1);
Shucai Xiao's avatar
Shucai Xiao committed
1602
        auto l2  = p.add_parameter("b", m2_shape);
1603
        auto l3  = p.add_parameter("c", m3_shape);
Shucai Xiao's avatar
Shucai Xiao committed
1604
1605
1606
1607
1608
1609
1610
1611
1612
1613
1614
1615
1616
1617
        p.add_instruction(migraphx::op::quant_dot{1, 3}, tl1, l2, l3);
        return p;
    }
};

struct quant_dot_3args_3 : verify_program<quant_dot_3args_3>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::int8_type, {2, 8}};
        migraphx::shape m2_shape{migraphx::shape::int8_type, {7, 8}};
        migraphx::shape m3_shape{migraphx::shape::int32_type, {2, 7}};

Shucai Xiao's avatar
Shucai Xiao committed
1618
1619
        auto l1  = p.add_parameter("a", m1_shape);
        auto l2  = p.add_parameter("b", m2_shape);
Shucai Xiao's avatar
Shucai Xiao committed
1620
        auto tl2 = p.add_instruction(migraphx::op::transpose{{1, 0}}, l2);
1621
        auto l3  = p.add_parameter("c", m3_shape);
Shucai Xiao's avatar
Shucai Xiao committed
1622
1623
1624
1625
1626
1627
1628
1629
1630
1631
1632
1633
1634
1635
        p.add_instruction(migraphx::op::quant_dot{2, 3}, l1, tl2, l3);
        return p;
    }
};

struct quant_dot_3args_4 : verify_program<quant_dot_3args_4>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::int8_type, {8, 2}};
        migraphx::shape m2_shape{migraphx::shape::int8_type, {7, 8}};
        migraphx::shape m3_shape{migraphx::shape::int32_type, {2, 7}};

Shucai Xiao's avatar
Shucai Xiao committed
1636
        auto l1  = p.add_parameter("a", m1_shape);
Shucai Xiao's avatar
Shucai Xiao committed
1637
        auto tl1 = p.add_instruction(migraphx::op::transpose{{1, 0}}, l1);
Shucai Xiao's avatar
Shucai Xiao committed
1638
        auto l2  = p.add_parameter("b", m2_shape);
Shucai Xiao's avatar
Shucai Xiao committed
1639
        auto tl2 = p.add_instruction(migraphx::op::transpose{{1, 0}}, l2);
1640
        auto l3  = p.add_parameter("c", m3_shape);
Shucai Xiao's avatar
Shucai Xiao committed
1641
1642
1643
1644
1645
        p.add_instruction(migraphx::op::quant_dot{3, 2}, tl1, tl2, l3);
        return p;
    }
};

1646
1647
1648
1649
1650
1651
1652
1653
1654
1655
1656
1657
1658
1659
1660
1661
1662
1663
1664
1665
1666
1667
1668
1669
1670
1671
1672
1673
struct batch_quant_dot_1 : verify_program<batch_quant_dot_1>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::int8_type, {3, 2, 8, 2}};
        migraphx::shape m2_shape{migraphx::shape::int8_type, {3, 2, 7, 8}};
        migraphx::shape m3_shape{migraphx::shape::int32_type, {3, 2, 2, 7}};

        auto l1  = p.add_parameter("a", m1_shape);
        auto tl1 = p.add_instruction(migraphx::op::transpose{{0, 1, 3, 2}}, l1);
        auto l2  = p.add_parameter("b", m2_shape);
        auto tl2 = p.add_instruction(migraphx::op::transpose{{0, 1, 3, 2}}, l2);
        auto l3  = p.add_parameter("c", m3_shape);
        p.add_instruction(migraphx::op::quant_dot{3, 2}, tl1, tl2, l3);
        return p;
    }
};

struct batch_quant_dot_2 : verify_program<batch_quant_dot_2>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape m1_shape{migraphx::shape::int8_type, {3, 2, 2, 8}};
        migraphx::shape m2_shape{migraphx::shape::int8_type, {3, 2, 8, 7}};
        migraphx::shape m3_shape{migraphx::shape::int32_type, {3, 2, 2, 7}};

Shucai Xiao's avatar
Shucai Xiao committed
1674
1675
1676
        auto l1 = p.add_parameter("a", m1_shape);
        auto l2 = p.add_parameter("b", m2_shape);
        auto l3 = p.add_parameter("c", m3_shape);
1677
1678
1679
1680
1681
        p.add_instruction(migraphx::op::quant_dot{1, 3}, l1, l2, l3);
        return p;
    }
};

Paul's avatar
Paul committed
1682
struct test_contiguous : verify_program<test_contiguous>
1683
{
Paul's avatar
Paul committed
1684
    migraphx::program create_program() const
1685
    {
Paul's avatar
Paul committed
1686
1687
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {4, 4, 4, 3}, {48, 4, 1, 16}};
1688
        auto x = p.add_parameter("x", s);
Paul's avatar
Paul committed
1689
        p.add_instruction(migraphx::op::contiguous{}, x);
1690
        EXPECT(p.get_output_shapes().back().standard());
1691
1692
1693
1694
        return p;
    }
};

1695
1696
1697
1698
1699
1700
1701
1702
struct test_contiguous_broadcast : verify_program<test_contiguous_broadcast>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {1, 2}, {0, 1}};
        auto x = p.add_parameter("x", s);
        p.add_instruction(migraphx::op::contiguous{}, x);
1703
        EXPECT(p.get_output_shapes().back().standard());
1704
1705
1706
1707
1708
1709
1710
1711
1712
1713
1714
1715
        return p;
    }
};

struct test_contiguous_broadcast_transpose : verify_program<test_contiguous_broadcast_transpose>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {1, 3072, 768}, {0, 1, 3072}};
        auto x = p.add_parameter("x", s);
        p.add_instruction(migraphx::op::contiguous{}, x);
1716
        EXPECT(p.get_output_shapes().back().standard());
1717
1718
1719
1720
        return p;
    }
};

Paul's avatar
Paul committed
1721
struct test_transpose : verify_program<test_transpose>
1722
{
Paul's avatar
Paul committed
1723
    migraphx::program create_program() const
1724
    {
Paul's avatar
Paul committed
1725
1726
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {4, 3, 4, 4}};
1727
1728
        auto x                    = p.add_parameter("x", s);
        std::vector<int64_t> perm = {0, 2, 3, 1};
Paul's avatar
Paul committed
1729
1730
        auto l                    = p.add_instruction(migraphx::op::transpose{perm}, x);
        p.add_instruction(migraphx::op::contiguous{}, l);
1731
1732
1733
        return p;
    }
};
1734

1735
1736
1737
1738
1739
1740
1741
1742
1743
1744
1745
1746
1747
struct test_trans_ret : verify_program<test_trans_ret>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        auto x  = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
        auto tx = p.add_instruction(migraphx::op::transpose{{0, 1, 3, 2}}, x);
        p.add_return({tx});

        return p;
    }
};

Paul's avatar
Paul committed
1748
struct test_batchnorm_inference_2 : verify_program<test_batchnorm_inference_2>
Paul's avatar
Paul committed
1749
1750
1751
1752
1753
1754
{
    const size_t width    = 14;
    const size_t height   = 14;
    const size_t channels = 256;
    const size_t batches  = 1;

Paul's avatar
Paul committed
1755
    migraphx::program create_program() const
Paul's avatar
Paul committed
1756
    {
Paul's avatar
Paul committed
1757
        migraphx::program p;
Paul's avatar
Paul committed
1758

Paul's avatar
Paul committed
1759
1760
        migraphx::shape s{migraphx::shape::float_type, {batches, channels, height, width}};
        migraphx::shape vars{migraphx::shape::float_type, {channels}};
Paul's avatar
Paul committed
1761
        auto x        = p.add_parameter("x", s);
Paul's avatar
Paul committed
1762
1763
1764
1765
1766
        auto scale    = p.add_literal(migraphx::abs(migraphx::generate_literal(vars, 1)));
        auto bias     = p.add_literal(migraphx::abs(migraphx::generate_literal(vars, 2)));
        auto mean     = p.add_literal(migraphx::abs(migraphx::generate_literal(vars, 3)));
        auto variance = p.add_literal(migraphx::abs(migraphx::generate_literal(vars, 4)));
        p.add_instruction(migraphx::op::batch_norm_inference{}, x, scale, bias, mean, variance);
Paul's avatar
Paul committed
1767
1768
1769
1770
        return p;
    }
};

Paul's avatar
Paul committed
1771
struct test_batchnorm_inference : verify_program<test_batchnorm_inference>
wsttiger's avatar
wsttiger committed
1772
1773
1774
1775
1776
1777
{
    const size_t width    = 3;
    const size_t height   = 3;
    const size_t channels = 3;
    const size_t batches  = 4;

Paul's avatar
Paul committed
1778
    migraphx::program create_program() const
wsttiger's avatar
wsttiger committed
1779
    {
Paul's avatar
Paul committed
1780
        migraphx::program p;
wsttiger's avatar
wsttiger committed
1781

Paul's avatar
Paul committed
1782
1783
        migraphx::shape s{migraphx::shape::float_type, {batches, channels, height, width}};
        migraphx::shape vars{migraphx::shape::float_type, {channels}};
wsttiger's avatar
wsttiger committed
1784
        auto x        = p.add_parameter("x", s);
Paul's avatar
Paul committed
1785
1786
1787
1788
1789
        auto scale    = p.add_literal(migraphx::abs(migraphx::generate_literal(vars, 1)));
        auto bias     = p.add_literal(migraphx::abs(migraphx::generate_literal(vars, 2)));
        auto mean     = p.add_literal(migraphx::abs(migraphx::generate_literal(vars, 3)));
        auto variance = p.add_literal(migraphx::abs(migraphx::generate_literal(vars, 4)));
        p.add_instruction(migraphx::op::batch_norm_inference{}, x, scale, bias, mean, variance);
wsttiger's avatar
wsttiger committed
1790
1791
1792
1793
        return p;
    }
};

Khalique's avatar
Khalique committed
1794
1795
1796
1797
1798
struct test_clip : verify_program<test_clip>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
kahmed10's avatar
kahmed10 committed
1799
1800
1801
1802
1803
1804
        auto x       = p.add_parameter("x", migraphx::shape{migraphx::shape::float_type, {3}});
        auto min_val = p.add_literal(0.0f);
        auto max_val = p.add_literal(6.0f);
        min_val      = p.add_instruction(migraphx::op::multibroadcast{{3}}, min_val);
        max_val      = p.add_instruction(migraphx::op::multibroadcast{{3}}, max_val);
        p.add_instruction(migraphx::op::clip{}, x, min_val, max_val);
Khalique's avatar
Khalique committed
1805
1806
1807
1808
        return p;
    }
};

Paul's avatar
Paul committed
1809
struct test_conv_bn : verify_program<test_conv_bn>
Paul's avatar
Paul committed
1810
{
Paul's avatar
Paul committed
1811
    migraphx::program create_program() const
Paul's avatar
Paul committed
1812
    {
Paul's avatar
Paul committed
1813
        migraphx::program p;
Paul's avatar
Paul committed
1814

Paul's avatar
Paul committed
1815
1816
1817
        migraphx::shape xs{migraphx::shape::float_type, {1, 3, 224, 224}};
        migraphx::shape ws{migraphx::shape::float_type, {64, 3, 7, 7}};
        migraphx::shape vars{migraphx::shape::float_type, {64}};
Paul's avatar
Paul committed
1818
1819
        auto x        = p.add_parameter("x", xs);
        auto w        = p.add_parameter("w", ws);
Paul's avatar
Paul committed
1820
1821
1822
1823
1824
1825
        auto conv     = p.add_instruction(migraphx::op::convolution{{3, 3}, {2, 2}, {1, 1}}, x, w);
        auto scale    = p.add_literal(migraphx::abs(migraphx::generate_literal(vars, 1)));
        auto bias     = p.add_literal(migraphx::abs(migraphx::generate_literal(vars, 2)));
        auto mean     = p.add_literal(migraphx::abs(migraphx::generate_literal(vars, 3)));
        auto variance = p.add_literal(migraphx::abs(migraphx::generate_literal(vars, 4)));
        p.add_instruction(migraphx::op::batch_norm_inference{}, conv, scale, bias, mean, variance);
Paul's avatar
Paul committed
1826
1827
1828
1829
        return p;
    }
};

Paul's avatar
Paul committed
1830
struct test_conv_bn_relu_pooling : verify_program<test_conv_bn_relu_pooling>
Paul's avatar
Paul committed
1831
{
Paul's avatar
Paul committed
1832
    migraphx::program create_program() const
Paul's avatar
Paul committed
1833
    {
Paul's avatar
Paul committed
1834
        migraphx::program p;
Paul's avatar
Paul committed
1835

Paul's avatar
Paul committed
1836
1837
1838
        migraphx::shape xs{migraphx::shape::float_type, {1, 3, 224, 224}};
        migraphx::shape ws{migraphx::shape::float_type, {64, 3, 7, 7}};
        migraphx::shape vars{migraphx::shape::float_type, {64}};
Paul's avatar
Paul committed
1839
1840
        auto x        = p.add_parameter("x", xs);
        auto w        = p.add_parameter("w", ws);
Paul's avatar
Paul committed
1841
1842
1843
1844
1845
        auto conv     = p.add_instruction(migraphx::op::convolution{{3, 3}, {2, 2}, {1, 1}}, x, w);
        auto scale    = p.add_literal(migraphx::abs(migraphx::generate_literal(vars, 1)));
        auto bias     = p.add_literal(migraphx::abs(migraphx::generate_literal(vars, 2)));
        auto mean     = p.add_literal(migraphx::abs(migraphx::generate_literal(vars, 3)));
        auto variance = p.add_literal(migraphx::abs(migraphx::generate_literal(vars, 4)));
wsttiger's avatar
wsttiger committed
1846
        auto bn       = p.add_instruction(
Paul's avatar
Paul committed
1847
1848
1849
            migraphx::op::batch_norm_inference{}, conv, scale, bias, mean, variance);
        auto relu = p.add_instruction(migraphx::op::relu{}, bn);
        p.add_instruction(migraphx::op::pooling{"average", {1, 1}, {2, 2}, {3, 3}}, relu);
Paul's avatar
Paul committed
1850
1851
1852
1853
        return p;
    }
};

1854
1855
struct quant_conv : verify_program<quant_conv>
{
Shucai Xiao's avatar
Shucai Xiao committed
1856
1857
    migraphx::program create_program()
    {
1858
1859
1860
1861
1862
1863
1864
1865
1866
1867
1868
1869
        migraphx::program p;
        migraphx::shape a_shape{migraphx::shape::int8_type, {2, 3, 4, 4}};
        auto pa = p.add_parameter("a", a_shape);
        migraphx::shape c_shape{migraphx::shape::int8_type, {2, 3, 3, 3}};
        auto pc = p.add_parameter("c", c_shape);
        p.add_instruction(migraphx::op::quant_convolution{}, pa, pc);
        return p;
    }
};

struct quant_conv_default_mode : verify_program<quant_conv_default_mode>
{
Shucai Xiao's avatar
Shucai Xiao committed
1870
1871
    migraphx::program create_program()
    {
1872
1873
1874
1875
1876
        migraphx::program p;
        migraphx::shape a_shape{migraphx::shape::int8_type, {2, 3, 4, 4}};
        auto pa = p.add_parameter("a", a_shape);
        migraphx::shape c_shape{migraphx::shape::int8_type, {2, 3, 3, 3}};
        auto pc = p.add_parameter("c", c_shape);
Shucai Xiao's avatar
Shucai Xiao committed
1877
1878
1879
1880
        p.add_instruction(
            migraphx::op::quant_convolution{{{0, 0}}, {{1, 1}}, {{1, 1}}, migraphx::op::same},
            pa,
            pc);
1881
1882
1883
1884
1885
1886
        return p;
    }
};

struct quant_conv_valid_mode : verify_program<quant_conv_valid_mode>
{
Shucai Xiao's avatar
Shucai Xiao committed
1887
1888
    migraphx::program create_program()
    {
1889
1890
1891
1892
1893
        migraphx::program p;
        migraphx::shape a_shape{migraphx::shape::int8_type, {2, 3, 4, 4}};
        auto pa = p.add_parameter("a", a_shape);
        migraphx::shape c_shape{migraphx::shape::int8_type, {2, 3, 3, 3}};
        auto pc = p.add_parameter("c", c_shape);
Shucai Xiao's avatar
Shucai Xiao committed
1894
1895
1896
1897
        p.add_instruction(
            migraphx::op::quant_convolution{{{0, 0}}, {{1, 1}}, {{1, 1}}, migraphx::op::valid},
            pa,
            pc);
1898
1899
1900
1901
1902
1903
        return p;
    }
};

struct quant_conv_padding : verify_program<quant_conv_padding>
{
Shucai Xiao's avatar
Shucai Xiao committed
1904
1905
    migraphx::program create_program()
    {
1906
1907
1908
1909
1910
1911
1912
1913
1914
1915
1916
1917
        migraphx::program p;
        migraphx::shape a_shape{migraphx::shape::int8_type, {2, 3, 4, 4}};
        auto pa = p.add_parameter("a", a_shape);
        migraphx::shape c_shape{migraphx::shape::int8_type, {2, 3, 3, 3}};
        auto pc = p.add_parameter("c", c_shape);
        p.add_instruction(migraphx::op::quant_convolution{{{1, 1}}, {{1, 1}}}, pa, pc);
        return p;
    }
};

struct quant_conv_padding_stride : verify_program<quant_conv_padding_stride>
{
Shucai Xiao's avatar
Shucai Xiao committed
1918
1919
    migraphx::program create_program()
    {
1920
1921
1922
1923
1924
1925
1926
1927
1928
1929
1930
        migraphx::program p;
        migraphx::shape a_shape{migraphx::shape::int8_type, {2, 3, 4, 4}};
        auto pa = p.add_parameter("a", a_shape);
        migraphx::shape c_shape{migraphx::shape::int8_type, {2, 3, 3, 3}};
        auto pc = p.add_parameter("c", c_shape);
        p.add_instruction(migraphx::op::quant_convolution{{{1, 1}}, {{2, 2}}}, pa, pc);

        return p;
    }
};

Shucai Xiao's avatar
Shucai Xiao committed
1931
struct test_concat_axis_1 : verify_program<test_concat_axis_1>
1932
{
Paul's avatar
Paul committed
1933
    migraphx::program create_program() const
1934
    {
Paul's avatar
Paul committed
1935
        migraphx::program p;
Shucai Xiao's avatar
Shucai Xiao committed
1936
        int axis = 1;
Paul's avatar
Paul committed
1937
1938
1939
        migraphx::shape s0{migraphx::shape::int32_type, {2, 2}};
        migraphx::shape s1{migraphx::shape::int32_type, {2, 3}};
        migraphx::shape s2{migraphx::shape::int32_type, {2, 1}};
1940
1941
1942
        auto l0 = p.add_parameter("x", s0);
        auto l1 = p.add_parameter("y", s1);
        auto l2 = p.add_parameter("z", s2);
Paul's avatar
Paul committed
1943
        p.add_instruction(migraphx::op::concat{axis}, l0, l1, l2);
1944
1945
1946
1947
        return p;
    }
};

Shucai Xiao's avatar
Shucai Xiao committed
1948
struct test_concat_axis_neg_1 : verify_program<test_concat_axis_neg_1>
1949
{
Paul's avatar
Paul committed
1950
    migraphx::program create_program() const
1951
    {
Paul's avatar
Paul committed
1952
        migraphx::program p;
Shucai Xiao's avatar
Shucai Xiao committed
1953
1954
1955
1956
1957
1958
1959
1960
1961
1962
1963
1964
1965
1966
1967
1968
1969
1970
        int axis = -1;
        migraphx::shape s0{migraphx::shape::int32_type, {2, 2}};
        migraphx::shape s1{migraphx::shape::int32_type, {2, 3}};
        migraphx::shape s2{migraphx::shape::int32_type, {2, 1}};
        auto l0 = p.add_parameter("x", s0);
        auto l1 = p.add_parameter("y", s1);
        auto l2 = p.add_parameter("z", s2);
        p.add_instruction(migraphx::op::concat{axis}, l0, l1, l2);
        return p;
    }
};

struct test_concat_axis_0 : verify_program<test_concat_axis_0>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        int axis = 0;
Paul's avatar
Paul committed
1971
1972
1973
        migraphx::shape s0{migraphx::shape::int32_type, {2, 2}};
        migraphx::shape s1{migraphx::shape::int32_type, {3, 2}};
        migraphx::shape s2{migraphx::shape::int32_type, {1, 2}};
1974
1975
1976
        auto l0 = p.add_parameter("x", s0);
        auto l1 = p.add_parameter("y", s1);
        auto l2 = p.add_parameter("z", s2);
Paul's avatar
Paul committed
1977
        p.add_instruction(migraphx::op::concat{axis}, l0, l1, l2);
1978
1979
1980
1981
1982
1983
1984
1985
1986
        return p;
    }
};

struct test_concat_transpose : verify_program<test_concat_transpose>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
Shucai Xiao's avatar
Shucai Xiao committed
1987
        int axis = 1;
1988
1989
1990
1991
1992
1993
1994
1995
1996
1997
1998
1999
2000
2001
2002
2003
2004
        migraphx::shape s0{migraphx::shape::int32_type, {2, 2}};
        migraphx::shape s1{migraphx::shape::int32_type, {3, 2}};
        migraphx::shape s2{migraphx::shape::int32_type, {2, 4}};
        auto l0  = p.add_parameter("x", s0);
        auto lp1 = p.add_parameter("y", s1);
        auto l1  = p.add_instruction(migraphx::op::transpose{{1, 0}}, lp1);
        auto l2  = p.add_parameter("z", s2);
        p.add_instruction(migraphx::op::concat{axis}, l0, l1, l2);
        return p;
    }
};

struct test_concat_transpose2 : verify_program<test_concat_transpose2>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
Shucai Xiao's avatar
Shucai Xiao committed
2005
        int axis = 1;
2006
2007
2008
2009
2010
2011
2012
2013
2014
2015
2016
2017
2018
2019
2020
2021
2022
        migraphx::shape s0{migraphx::shape::int32_type, {2, 2}};
        migraphx::shape s1{migraphx::shape::int32_type, {2, 3}};
        migraphx::shape s2{migraphx::shape::int32_type, {5, 2}};
        auto l0  = p.add_parameter("x", s0);
        auto l1  = p.add_parameter("y", s1);
        auto lp2 = p.add_parameter("z", s2);
        auto l2  = p.add_instruction(migraphx::op::transpose{{1, 0}}, lp2);
        p.add_instruction(migraphx::op::concat{axis}, l0, l1, l2);
        return p;
    }
};

struct test_concat_transpose3 : verify_program<test_concat_transpose3>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
Shucai Xiao's avatar
Shucai Xiao committed
2023
        int axis = 1;
2024
2025
2026
2027
2028
2029
2030
2031
2032
        migraphx::shape s0{migraphx::shape::int32_type, {2, 2}};
        migraphx::shape s1{migraphx::shape::int32_type, {3, 2}};
        migraphx::shape s2{migraphx::shape::int32_type, {5, 2}};
        auto l0  = p.add_parameter("x", s0);
        auto lp1 = p.add_parameter("y", s1);
        auto l1  = p.add_instruction(migraphx::op::transpose{{1, 0}}, lp1);
        auto lp2 = p.add_parameter("z", s2);
        auto l2  = p.add_instruction(migraphx::op::transpose{{1, 0}}, lp2);
        p.add_instruction(migraphx::op::concat{axis}, l0, l1, l2);
2033
2034
2035
2036
        return p;
    }
};

Paul's avatar
Paul committed
2037
struct test_concat_relu : verify_program<test_concat_relu>
wsttiger's avatar
wsttiger committed
2038
{
Paul's avatar
Paul committed
2039
    migraphx::program create_program() const
wsttiger's avatar
wsttiger committed
2040
    {
Paul's avatar
Paul committed
2041
        migraphx::program p;
Shucai Xiao's avatar
Shucai Xiao committed
2042
        int axis = 0;
Paul's avatar
Paul committed
2043
2044
2045
        migraphx::shape s0{migraphx::shape::float_type, {2, 2}};
        migraphx::shape s1{migraphx::shape::float_type, {3, 2}};
        migraphx::shape s2{migraphx::shape::float_type, {1, 2}};
wsttiger's avatar
wsttiger committed
2046
2047
2048
        auto l0 = p.add_parameter("x", s0);
        auto l1 = p.add_parameter("y", s1);
        auto l2 = p.add_parameter("z", s2);
Paul's avatar
Paul committed
2049
2050
2051
2052
2053
        auto r0 = p.add_instruction(migraphx::op::relu{}, l0);
        auto r1 = p.add_instruction(migraphx::op::relu{}, l1);
        auto r2 = p.add_instruction(migraphx::op::relu{}, l2);
        auto c0 = p.add_instruction(migraphx::op::concat{axis}, r0, r1, r2);
        p.add_instruction(migraphx::op::relu{}, c0);
wsttiger's avatar
wsttiger committed
2054
2055
2056
2057
        return p;
    }
};

Paul's avatar
Paul committed
2058
struct test_pad : verify_program<test_pad>
2059
2060
2061
2062
2063
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s0{migraphx::shape::int32_type, {1, 96, 165, 165}};
Khalique's avatar
Khalique committed
2064
2065
2066
2067
        std::vector<int64_t> pads0 = {0, 0, 0, 0, 0, 0, 1, 1};
        std::vector<int64_t> pads1 = {0, 0, 0, 0, 1, 1, 1, 1};
        std::vector<int64_t> pads2 = {1, 1, 1, 1, 0, 0, 0, 0};
        std::vector<int64_t> pads3 = {1, 0, 1, 0, 1, 0, 2, 0};
Khalique's avatar
Khalique committed
2068
        auto l0                    = p.add_parameter("x", s0);
Khalique's avatar
Khalique committed
2069
2070
2071
        p.add_instruction(migraphx::op::pad{pads0}, l0);
        p.add_instruction(migraphx::op::pad{pads1}, l0);
        p.add_instruction(migraphx::op::pad{pads2}, l0);
Khalique's avatar
Khalique committed
2072
        p.add_instruction(migraphx::op::pad{pads3}, l0);
Khalique's avatar
Khalique committed
2073
        return p;
Khalique's avatar
Khalique committed
2074
    }
Khalique's avatar
Khalique committed
2075
};
Khalique's avatar
Khalique committed
2076

2077
2078
2079
2080
2081
2082
2083
2084
2085
2086
2087
2088
2089
struct test_pad_transposed : verify_program<test_pad_transposed>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::int32_type, {1, 224, 224, 3}};
        auto x = p.add_parameter("x", s);
        auto t = p.add_instruction(migraphx::op::transpose{{0, 3, 1, 2}}, x);
        p.add_instruction(migraphx::op::pad{{0, 0, 2, 2, 0, 0, 3, 3}}, t);
        return p;
    }
};

Khalique's avatar
Khalique committed
2090
2091
2092
2093
2094
2095
2096
2097
2098
2099
2100
2101
2102
2103
2104
2105
struct test_pad_int8 : verify_program<test_pad_int8>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        std::vector<int8_t> data0 = {0, 1, 2, 3};
        migraphx::shape s0{migraphx::shape::float_type, {2, 2}};
        auto l0 = p.add_literal(migraphx::literal{s0, data0});
        migraphx::op::pad op{};
        op.value = std::numeric_limits<int8_t>::lowest();
        op.pads  = {0, 0, 1, 1};
        p.add_instruction(op, l0);
        return p;
    }
};

2106
2107
2108
2109
2110
2111
2112
2113
2114
2115
2116
2117
2118
2119
2120
2121
2122
2123
2124
2125
2126
2127
2128
2129
2130
2131
2132
2133
2134
2135
2136
2137
2138
2139
struct test_pad_lowest : verify_program<test_pad_lowest>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        std::vector<migraphx::half> data0(4);
        std::iota(data0.begin(), data0.end(), 0);
        migraphx::shape s0{migraphx::shape::half_type, {2, 2}};
        auto l0 = p.add_literal(migraphx::literal{s0, data0});
        migraphx::op::pad op{};
        op.value = std::numeric_limits<float>::lowest();
        op.pads  = {0, 0, 1, 1};
        p.add_instruction(op, l0);
        return p;
    }
};

struct test_pad_highest : verify_program<test_pad_highest>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        std::vector<migraphx::half> data0(4);
        std::iota(data0.begin(), data0.end(), 0);
        migraphx::shape s0{migraphx::shape::half_type, {2, 2}};
        auto l0 = p.add_literal(migraphx::literal{s0, data0});
        migraphx::op::pad op{};
        op.value = std::numeric_limits<float>::max();
        op.pads  = {0, 0, 1, 1};
        p.add_instruction(op, l0);
        return p;
    }
};

Paul's avatar
Paul committed
2140
struct test_pooling_autopad : verify_program<test_pooling_autopad>
2141
2142
2143
2144
2145
2146
2147
2148
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s0{migraphx::shape::float_type, {1, 3, 63, 63}};
        auto l0 = p.add_parameter("x", s0);
        migraphx::op::pooling op{"max"};
        op.padding_mode = migraphx::op::padding_mode_t::same;
Khalique's avatar
Khalique committed
2149
2150
        op.lengths      = {2, 2};
        op.stride       = {2, 2};
2151
2152
2153
2154
2155
        p.add_instruction(op, l0);
        return p;
    }
};

Paul's avatar
Paul committed
2156
struct test_gather : verify_program<test_gather>
2157
2158
2159
2160
2161
2162
2163
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {3, 3}};
        migraphx::shape s_indices{migraphx::shape::int32_type, {2, 2}};
        std::vector<int> indices{1, 2, 2, 1};
Shucai Xiao's avatar
Shucai Xiao committed
2164
2165
        auto a0  = p.add_parameter("data", s);
        auto a1  = p.add_literal(migraphx::literal{s_indices, indices});
2166
        int axis = 0;
2167
        p.add_instruction(migraphx::op::gather{axis}, a0, a1);
2168
2169
2170
2171
        return p;
    }
};

Paul's avatar
Paul committed
2172
struct test_gather_neg_axis : verify_program<test_gather_neg_axis>
2173
2174
2175
2176
2177
2178
2179
2180
2181
2182
2183
2184
2185
2186
2187
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {3, 3}};
        migraphx::shape s_indices{migraphx::shape::int32_type, {2, 2}};
        std::vector<int> indices{1, 2, 2, 1};
        auto a0  = p.add_parameter("data", s);
        auto a1  = p.add_literal(migraphx::literal{s_indices, indices});
        int axis = -1;
        p.add_instruction(migraphx::op::gather{axis}, a0, a1);
        return p;
    }
};

Shucai Xiao's avatar
Shucai Xiao committed
2188
2189
2190
2191
2192
2193
2194
2195
2196
2197
2198
2199
2200
2201
2202
2203
struct test_gather_neg_indices : verify_program<test_gather_neg_indices>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {3, 3}};
        migraphx::shape s_indices{migraphx::shape::int32_type, {2, 2}};
        std::vector<int> indices{-2, -1, -1, -2};
        auto a0  = p.add_parameter("data", s);
        auto a1  = p.add_literal(migraphx::literal{s_indices, indices});
        int axis = -1;
        p.add_instruction(migraphx::op::gather{axis}, a0, a1);
        return p;
    }
};

Paul's avatar
Paul committed
2204
struct test_gather_scalar_output : verify_program<test_gather_scalar_output>
2205
2206
2207
2208
2209
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {3}};
2210
        migraphx::shape s_indices{migraphx::shape::int32_type};
2211
2212
2213
2214
2215
2216
2217
2218
2219
        std::vector<int> indices{1};
        auto a0  = p.add_parameter("data", s);
        auto a1  = p.add_literal(migraphx::literal{s_indices, indices});
        int axis = 0;
        p.add_instruction(migraphx::op::gather{axis}, a0, a1);
        return p;
    }
};

Paul's avatar
Paul committed
2220
struct test_gather_scalar_index : verify_program<test_gather_scalar_index>
2221
2222
2223
2224
2225
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {3, 3}};
2226
        migraphx::shape s_indices{migraphx::shape::int32_type};
2227
2228
2229
2230
2231
2232
2233
2234
2235
        std::vector<int> indices{1};
        auto a0  = p.add_parameter("data", s);
        auto a1  = p.add_literal(migraphx::literal{s_indices, indices});
        int axis = -1;
        p.add_instruction(migraphx::op::gather{axis}, a0, a1);
        return p;
    }
};

Paul's avatar
Paul committed
2236
struct test_gather_1d_index : verify_program<test_gather_1d_index>
2237
2238
2239
2240
2241
2242
2243
2244
2245
2246
2247
2248
2249
2250
2251
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {3, 3}};
        migraphx::shape s_indices{migraphx::shape::int32_type, {1}};
        std::vector<int> indices{1};
        auto a0  = p.add_parameter("data", s);
        auto a1  = p.add_literal(migraphx::literal{s_indices, indices});
        int axis = -1;
        p.add_instruction(migraphx::op::gather{axis}, a0, a1);
        return p;
    }
};

wsttiger's avatar
wsttiger committed
2252
2253
void manual_identity()
{
Paul's avatar
Paul committed
2254
    migraphx::program p;
wsttiger's avatar
wsttiger committed
2255
    std::vector<float> data0 = {0, 1, 2, 3};
Paul's avatar
Paul committed
2256
2257
2258
2259
2260
    migraphx::shape s0{migraphx::shape::float_type, {2, 2}};
    auto l0 = p.add_literal(migraphx::literal{s0, data0});
    p.add_instruction(migraphx::op::identity{}, l0);
    p.compile(migraphx::gpu::target{});
    migraphx::program::parameter_map m;
wsttiger's avatar
wsttiger committed
2261
2262
    for(auto&& x : p.get_parameter_shapes())
    {
Paul's avatar
Paul committed
2263
        m[x.first] = migraphx::gpu::to_gpu(migraphx::generate_argument(x.second));
wsttiger's avatar
wsttiger committed
2264
    }
2265
    auto result = migraphx::gpu::from_gpu(p.eval(m).back());
wsttiger's avatar
wsttiger committed
2266
2267
2268
2269
2270
    std::cout << result << std::endl;
}

void manual_test_concat_relu()
{
Paul's avatar
Paul committed
2271
    migraphx::program p;
Shucai Xiao's avatar
Shucai Xiao committed
2272
    int axis                 = 0;
wsttiger's avatar
wsttiger committed
2273
2274
2275
    std::vector<float> data0 = {0, 1, 2, 3};
    std::vector<float> data1 = {4, 5, 6, 7, 8, 9};
    std::vector<float> data2 = {10, 11};
Paul's avatar
Paul committed
2276
2277
2278
2279
2280
2281
2282
2283
2284
2285
2286
2287
2288
2289
    migraphx::shape s0{migraphx::shape::float_type, {2, 2}};
    migraphx::shape s1{migraphx::shape::float_type, {3, 2}};
    migraphx::shape s2{migraphx::shape::float_type, {1, 2}};
    auto l0 = p.add_literal(migraphx::literal{s0, data0});
    auto l1 = p.add_literal(migraphx::literal{s1, data1});
    auto l2 = p.add_literal(migraphx::literal{s2, data2});
    auto r0 = p.add_instruction(migraphx::op::relu{}, l0);
    auto r1 = p.add_instruction(migraphx::op::relu{}, l1);
    auto r2 = p.add_instruction(migraphx::op::relu{}, l2);
    auto c0 = p.add_instruction(migraphx::op::concat{axis}, r0, r1, r2);
    p.add_instruction(migraphx::op::relu{}, c0);

    p.compile(migraphx::gpu::target{});
    migraphx::program::parameter_map m;
wsttiger's avatar
wsttiger committed
2290
2291
    for(auto&& x : p.get_parameter_shapes())
    {
Paul's avatar
Paul committed
2292
        m[x.first] = migraphx::gpu::to_gpu(migraphx::generate_argument(x.second));
wsttiger's avatar
wsttiger committed
2293
    }
2294
    auto result = migraphx::gpu::from_gpu(p.eval(m).back());
wsttiger's avatar
wsttiger committed
2295
2296
2297
    std::cout << result << std::endl;
}

Paul's avatar
Paul committed
2298
struct test_conv_bn_relu_pooling2 : verify_program<test_conv_bn_relu_pooling2>
Paul's avatar
Paul committed
2299
{
Paul's avatar
Paul committed
2300
2301
    static migraphx::instruction_ref
    add_bn(migraphx::program& p, migraphx::instruction_ref x, std::size_t channels)
Paul's avatar
Paul committed
2302
    {
Paul's avatar
Paul committed
2303
        migraphx::shape vars{migraphx::shape::float_type, {channels}};
Paul's avatar
Paul committed
2304
2305
2306
2307
2308
        auto scale = p.add_literal(migraphx::abs(migraphx::generate_literal(vars, 1 + channels)));
        auto bias  = p.add_literal(migraphx::abs(migraphx::generate_literal(vars, 2 + channels)));
        auto mean  = p.add_literal(migraphx::abs(migraphx::generate_literal(vars, 3 + channels)));
        auto variance =
            p.add_literal(migraphx::abs(migraphx::generate_literal(vars, 4 + channels)));
wsttiger's avatar
wsttiger committed
2309
        return p.add_instruction(
Paul's avatar
Paul committed
2310
            migraphx::op::batch_norm_inference{}, x, scale, bias, mean, variance);
Paul's avatar
Paul committed
2311
    }
Paul's avatar
Paul committed
2312
    migraphx::program create_program() const
Paul's avatar
Paul committed
2313
    {
Paul's avatar
Paul committed
2314
        migraphx::program p;
Paul's avatar
Paul committed
2315

Paul's avatar
Paul committed
2316
2317
2318
2319
        migraphx::shape xs1{migraphx::shape::float_type, {1, 512, 7, 7}};
        migraphx::shape xs2{migraphx::shape::float_type, {1, 1024, 14, 14}};
        migraphx::shape ws1{migraphx::shape::float_type, {2048, 512, 1, 1}};
        migraphx::shape ws2{migraphx::shape::float_type, {2048, 1024, 1, 1}};
Paul's avatar
Paul committed
2320
2321
        auto x1    = p.add_parameter("x1", xs1);
        auto w1    = p.add_parameter("w1", ws1);
Paul's avatar
Paul committed
2322
        auto conv1 = p.add_instruction(migraphx::op::convolution{{0, 0}, {1, 1}, {1, 1}}, x1, w1);
Paul's avatar
Paul committed
2323
2324
2325
        auto bn1   = add_bn(p, conv1, 2048);
        auto x2    = p.add_parameter("x2", xs2);
        auto w2    = p.add_parameter("w2", ws2);
Paul's avatar
Paul committed
2326
        auto conv2 = p.add_instruction(migraphx::op::convolution{{0, 0}, {2, 2}, {1, 1}}, x2, w2);
Paul's avatar
Paul committed
2327
        auto bn2   = add_bn(p, conv2, 2048);
Paul's avatar
Paul committed
2328
2329
2330
        auto add   = p.add_instruction(migraphx::op::add{}, bn1, bn2);
        auto relu  = p.add_instruction(migraphx::op::relu{}, add);
        p.add_instruction(migraphx::op::pooling{"average", {1, 1}, {2, 2}, {3, 3}}, relu);
Paul's avatar
Paul committed
2331
2332
2333
2334
        return p;
    }
};

Shucai Xiao's avatar
Shucai Xiao committed
2335
2336
template <int Axis, migraphx::shape::type_t T>
struct test_logsoftmax : verify_program<test_logsoftmax<Axis, T>>
2337
2338
2339
2340
{
    migraphx::program create_program() const
    {
        migraphx::program p;
Shucai Xiao's avatar
Shucai Xiao committed
2341
        migraphx::shape s{T, {10, 4, 2080, 6}};
2342
        auto param = p.add_parameter("0", s);
Shucai Xiao's avatar
Shucai Xiao committed
2343
        p.add_instruction(migraphx::op::logsoftmax{Axis}, param);
2344
2345
2346
2347
2348

        return p;
    }
};

Shucai Xiao's avatar
Shucai Xiao committed
2349
2350
2351
2352
2353
2354
2355
2356
2357
2358
template struct test_logsoftmax<0, migraphx::shape::float_type>;
template struct test_logsoftmax<1, migraphx::shape::float_type>;
template struct test_logsoftmax<2, migraphx::shape::float_type>;
template struct test_logsoftmax<3, migraphx::shape::float_type>;
template struct test_logsoftmax<1, migraphx::shape::double_type>;
template struct test_logsoftmax<3, migraphx::shape::double_type>;
template struct test_logsoftmax<1, migraphx::shape::half_type>;
template struct test_logsoftmax<0, migraphx::shape::half_type>;
template struct test_logsoftmax<2, migraphx::shape::half_type>;
template struct test_logsoftmax<3, migraphx::shape::half_type>;
Paul's avatar
Paul committed
2359

Shucai Xiao's avatar
Shucai Xiao committed
2360
2361
2362
2363
2364
2365
2366
2367
2368
2369
2370
struct test_fp32_fp16_lall : verify_program<test_fp32_fp16_lall>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {2, 3}};
        std::vector<float> data(2 * 3);
        std::iota(data.begin(), data.end(), 1.0f);
        auto l1 = p.add_literal(migraphx::literal(s, data));
        auto l2 = p.add_parameter("p2", s);
        p.add_instruction(migraphx::op::add{}, l1, l2);
Shucai Xiao's avatar
Shucai Xiao committed
2371
        migraphx::quantize_fp16(p, {"all"});
Shucai Xiao's avatar
Shucai Xiao committed
2372
2373
2374
2375
2376
2377
2378
2379
2380
2381
2382
2383
2384
2385
2386
        return p;
    };
};

struct test_fp32_fp16_ladd : verify_program<test_fp32_fp16_ladd>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {2, 3}};
        std::vector<float> data(2 * 3);
        std::iota(data.begin(), data.end(), 1.0f);
        auto l1 = p.add_literal(migraphx::literal(s, data));
        auto l2 = p.add_parameter("p2", s);
        p.add_instruction(migraphx::op::add{}, l1, l2);
Shucai Xiao's avatar
Shucai Xiao committed
2387
        migraphx::quantize_fp16(p, {"add"});
Shucai Xiao's avatar
Shucai Xiao committed
2388
2389
2390
        return p;
    };
};
2391
2392
2393

struct test_fp32_fp16_add : verify_program<test_fp32_fp16_add>
{
Shucai Xiao's avatar
Shucai Xiao committed
2394
    migraphx::program create_program()
2395
2396
2397
2398
2399
2400
2401
2402
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {2, 3}};
        auto p1   = p.add_parameter("x", s);
        auto p2   = p.add_parameter("y", s);
        auto sum  = p.add_instruction(migraphx::op::add{}, p1, p2);
        auto diff = p.add_instruction(migraphx::op::sub{}, sum, p2);
        p.add_instruction(migraphx::op::add{}, diff, p1);
Shucai Xiao's avatar
Shucai Xiao committed
2403
        migraphx::quantize_fp16(p, {"add"});
2404
2405
2406
2407
2408
2409
2410

        return p;
    };
};

struct test_fp32_fp16_sub : verify_program<test_fp32_fp16_sub>
{
Shucai Xiao's avatar
Shucai Xiao committed
2411
    migraphx::program create_program()
2412
2413
2414
2415
2416
2417
2418
2419
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::float_type, {2, 3}};
        auto p1   = p.add_parameter("x", s);
        auto p2   = p.add_parameter("y", s);
        auto sum  = p.add_instruction(migraphx::op::add{}, p1, p2);
        auto diff = p.add_instruction(migraphx::op::sub{}, sum, p2);
        p.add_instruction(migraphx::op::add{}, diff, p1);
Shucai Xiao's avatar
Shucai Xiao committed
2420
        migraphx::quantize_fp16(p, {"sub"});
2421
2422
2423
2424
2425

        return p;
    };
};

Shucai Xiao's avatar
Shucai Xiao committed
2426
2427
template <class Op, int Axis, migraphx::shape::type_t T>
struct test_reduce_op_large : verify_program<test_reduce_op_large<Op, Axis, T>>
Paul's avatar
Paul committed
2428
2429
2430
2431
{
    migraphx::program create_program() const
    {
        migraphx::program p;
Shucai Xiao's avatar
Shucai Xiao committed
2432
        migraphx::shape s{T, {3, 1026, 4, 3}};
Paul's avatar
Paul committed
2433
        auto x = p.add_parameter("x", s);
Shucai Xiao's avatar
Shucai Xiao committed
2434
        p.add_instruction(Op{{1}}, x);
Paul's avatar
Paul committed
2435
2436
2437
2438
        return p;
    };
};

Shucai Xiao's avatar
Shucai Xiao committed
2439
template struct test_reduce_op_large<migraphx::op::reduce_max, 1, migraphx::shape::float_type>;
Shucai Xiao's avatar
Shucai Xiao committed
2440
template struct test_reduce_op_large<migraphx::op::reduce_mean, 1, migraphx::shape::float_type>;
Shucai Xiao's avatar
Shucai Xiao committed
2441
template struct test_reduce_op_large<migraphx::op::reduce_min, 1, migraphx::shape::float_type>;
Shucai Xiao's avatar
Shucai Xiao committed
2442
2443
template struct test_reduce_op_large<migraphx::op::reduce_prod, 2, migraphx::shape::float_type>;
template struct test_reduce_op_large<migraphx::op::reduce_sum, 1, migraphx::shape::float_type>;
Paul's avatar
Paul committed
2444

Shucai Xiao's avatar
Shucai Xiao committed
2445
2446
template <class Op, int Axis, migraphx::shape::type_t T>
struct test_reduce_op_small : verify_program<test_reduce_op_small<Op, Axis, T>>
Paul's avatar
Paul committed
2447
2448
2449
2450
{
    migraphx::program create_program() const
    {
        migraphx::program p;
Shucai Xiao's avatar
Shucai Xiao committed
2451
        migraphx::shape s{T, {3, 4, 8, 8}};
Paul's avatar
Paul committed
2452
        auto x = p.add_parameter("x", s);
Shucai Xiao's avatar
Shucai Xiao committed
2453
        p.add_instruction(Op{{1}}, x);
Paul's avatar
Paul committed
2454
2455
2456
        return p;
    };
};
Shucai Xiao's avatar
Shucai Xiao committed
2457
2458
2459
2460
2461
2462
2463
2464
2465
template struct test_reduce_op_small<migraphx::op::reduce_sum, 2, migraphx::shape::int32_type>;
template struct test_reduce_op_small<migraphx::op::reduce_mean, 2, migraphx::shape::int32_type>;
template struct test_reduce_op_small<migraphx::op::reduce_max, 2, migraphx::shape::int32_type>;
template struct test_reduce_op_small<migraphx::op::reduce_min, 2, migraphx::shape::int32_type>;

template struct test_reduce_op_small<migraphx::op::reduce_sum, 2, migraphx::shape::half_type>;
template struct test_reduce_op_small<migraphx::op::reduce_mean, 2, migraphx::shape::half_type>;
template struct test_reduce_op_small<migraphx::op::reduce_max, 2, migraphx::shape::half_type>;
template struct test_reduce_op_small<migraphx::op::reduce_min, 2, migraphx::shape::half_type>;
Shucai Xiao's avatar
Shucai Xiao committed
2466
template struct test_reduce_op_small<migraphx::op::reduce_prod, -2, migraphx::shape::half_type>;
Paul's avatar
Paul committed
2467

Khalique's avatar
Khalique committed
2468
2469
2470
2471
2472
struct test_rsqrt : verify_program<test_rsqrt>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
kahmed10's avatar
kahmed10 committed
2473
2474
2475
2476
2477
2478
2479
2480
        std::vector<size_t> input_lens{1, 3, 16, 16};
        migraphx::shape s{migraphx::shape::float_type, input_lens};
        auto x       = p.add_parameter("x", s);
        auto min_val = p.add_literal(1.0f);
        auto max_val = p.add_literal(std::numeric_limits<float>::max());
        min_val      = p.add_instruction(migraphx::op::multibroadcast{input_lens}, min_val);
        max_val      = p.add_instruction(migraphx::op::multibroadcast{input_lens}, max_val);
        auto l0      = p.add_instruction(migraphx::op::clip{}, x, min_val, max_val);
Khalique's avatar
Khalique committed
2481
2482
2483
2484
2485
        p.add_instruction(migraphx::op::rsqrt{}, l0);
        return p;
    };
};

2486
struct test_round : verify_program<test_round>
2487
2488
2489
2490
2491
{
    migraphx::program create_program() const
    {
        migraphx::program p;

2492
2493
2494
        migraphx::shape s{migraphx::shape::float_type, {2, 3, 4, 6}};
        auto param = p.add_parameter("x", s);
        p.add_instruction(migraphx::op::round{}, param);
2495
2496
2497
2498
        return p;
    };
};

Shucai Xiao's avatar
Shucai Xiao committed
2499
2500
2501
2502
2503
2504
2505
2506
2507
2508
2509
2510
2511
2512
2513
2514
2515
2516
2517
2518
2519
2520
2521
2522
2523
2524
struct test_ceil : verify_program<test_ceil>
{
    migraphx::program create_program() const
    {
        migraphx::program p;

        migraphx::shape s{migraphx::shape::double_type, {2, 3, 4, 6}};
        auto param = p.add_parameter("x", s);
        p.add_instruction(migraphx::op::ceil{}, param);
        return p;
    };
};

struct test_floor : verify_program<test_floor>
{
    migraphx::program create_program() const
    {
        migraphx::program p;

        migraphx::shape s{migraphx::shape::float_type, {2, 3, 4, 6}};
        auto param = p.add_parameter("x", s);
        p.add_instruction(migraphx::op::floor{}, param);
        return p;
    };
};

2525
struct test_convert : verify_program<test_convert>
2526
2527
2528
2529
{
    migraphx::program create_program() const
    {
        migraphx::program p;
2530
2531
2532
2533
2534
2535
2536
        migraphx::shape sa{migraphx::shape::float_type, {8, 24}};
        migraphx::shape sb{migraphx::shape::float_type, {24, 6}};
        auto pa = p.add_parameter("a", sa);
        auto pb = p.add_parameter("b", sb);
        auto ia = p.add_instruction(migraphx::op::convert{migraphx::shape::int8_type}, pa);
        auto ib = p.add_instruction(migraphx::op::convert{migraphx::shape::int8_type}, pb);
        p.add_instruction(migraphx::op::quant_dot{}, ia, ib);
2537

2538
        return p;
2539
    };
2540
2541
};

kahmed10's avatar
kahmed10 committed
2542
2543
2544
2545
2546
2547
2548
2549
2550
2551
2552
2553
struct test_recip : verify_program<test_recip>
{
    migraphx::program create_program() const
    {
        migraphx::program p;
        migraphx::shape s{migraphx::shape::double_type, {3}};
        auto x = p.add_parameter("x", s);
        p.add_instruction(migraphx::op::recip{}, x);
        return p;
    }
};

Shucai Xiao's avatar
Shucai Xiao committed
2554
2555
2556
2557
2558
2559
2560
2561
2562
2563
2564
2565
2566
struct test_neg : verify_program<test_neg>
{
    migraphx::program create_program() const
    {
        migraphx::program p;

        migraphx::shape s{migraphx::shape::double_type, {2, 3, 4, 6}};
        auto input = p.add_parameter("x", s);
        p.add_instruction(migraphx::op::neg{}, input);
        return p;
    };
};

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