matcher.cpp 26.1 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
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);
Paul's avatar
Paul committed
359
360
    auto m = match::name("sum")(
        match::either_arg(0, 1)(match::any().bind("x"), match::name("@literal").bind("y")));
Paul's avatar
Paul committed
361
362
363
364
365
366
367
368
369
370
371
372
373
    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);
Paul's avatar
Paul committed
374
375
    auto m = match::name("sum")(
        match::either_arg(0, 1)(match::name("@literal").bind("x"), match::any().bind("y")));
Paul's avatar
Paul committed
376
377
378
379
380
381
382
383
384
385
386
387
388
    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);
Paul's avatar
Paul committed
389
390
    auto m = match::name("sum")(
        match::either_arg(0, 1)(match::name("sum").bind("x"), match::any().bind("y")));
Paul's avatar
Paul committed
391
392
393
394
395
396
397
398
399
400
401
402
403
    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);
Paul's avatar
Paul committed
404
405
    auto m = match::name("sum")(
        match::either_arg(0, 1)(match::any().bind("x"), match::name("sum").bind("y")));
Paul's avatar
Paul committed
406
407
408
409
410
    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
TEST_CASE(match_lazy_any_of)
{
    migraphx::program p;
Paul's avatar
Paul committed
453
    auto one = p.add_literal(1);
Paul's avatar
Paul committed
454
455
456
457
458
459
460
461
462
    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;
Paul's avatar
Paul committed
463
    auto one = p.add_literal(1);
Paul's avatar
Paul committed
464
465
466
467
468
469
470
471
472
    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;
Paul's avatar
Paul committed
473
    auto one = p.add_literal(1);
Paul's avatar
Paul committed
474
475
476
477
478
479
    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
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")(
Paul's avatar
Paul committed
514
515
        match::any_of(match::args(match::any(), match::any()).bind("x"),
                      match::args(match::name("sum"), match::name("sum")).bind("y")));
Paul's avatar
Paul committed
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
    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")(
Paul's avatar
Paul committed
531
532
        match::any_of(match::args(match::name("@literal"), match::name("@literal")).bind("x"),
                      match::args(match::any(), match::any()).bind("y")));
Paul's avatar
Paul committed
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
    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")(
Paul's avatar
Paul committed
548
549
        match::any_of(match::args(match::any(), match::any()).bind("x"),
                      match::args(match::name("@literal"), match::name("@literal")).bind("y")));
Paul's avatar
Paul committed
550
551
552
553
554
555
556
557
558
559
560
561
562
563
    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);
Paul's avatar
Paul committed
564
565
566
    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"))));
Paul's avatar
Paul committed
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
    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);
Paul's avatar
Paul committed
584
585
586
    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"))));
Paul's avatar
Paul committed
587
588
589
590
591
592
593
594
595
596
    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
597
TEST_CASE(match_none_of1)
Paul's avatar
Paul committed
598
{
Paul's avatar
Paul committed
599
    migraphx::program p;
Paul's avatar
Paul committed
600
601
602
603
    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
604
605
    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
606
607
608
609
    auto r = find_match(p, m);
    EXPECT(bool{r.result == sum});
}

Paul's avatar
Paul committed
610
TEST_CASE(match_none_of2)
Paul's avatar
Paul committed
611
{
Paul's avatar
Paul committed
612
    migraphx::program p;
Paul's avatar
Paul committed
613
614
615
616
    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
617
    auto m = match::name("sum")(match::none_of(match::arg(0)(match::name("@literal")),
Paul's avatar
Paul committed
618
                                               match::arg(1)(match::name("@literal"))));
Paul's avatar
Paul committed
619
620
621
622
    auto r = find_match(p, m);
    EXPECT(bool{r.result == p.end()});
}

