matcher.cpp 26 KB
Newer Older
Paul's avatar
Paul committed
1
2
#include <migraphx/matcher.hpp>
#include <migraphx/iterator_for.hpp>
Paul's avatar
Paul committed
3
4
5
#include <test.hpp>
#include <basic_ops.hpp>

Paul's avatar
Paul committed
6
namespace match = migraphx::match;
Paul's avatar
Paul committed
7

Paul's avatar
Paul committed
8
9
MIGRAPHX_PRED_MATCHER(throws, migraphx::instruction_ref) { MIGRAPHX_THROW("Matcher throws"); }

Paul's avatar
Paul committed
10
template <class M>
Paul's avatar
Paul committed
11
migraphx::match::matcher_result find_match(migraphx::program& p, M&& m)
Paul's avatar
Paul committed
12
{
Paul's avatar
Paul committed
13
14
    migraphx::match::matcher_result result;
    for(auto ins : migraphx::iterator_for(p))
Paul's avatar
Paul committed
15
    {
Paul's avatar
Paul committed
16
        result = migraphx::match::match_instruction(p, ins, m);
Paul's avatar
Paul committed
17
18
19
20
21
22
23
24
        if(result.result != p.end())
            return result;
    }
    return result;
}

void match1()
{
Paul's avatar
Paul committed
25
    migraphx::program p;
Paul's avatar
Paul committed
26
    auto l = p.add_literal(1);
Paul's avatar
Paul committed
27
    auto m = match::standard_shape();
Paul's avatar
Paul committed
28
29
30
31
    auto r = find_match(p, m);
    EXPECT(bool{r.result == l});
}

Paul's avatar
Paul committed
32
TEST_CASE(match_name1)
Paul's avatar
Paul committed
33
{
Paul's avatar
Paul committed
34
    migraphx::program p;
Paul's avatar
Paul committed
35
36
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
Paul's avatar
Paul committed
37
38
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
39
    auto m = match::name("sum");
Paul's avatar
Paul committed
40
41
42
43
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum});
}

Paul's avatar
Paul committed
44
TEST_CASE(match_name2)
Paul's avatar
Paul committed
45
{
Paul's avatar
Paul committed
46
    migraphx::program p;
Paul's avatar
Paul committed
47
48
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
Paul's avatar
Paul committed
49
50
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
51
    auto m = match::name("min");
Paul's avatar
Paul committed
52
53
54
55
    auto r = find_match(p, m);
    EXPECT(bool{r.result == p.end()});
}

Paul's avatar
Paul committed
56
TEST_CASE(match_name3)
Paul's avatar
Paul committed
57
{
Paul's avatar
Paul committed
58
    migraphx::program p;
Paul's avatar
Paul committed
59
60
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
Paul's avatar
Paul committed
61
62
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
63
    auto m = match::name("sum")(match::standard_shape());
Paul's avatar
Paul committed
64
65
66
67
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum});
}

Paul's avatar
Paul committed
68
TEST_CASE(match_arg1)
Paul's avatar
Paul committed
69
{
Paul's avatar
Paul committed
70
    migraphx::program p;
Paul's avatar
Paul committed
71
72
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
Paul's avatar
Paul committed
73
74
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
75
    auto m = match::name("sum")(match::arg(0)(match::name("@literal")), match::standard_shape());
Paul's avatar
Paul committed
76
77
78
79
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum});
}

Paul's avatar
Paul committed
80
TEST_CASE(match_arg2)
Paul's avatar
Paul committed
81
{
Paul's avatar
Paul committed
82
    migraphx::program p;
Paul's avatar
Paul committed
83
84
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
Paul's avatar
Paul committed
85
86
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
87
    auto m = match::name("sum")(match::arg(0)(match::name("sum")), match::standard_shape());
Paul's avatar
Paul committed
88
89
90
91
    auto r = find_match(p, m);
    EXPECT(bool{r.result == p.end()});
}

Paul's avatar
Paul committed
92
TEST_CASE(match_arg3)
Paul's avatar
Paul committed
93
{
Paul's avatar
Paul committed
94
    migraphx::program p;
Paul's avatar
Paul committed
95
96
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
Paul's avatar
Paul committed
97
98
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
99
    auto m = match::name("sum")(match::arg(1)(match::name("@literal")), match::standard_shape());
Paul's avatar
Paul committed
100
101
102
103
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum});
}

Paul's avatar
Paul committed
104
TEST_CASE(match_arg4)
Paul's avatar
Paul committed
105
{
Paul's avatar
Paul committed
106
    migraphx::program p;
Paul's avatar
Paul committed
107
108
    auto one  = p.add_literal(1);
    auto two  = p.add_literal(2);
Paul's avatar
Paul committed
109
    auto sum  = p.add_instruction(sum_op{}, one, two);
Paul's avatar
Paul committed
110
    auto pass = p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
111
112
    auto m    = match::name("pass")(match::arg(0)(match::name("sum")), match::standard_shape());
    auto r    = find_match(p, m);
Paul's avatar
Paul committed
113
114
115
    EXPECT(bool{r.result == pass});
}

Paul's avatar
Paul committed
116
TEST_CASE(match_arg5)
Paul's avatar
Paul committed
117
{
Paul's avatar
Paul committed
118
    migraphx::program p;
Paul's avatar
Paul committed
119
120
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
Paul's avatar
Paul committed
121
122
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
123
    auto m = match::name("pass")(match::arg(1)(match::name("sum")), match::standard_shape());
Paul's avatar
Paul committed
124
125
126
127
    auto r = find_match(p, m);
    EXPECT(bool{r.result == p.end()});
}

Paul's avatar
Paul committed
128
TEST_CASE(match_arg6)
Paul's avatar
Paul committed
129
{
Paul's avatar
Paul committed
130
    migraphx::program p;
Paul's avatar
Paul committed
131
132
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
Paul's avatar
Paul committed
133
134
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
135
    auto m = match::name("sum")(match::arg(0)(match::name("@literal")));
Paul's avatar
Paul committed
136
137
138
139
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum});
}

Paul's avatar
Paul committed
140
TEST_CASE(match_arg7)
Paul's avatar
Paul committed
141
{
Paul's avatar
Paul committed
142
    migraphx::program p;
Paul's avatar
Paul committed
143
144
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
Paul's avatar
Paul committed
145
146
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
147
    auto m = match::name("sum")(match::arg(0)(match::name("@literal")),
Paul's avatar
Paul committed
148
                                match::arg(1)(match::name("@literal")));
Paul's avatar
Paul committed
149
150
151
152
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum});
}

Paul's avatar
Paul committed
153
154
155
156
157
158
159
160
TEST_CASE(match_arg8)
{
    migraphx::program p;
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
    auto m = match::name("sum")(match::all_of(match::arg(0)(match::name("@literal")),
Paul's avatar
Paul committed
161
                                              match::arg(1)(match::name("@literal"))),
Paul's avatar
Paul committed
162
163
164
165
166
                                match::standard_shape());
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum});
}

Paul's avatar
Paul committed
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
TEST_CASE(match_nargs1)
{
    migraphx::program p;
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
    auto m = match::name("sum")(match::nargs(2));
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum});
}

TEST_CASE(match_nargs2)
{
    migraphx::program p;
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
186
    auto m = match::name("sum")(match::nargs(2), match::standard_shape());
Paul's avatar
Paul committed
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum});
}

TEST_CASE(match_nargs3)
{
    migraphx::program p;
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
    auto m = match::name("sum")(match::all_of(match::nargs(2)));
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum});
}

Paul's avatar
Paul committed
203
TEST_CASE(match_args1)
Paul's avatar
Paul committed
204
{
Paul's avatar
Paul committed
205
    migraphx::program p;
Paul's avatar
Paul committed
206
207
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
Paul's avatar
Paul committed
208
209
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
210
211
    auto m = match::name("sum")(match::args(match::name("@literal"), match::name("@literal")),
                                match::standard_shape());
Paul's avatar
Paul committed
212
213
214
215
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum});
}

Paul's avatar
Paul committed
216
TEST_CASE(match_args2)
Paul's avatar
Paul committed
217
{
Paul's avatar
Paul committed
218
    migraphx::program p;
Paul's avatar
Paul committed
219
220
221
222
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
223
224
    auto m = match::name("sum")(match::args(match::name("@literal"), match::name("sum")),
                                match::standard_shape());
Paul's avatar
Paul committed
225
226
227
228
    auto r = find_match(p, m);
    EXPECT(bool{r.result == p.end()});
}