Paul's avatar
Paul committed
623
624
625
TEST_CASE(match_output1)
{
    migraphx::program p;
Paul's avatar
Paul committed
626
627
    auto one   = p.add_literal(1);
    auto two   = p.add_literal(2);
Paul's avatar
Paul committed
628
    auto minus = p.add_instruction(minus_op{}, two, one);
Paul's avatar
Paul committed
629
    auto sum   = p.add_instruction(sum_op{}, minus, two);
Paul's avatar
Paul committed
630
631
632
633
634
635
636
637
638
    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
639
640
    auto one   = p.add_literal(1);
    auto two   = p.add_literal(2);
Paul's avatar
Paul committed
641
    auto minus = p.add_instruction(minus_op{}, two, one);
Paul's avatar
Paul committed
642
    auto sum   = p.add_instruction(sum_op{}, minus, two);
Paul's avatar
Paul committed
643
644
645
646
647
648
649
650
651
    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
652
653
    auto one   = p.add_literal(1);
    auto two   = p.add_literal(2);
Paul's avatar
Paul committed
654
    auto minus = p.add_instruction(minus_op{}, two, one);
Paul's avatar
Paul committed
655
    auto sum   = p.add_instruction(sum_op{}, minus, two);
Paul's avatar
Paul committed
656
657
658
659
660
661
662
663
664
    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
665
666
667
    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
668
    auto minus_pass = p.add_instruction(pass_op{}, minus);
Paul's avatar
Paul committed
669
    auto sum        = p.add_instruction(sum_op{}, minus_pass, two);
Paul's avatar
Paul committed
670
671
672
673
674
675
676
677
678
    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
679
680
681
    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
682
683
684
    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
685
    auto sum         = p.add_instruction(sum_op{}, minus_pass3, two);
Paul's avatar
Paul committed
686
687
688
689
690
691
692
693
694
    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
695
696
    auto one  = p.add_literal(1);
    auto two  = p.add_literal(2);
Paul's avatar
Paul committed
697
    auto pass = p.add_instruction(pass_op{}, one);
Paul's avatar
Paul committed
698
    auto sum  = p.add_instruction(sum_op{}, pass, two);
Paul's avatar
Paul committed
699
700
701
702
703
704
705
706
707
    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
708
709
    auto one  = p.add_literal(1);
    auto two  = p.add_literal(2);
Paul's avatar
Paul committed
710
711
712
713
714
715
716
717
718
719
720
721
722
    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
723
724
    auto one   = p.add_literal(1);
    auto two   = p.add_literal(2);
Paul's avatar
Paul committed
725
    auto minus = p.add_instruction(minus_op{}, two, one);
Paul's avatar
Paul committed
726
727
728
    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
729
730
731
732
733
734
735
736
737
    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
738
739
    auto one    = p.add_literal(1);
    auto two    = p.add_literal(2);
Paul's avatar
Paul committed
740
741
    auto minus1 = p.add_instruction(minus_op{}, two, one);
    auto minus2 = p.add_instruction(minus_op{}, two, minus1);
Paul's avatar
Paul committed
742
    auto sum    = p.add_instruction(sum_op{}, one, minus2);
Paul's avatar
Paul committed
743
744
745
746
747
748
    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
749
TEST_CASE(match_bind1)
Paul's avatar
Paul committed
750
{
Paul's avatar
Paul committed
751
    migraphx::program p;
Paul's avatar
Paul committed
752
753
754
    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
755
    auto pass = p.add_instruction(pass_op{}, sum);
Paul's avatar
Paul committed
756
    auto m    = match::name("pass")(
Paul's avatar
Paul committed
757
758
759
                 match::args(match::name("sum")(match::args(match::name("@literal").bind("one"),
                                                            match::name("@literal").bind("two")))
                                 .bind("sum")),
Paul's avatar
Paul committed
760
                 match::standard_shape())
Paul's avatar
Paul committed
761
                 .bind("pass");
Paul's avatar
Paul committed
762
763
764
765
766
767
768
769
    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
770
771
struct match_find_sum
{
Paul's avatar
Paul committed
772
    migraphx::instruction_ref ins;
Paul's avatar
Paul committed
773
    auto matcher() const { return match::name("sum"); }
Paul's avatar
Paul committed
774

Paul's avatar
Paul committed
775
776
777
778
    void apply(migraphx::program&, const match::matcher_result& r) const
    {
        EXPECT(bool{r.result == ins});
    }
Paul's avatar
Paul committed
779
780
781
782
};

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

Paul's avatar
Paul committed
786
    void apply(migraphx::program&, const match::matcher_result& r) const
Paul's avatar
Paul committed
787
788
789
790
791
792
    {
        EXPECT(bool{r.result != ins});
        EXPECT(r.result->name() == "@literal");
    }
};

Paul's avatar
Paul committed
793
TEST_CASE(match_finder)
Paul's avatar
Paul committed
794
{
Paul's avatar
Paul committed
795
    migraphx::program p;
Paul's avatar
Paul committed
796
797
798
799
    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
800
    match::find_matches(p, match_find_sum{sum}, match_find_literal{sum});
Paul's avatar
Paul committed
801
802
}

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