Paul's avatar
Paul committed
229
TEST_CASE(match_args3)
Paul's avatar
Paul committed
230
{
Paul's avatar
Paul committed
231
    migraphx::program p;
Paul's avatar
Paul committed
232
233
234
235
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
236
    auto m = match::name("sum")(match::args(match::name("@literal")), match::standard_shape());
Paul's avatar
Paul committed
237
    auto r = find_match(p, m);
238
    EXPECT(bool{r.result == p.end()});
Paul's avatar
Paul committed
239
240
}

Paul's avatar
Paul committed
241
TEST_CASE(match_args4)
Paul's avatar
Paul committed
242
{
Paul's avatar
Paul committed
243
    migraphx::program p;
Paul's avatar
Paul committed
244
245
    auto one  = p.add_literal(1);
    auto two  = p.add_literal(2);
Paul's avatar
Paul committed
246
247
248
    auto sum1 = p.add_instruction(sum_op{}, one, two);
    auto sum2 = p.add_instruction(sum_op{}, sum1, two);
    p.add_instruction(pass_op{}, sum2);
Paul's avatar
Paul committed
249
250
    auto m = match::name("sum")(match::args(match::name("sum"), match::name("@literal")),
                                match::standard_shape());
Paul's avatar
Paul committed
251
252
253
254
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum2});
}

Paul's avatar
Paul committed
255
TEST_CASE(match_args5)
Paul's avatar
Paul committed
256
{
Paul's avatar
Paul committed
257
    migraphx::program p;
Paul's avatar
Paul committed
258
259
260
261
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
262
263
    auto m = match::name("sum")(match::args(match::name("sum"), match::name("@literal")),
                                match::standard_shape());
Paul's avatar
Paul committed
264
265
266
267
    auto r = find_match(p, m);
    EXPECT(bool{r.result == p.end()});
}

Paul's avatar
Paul committed
268
TEST_CASE(match_args6)
Paul's avatar
Paul committed
269
{
Paul's avatar
Paul committed
270
    migraphx::program p;
Paul's avatar
Paul committed
271
272
273
    auto one  = p.add_literal(1);
    auto two  = p.add_literal(2);
    auto sum  = p.add_instruction(sum_op{}, one, two);
Paul's avatar
Paul committed
274
    auto pass = p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
275
276
    auto m    = match::name("pass")(match::args(match::name("sum")), match::standard_shape());
    auto r    = find_match(p, m);
Paul's avatar
Paul committed
277
278
279
    EXPECT(bool{r.result == pass});
}

Paul's avatar
Paul committed
280
TEST_CASE(match_args7)
Paul's avatar
Paul committed
281
{
Paul's avatar
Paul committed
282
    migraphx::program p;
Paul's avatar
Paul committed
283
284
285
    auto one  = p.add_literal(1);
    auto two  = p.add_literal(2);
    auto sum  = p.add_instruction(sum_op{}, one, two);
Paul's avatar
Paul committed
286
    auto pass = p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
287
    auto m    = match::name("pass")(match::args(match::name("sum")(match::args(
Paul's avatar
Paul committed
288
289
                                     match::name("@literal"), match::name("@literal")))),
                                 match::standard_shape());
Paul's avatar
Paul committed
290
    auto r    = find_match(p, m);
Paul's avatar
Paul committed
291
292
293
    EXPECT(bool{r.result == pass});
}

Paul's avatar
Paul committed
294
TEST_CASE(match_either_args1)
Paul's avatar
Paul committed
295
{
Paul's avatar
Paul committed
296
    migraphx::program p;
Paul's avatar
Paul committed
297
298
    auto one  = p.add_literal(1);
    auto two  = p.add_literal(2);
Paul's avatar
Paul committed
299
    auto sum1 = p.add_instruction(sum_op{}, one, two);
Paul's avatar
Paul committed
300
301
    auto sum2 = p.add_instruction(sum_op{}, sum1, two);
    p.add_instruction(pass_op{}, sum2);
Paul's avatar
Paul committed
302
303
304
    auto m =
        match::name("sum")(match::either_arg(0, 1)(match::name("sum"), match::name("@literal")));
    auto r = find_match(p, m);
Paul's avatar
Paul committed
305
306
307
    EXPECT(bool{r.result == sum2});
}

Paul's avatar
Paul committed
308
TEST_CASE(match_either_args2)
Paul's avatar
Paul committed
309
{
Paul's avatar
Paul committed
310
    migraphx::program p;
Paul's avatar
Paul committed
311
312
    auto one  = p.add_literal(1);
    auto two  = p.add_literal(2);
Paul's avatar
Paul committed
313
    auto sum1 = p.add_instruction(sum_op{}, one, two);
Paul's avatar
Paul committed
314
315
    auto sum2 = p.add_instruction(sum_op{}, sum1, two);
    p.add_instruction(pass_op{}, sum2);
Paul's avatar
Paul committed
316
317
318
    auto m =
        match::name("sum")(match::either_arg(0, 1)(match::name("@literal"), match::name("sum")));
    auto r = find_match(p, m);
Paul's avatar
Paul committed
319
320
321
    EXPECT(bool{r.result == sum2});
}

Paul's avatar
Paul committed
322
TEST_CASE(match_either_args3)
Paul's avatar
Paul committed
323
{
Paul's avatar
Paul committed
324
    migraphx::program p;
Paul's avatar
Paul committed
325
326
    auto one  = p.add_literal(1);
    auto two  = p.add_literal(2);
Paul's avatar
Paul committed
327
    auto sum1 = p.add_instruction(sum_op{}, one, two);
Paul's avatar
Paul committed
328
329
    auto sum2 = p.add_instruction(sum_op{}, sum1, two);
    p.add_instruction(pass_op{}, sum2);
Paul's avatar
Paul committed
330
331
332
    auto m =
        match::name("sum")(match::either_arg(0, 1)(match::name("pass"), match::name("@literal")));
    auto r = find_match(p, m);
Paul's avatar
Paul committed
333
334
335
    EXPECT(bool{r.result == p.end()});
}

Paul's avatar
Paul committed
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
TEST_CASE(match_either_args_any1)
{
    migraphx::program p;
    auto one  = p.add_literal(1);
    auto two  = p.add_literal(2);
    auto sum1 = p.add_instruction(sum_op{}, one, two);
    auto sum2 = p.add_instruction(sum_op{}, sum1, two);
    p.add_instruction(pass_op{}, sum2);
    auto m =
        match::name("sum")(match::either_arg(0, 1)(match::any().bind("x"), match::any().bind("y")));
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum1});
    EXPECT(bool{r.instructions.at("x") != r.instructions.at("y")});
}

TEST_CASE(match_either_args_any2)
{
    migraphx::program p;
    auto one  = p.add_literal(1);
    auto two  = p.add_literal(2);
    auto sum1 = p.add_instruction(sum_op{}, one, two);
    auto sum2 = p.add_instruction(sum_op{}, sum1, two);
    p.add_instruction(pass_op{}, sum2);
    auto m =
        match::name("sum")(match::either_arg(0, 1)(match::any().bind("x"), match::name("@literal").bind("y")));
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum1});
    EXPECT(bool{r.instructions.at("x") != r.instructions.at("y")});
}

TEST_CASE(match_either_args_any3)
{
    migraphx::program p;
    auto one  = p.add_literal(1);
    auto two  = p.add_literal(2);
    auto sum1 = p.add_instruction(sum_op{}, one, two);
    auto sum2 = p.add_instruction(sum_op{}, sum1, two);
    p.add_instruction(pass_op{}, sum2);
    auto m =
        match::name("sum")(match::either_arg(0, 1)(match::name("@literal").bind("x"), match::any().bind("y")));
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum1});
    EXPECT(bool{r.instructions.at("x") != r.instructions.at("y")});
}

TEST_CASE(match_either_args_any4)
{
    migraphx::program p;
    auto one  = p.add_literal(1);
    auto two  = p.add_literal(2);
    auto sum1 = p.add_instruction(sum_op{}, one, two);
    auto sum2 = p.add_instruction(sum_op{}, sum1, two);
    p.add_instruction(pass_op{}, sum2);
    auto m =
        match::name("sum")(match::either_arg(0, 1)(match::name("sum").bind("x"), match::any().bind("y")));
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum2});
    EXPECT(bool{r.instructions.at("x") != r.instructions.at("y")});
}

TEST_CASE(match_either_args_any5)
{
    migraphx::program p;
    auto one  = p.add_literal(1);
    auto two  = p.add_literal(2);
    auto sum1 = p.add_instruction(sum_op{}, one, two);
    auto sum2 = p.add_instruction(sum_op{}, sum1, two);
    p.add_instruction(pass_op{}, sum2);
    auto m =
        match::name("sum")(match::either_arg(0, 1)(match::any().bind("x"), match::name("sum").bind("y")));
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum2});
    EXPECT(bool{r.instructions.at("x") != r.instructions.at("y")});
}

Paul's avatar
Paul committed
411
TEST_CASE(match_all_of1)
Paul's avatar
Paul committed
412
{
Paul's avatar
Paul committed
413
    migraphx::program p;
Paul's avatar
Paul committed
414
415
416
417
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
418
    auto m = match::name("sum")(match::all_of(match::arg(0)(match::name("@literal")),
Paul's avatar
Paul committed
419
                                              match::arg(1)(match::name("@literal"))));
Paul's avatar
Paul committed
420
421
422
423
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum});
}

Paul's avatar
Paul committed
424
TEST_CASE(match_all_of2)
Paul's avatar
Paul committed
425
{
Paul's avatar
Paul committed
426
    migraphx::program p;
Paul's avatar
Paul committed
427
428
429
430
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
431
432
    auto m = match::name("sum")(
        match::all_of(match::arg(0)(match::name("sum")), match::arg(1)(match::name("@literal"))));
Paul's avatar
Paul committed
433
434
435
436
    auto r = find_match(p, m);
    EXPECT(bool{r.result == p.end()});
}

Paul's avatar
Paul committed
437
438
439
440
441
442
443
TEST_CASE(match_all_of3)
{
    migraphx::program p;
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
444
445
    auto m = match::name("sum")(match::all_of(match::all_of(
        match::arg(0)(match::name("@literal")), match::arg(1)(match::name("@literal")))));
Paul's avatar
Paul committed
446
447
448
449
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum});
}

Paul's avatar
Paul committed
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
TEST_CASE(match_lazy_any_of)
{
    migraphx::program p;
    auto one    = p.add_literal(1);
    p.add_instruction(pass_op{}, one);
    auto m = match::any_of(match::any(), throws());
    auto r = find_match(p, m);
    EXPECT(bool{r.result == one});
}

TEST_CASE(match_lazy_all_of)
{
    migraphx::program p;
    auto one    = p.add_literal(1);
    p.add_instruction(pass_op{}, one);
    auto m = match::all_of(match::none(), throws());
    auto r = find_match(p, m);
    EXPECT(bool{r.result == p.end()});
}

TEST_CASE(match_lazy_none_of)
{
    migraphx::program p;
    auto one    = p.add_literal(1);
    p.add_instruction(pass_op{}, one);
    auto m = match::none_of(match::any(), throws());
    auto r = find_match(p, m);
    EXPECT(bool{r.result == p.end()});
}

Paul's avatar
Paul committed
480
TEST_CASE(match_any_of1)
Paul's avatar
Paul committed
481
{
Paul's avatar
Paul committed
482
    migraphx::program p;
Paul's avatar
Paul committed
483
484
485
486
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
487
488
    auto m = match::name("sum")(
        match::any_of(match::arg(0)(match::name("sum")), match::arg(1)(match::name("@literal"))));
Paul's avatar
Paul committed
489
490
491
492
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum});
}

Paul's avatar
Paul committed
493
TEST_CASE(match_any_of2)
Paul's avatar
Paul committed
494
{
Paul's avatar
Paul committed
495
    migraphx::program p;
Paul's avatar
Paul committed
496
497
498
499
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
500
501
    auto m = match::name("sum")(
        match::any_of(match::arg(0)(match::name("sum")), match::arg(1)(match::name("sum"))));
Paul's avatar
Paul committed
502
503
504
505
    auto r = find_match(p, m);
    EXPECT(bool{r.result == p.end()});
}

Paul's avatar
Paul committed
506
507
508
509
510
511
512
513
514
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
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
TEST_CASE(match_any_of_lazy1)
{
    migraphx::program p;
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
    auto m = match::name("sum")(
        match::any_of(match::args(match::any(), match::any()).bind("x"), match::args(match::name("sum"), match::name("sum")).bind("y")));
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum});
    EXPECT(migraphx::contains(r.instructions, "x"));
    EXPECT(bool{r.instructions["x"] == sum});
    EXPECT(not migraphx::contains(r.instructions, "y"));
}

TEST_CASE(match_any_of_lazy2)
{
    migraphx::program p;
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
    auto m = match::name("sum")(
        match::any_of(match::args(match::name("@literal"), match::name("@literal")).bind("x"), match::args(match::any(), match::any()).bind("y")));
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum});
    EXPECT(migraphx::contains(r.instructions, "x"));
    EXPECT(bool{r.instructions["x"] == sum});
    EXPECT(not migraphx::contains(r.instructions, "y"));
}

TEST_CASE(match_any_of_lazy3)
{
    migraphx::program p;
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
    auto m = match::name("sum")(
        match::any_of(match::args(match::any(), match::any()).bind("x"), match::args(match::name("@literal"), match::name("@literal")).bind("y")));
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum});
    EXPECT(migraphx::contains(r.instructions, "x"));
    EXPECT(bool{r.instructions["x"] == sum});
    EXPECT(not migraphx::contains(r.instructions, "y"));
}

TEST_CASE(match_any_of_lazy4)
{
    migraphx::program p;
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
    auto m = match::name("sum")(
        match::any_of(match::args(match::name("@literal").bind("x1"), match::name("@literal").bind("y1")), match::args(match::any().bind("x2"), match::any().bind("y2"))));
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum});
    EXPECT(migraphx::contains(r.instructions, "x1"));
    EXPECT(migraphx::contains(r.instructions, "y1"));
    EXPECT(bool{r.instructions["x1"] == one});
    EXPECT(bool{r.instructions["y1"] == two});
    EXPECT(not migraphx::contains(r.instructions, "x2"));
    EXPECT(not migraphx::contains(r.instructions, "y2"));
}

TEST_CASE(match_any_of_lazy5)
{
    migraphx::program p;
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
    auto m = match::name("sum")(
        match::any_of(match::args(match::any().bind("x1"), match::any().bind("y1")), match::args(match::name("@literal").bind("x2"), match::name("@literal").bind("y2"))));
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum});
    EXPECT(migraphx::contains(r.instructions, "x1"));
    EXPECT(migraphx::contains(r.instructions, "y1"));
    EXPECT(bool{r.instructions["x1"] == one});
    EXPECT(bool{r.instructions["y1"] == two});
    EXPECT(not migraphx::contains(r.instructions, "x2"));
    EXPECT(not migraphx::contains(r.instructions, "y2"));
}

Paul's avatar
Paul committed
592
TEST_CASE(match_none_of1)
Paul's avatar
Paul committed
593
{
Paul's avatar
Paul committed
594
    migraphx::program p;
Paul's avatar
Paul committed
595
596
597
598
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
599
600
    auto m = match::name("sum")(
        match::none_of(match::arg(0)(match::name("sum")), match::arg(1)(match::name("sum"))));
Paul's avatar
Paul committed
601
602
603
604
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum});
}

Paul's avatar
Paul committed
605
TEST_CASE(match_none_of2)
Paul's avatar
Paul committed
606
{
Paul's avatar
Paul committed
607
    migraphx::program p;
Paul's avatar
Paul committed
608
609
610
611
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
612
    auto m = match::name("sum")(match::none_of(match::arg(0)(match::name("@literal")),
Paul's avatar
Paul committed
613
                                               match::arg(1)(match::name("@literal"))));
Paul's avatar
Paul committed
614
615
616
617
    auto r = find_match(p, m);
    EXPECT(bool{r.result == p.end()});
}

Paul's avatar
Paul committed
618
619
620
TEST_CASE(match_output1)
{
    migraphx::program p;
Paul's avatar
Paul committed
621
622
    auto one   = p.add_literal(1);
    auto two   = p.add_literal(2);
Paul's avatar
Paul committed
623
    auto minus = p.add_instruction(minus_op{}, two, one);
Paul's avatar
Paul committed
624
    auto sum   = p.add_instruction(sum_op{}, minus, two);
Paul's avatar
Paul committed
625
626
627
628
629
630
631
632
633
    p.add_instruction(pass_op{}, sum);
    auto m = match::name("minus")(match::output(match::name("sum")));
    auto r = find_match(p, m);
    EXPECT(bool{r.result == minus});
}

TEST_CASE(match_output2)
{
    migraphx::program p;
Paul's avatar
Paul committed
634
635
    auto one   = p.add_literal(1);
    auto two   = p.add_literal(2);
Paul's avatar
Paul committed
636
    auto minus = p.add_instruction(minus_op{}, two, one);
Paul's avatar
Paul committed
637
    auto sum   = p.add_instruction(sum_op{}, minus, two);
Paul's avatar
Paul committed
638
639
640
641
642
643
644
645
646
    p.add_instruction(pass_op{}, sum);
    auto m = match::name("@literal")(match::output(match::name("sum")));
    auto r = find_match(p, m);
    EXPECT(bool{r.result == p.end()});
}

TEST_CASE(match_skip_output1)
{
    migraphx::program p;
Paul's avatar
Paul committed
647
648
    auto one   = p.add_literal(1);
    auto two   = p.add_literal(2);
Paul's avatar
Paul committed
649
    auto minus = p.add_instruction(minus_op{}, two, one);
Paul's avatar
Paul committed
650
    auto sum   = p.add_instruction(sum_op{}, minus, two);
Paul's avatar
Paul committed
651
652
653
654
655
656
657
658
659
    p.add_instruction(pass_op{}, sum);
    auto m = match::name("minus")(match::skip_output(match::name("pass"))(match::name("sum")));
    auto r = find_match(p, m);
    EXPECT(bool{r.result == minus});
}

TEST_CASE(match_skip_output2)
{
    migraphx::program p;
Paul's avatar
Paul committed
660
661
662
    auto one        = p.add_literal(1);
    auto two        = p.add_literal(2);
    auto minus      = p.add_instruction(minus_op{}, two, one);
Paul's avatar
Paul committed
663
    auto minus_pass = p.add_instruction(pass_op{}, minus);
Paul's avatar
Paul committed
664
    auto sum        = p.add_instruction(sum_op{}, minus_pass, two);
Paul's avatar
Paul committed
665
666
667
668
669
670
671
672
673
    p.add_instruction(pass_op{}, sum);
    auto m = match::name("minus")(match::skip_output(match::name("pass"))(match::name("sum")));
    auto r = find_match(p, m);
    EXPECT(bool{r.result == minus});
}

TEST_CASE(match_skip_output3)
{
    migraphx::program p;
Paul's avatar
Paul committed
674
675
676
    auto one         = p.add_literal(1);
    auto two         = p.add_literal(2);
    auto minus       = p.add_instruction(minus_op{}, two, one);
Paul's avatar
Paul committed
677
678
679
    auto minus_pass1 = p.add_instruction(pass_op{}, minus);
    auto minus_pass2 = p.add_instruction(pass_op{}, minus_pass1);
    auto minus_pass3 = p.add_instruction(pass_op{}, minus_pass2);
Paul's avatar
Paul committed
680
    auto sum         = p.add_instruction(sum_op{}, minus_pass3, two);
Paul's avatar
Paul committed
681
682
683
684
685
686
687
688
689
    p.add_instruction(pass_op{}, sum);
    auto m = match::name("minus")(match::skip_output(match::name("pass"))(match::name("sum")));
    auto r = find_match(p, m);
    EXPECT(bool{r.result == minus});
}

TEST_CASE(match_skip_output4)
{
    migraphx::program p;
Paul's avatar
Paul committed
690
691
    auto one  = p.add_literal(1);
    auto two  = p.add_literal(2);
Paul's avatar
Paul committed
692
    auto pass = p.add_instruction(pass_op{}, one);
Paul's avatar
Paul committed
693
    auto sum  = p.add_instruction(sum_op{}, pass, two);
Paul's avatar
Paul committed
694
695
696
697
698
699
700
701
702
    p.add_instruction(pass_op{}, sum);
    auto m = match::name("@literal")(match::skip_output(match::name("pass"))(match::name("sum")));
    auto r = find_match(p, m);
    EXPECT(bool{r.result == two});
}

TEST_CASE(match_skip_output5)
{
    migraphx::program p;
Paul's avatar
Paul committed
703
704
    auto one  = p.add_literal(1);
    auto two  = p.add_literal(2);
Paul's avatar
Paul committed
705
706
707
708
709
710
711
712
713
714
715
716
717
    auto pass = p.add_instruction(pass_op{}, one);
    auto sum1 = p.add_instruction(sum_op{}, pass, two);
    auto sum2 = p.add_instruction(sum_op{}, sum1, one);
    auto sum3 = p.add_instruction(sum_op{}, sum2, two);
    p.add_instruction(pass_op{}, sum3);
    auto m = match::name("@literal")(match::skip_output(match::name("pass"))(match::name("sum")));
    auto r = find_match(p, m);
    EXPECT(bool{r.result == p.end()});
}

TEST_CASE(match_skip_output6)
{
    migraphx::program p;
Paul's avatar
Paul committed
718
719
    auto one   = p.add_literal(1);
    auto two   = p.add_literal(2);
Paul's avatar
Paul committed
720
    auto minus = p.add_instruction(minus_op{}, two, one);
Paul's avatar
Paul committed
721
722
723
    auto sum1  = p.add_instruction(sum_op{}, minus, two);
    auto sum2  = p.add_instruction(sum_op{}, sum1, one);
    auto sum3  = p.add_instruction(sum_op{}, sum2, two);
Paul's avatar
Paul committed
724
725
726
727
728
729
730
731
732
    p.add_instruction(pass_op{}, sum3);
    auto m = match::name("minus")(match::skip_output(match::name("pass"))(match::name("sum")));
    auto r = find_match(p, m);
    EXPECT(bool{r.result == minus});
}

TEST_CASE(match_skip_output7)
{
    migraphx::program p;
Paul's avatar
Paul committed
733
734
    auto one    = p.add_literal(1);
    auto two    = p.add_literal(2);
Paul's avatar
Paul committed
735
736
    auto minus1 = p.add_instruction(minus_op{}, two, one);
    auto minus2 = p.add_instruction(minus_op{}, two, minus1);
Paul's avatar
Paul committed
737
    auto sum    = p.add_instruction(sum_op{}, one, minus2);
Paul's avatar
Paul committed
738
739
740
741
742
743
    p.add_instruction(pass_op{}, sum);
    auto m = match::name("minus")(match::skip_output(match::name("pass"))(match::name("minus")));
    auto r = find_match(p, m);
    EXPECT(bool{r.result == minus1});
}

Paul's avatar
Paul committed
744
TEST_CASE(match_bind1)
Paul's avatar
Paul committed
745
{
Paul's avatar
Paul committed
746
    migraphx::program p;
Paul's avatar
Paul committed
747
748
749
    auto one  = p.add_literal(1);
    auto two  = p.add_literal(2);
    auto sum  = p.add_instruction(sum_op{}, one, two);
Paul's avatar
Paul committed
750
    auto pass = p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
751
    auto m    = match::name("pass")(
Paul's avatar
Paul committed
752
753
754
                 match::args(match::name("sum")(match::args(match::name("@literal").bind("one"),
                                                            match::name("@literal").bind("two")))
                                 .bind("sum")),
Paul's avatar
Paul committed
755
                 match::standard_shape())
Paul's avatar
Paul committed
756
                 .bind("pass");
Paul's avatar
Paul committed
757
758
759
760
761
762
763
764
    auto r = find_match(p, m);
    EXPECT(bool{r.instructions.at("one") == one});
    EXPECT(bool{r.instructions.at("two") == two});
    EXPECT(bool{r.instructions.at("sum") == sum});
    EXPECT(bool{r.instructions.at("pass") == pass});
    EXPECT(bool{r.result == pass});
}

Paul's avatar
Paul committed
765
766
struct match_find_sum
{
Paul's avatar
Paul committed
767
    migraphx::instruction_ref ins;
Paul's avatar
Paul committed
768
    auto matcher() const { return match::name("sum"); }
Paul's avatar
Paul committed
769

Paul's avatar
Paul committed
770
771
772
773
    void apply(migraphx::program&, const match::matcher_result& r) const
    {
        EXPECT(bool{r.result == ins});
    }
Paul's avatar
Paul committed
774
775
776
777
};

struct match_find_literal
{
Paul's avatar
Paul committed
778
    migraphx::instruction_ref ins;
Paul's avatar
Paul committed
779
    auto matcher() const { return match::name("@literal"); }
Paul's avatar
Paul committed
780

Paul's avatar
Paul committed
781
    void apply(migraphx::program&, const match::matcher_result& r) const
Paul's avatar
Paul committed
782
783
784
785
786
787
    {
        EXPECT(bool{r.result != ins});
        EXPECT(r.result->name() == "@literal");
    }
};

Paul's avatar
Paul committed
788
TEST_CASE(match_finder)
Paul's avatar
Paul committed
789
{
Paul's avatar
Paul committed
790
    migraphx::program p;
Paul's avatar
Paul committed
791
792
793
794
    auto one = p.add_literal(1);
    auto two = p.add_literal(2);
    auto sum = p.add_instruction(sum_op{}, one, two);
    p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
795
    match::find_matches(p, match_find_sum{sum}, match_find_literal{sum});
Paul's avatar
Paul committed
796
797
}

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