xref: /llvm-project/mlir/utils/tree-sitter-mlir/dialect/arith.js (revision 8a6e54c9b3080f1b8e1a925bf1a39730223b99f9)
1'use strict';
2
3module.exports = {
4  arith_dialect : $ => choice(
5                    // operation ::= `arith.constant` attr-dict $value
6                    seq('arith.constant',
7                        field('attributes', optional($.attribute)),
8                        field('value', $._literal_and_type)),
9
10                    // operation ::= `arith.addi` $lhs `,` $rhs attr-dict `:`
11                    // type($result) operation ::= `arith.subi` $lhs `,` $rhs
12                    // attr-dict `:` type($result) operation ::= `arith.divsi`
13                    // $lhs `,` $rhs attr-dict `:` type($result) operation ::=
14                    // `arith.divui` $lhs `,` $rhs attr-dict `:` type($result)
15                    // operation ::= `arith.ceildivsi` $lhs `,` $rhs attr-dict
16                    // `:` type($result) operation ::= `arith.ceildivui` $lhs
17                    // `,` $rhs attr-dict `:` type($result) operation ::=
18                    // `arith.floordivsi` $lhs `,` $rhs attr-dict `:`
19                    // type($result) operation ::= `arith.remsi` $lhs `,` $rhs
20                    // attr-dict `:` type($result) operation ::= `arith.remui`
21                    // $lhs `,` $rhs attr-dict `:` type($result) operation ::=
22                    // `arith.muli` $lhs `,` $rhs attr-dict `:` type($result)
23                    // operation ::= `arith.mulsi_extended` $lhs `,` $rhs
24                    // attr-dict `:` type($lhs) operation ::=
25                    // `arith.mului_extended` $lhs `,` $rhs attr-dict `:`
26                    // type($lhs) operation ::= `arith.andi` $lhs `,` $rhs
27                    // attr-dict `:` type($result) operation ::= `arith.ori`
28                    // $lhs `,` $rhs attr-dict `:` type($result) operation ::=
29                    // `arith.xori` $lhs `,` $rhs attr-dict `:` type($result)
30                    // operation ::= `arith.maxsi` $lhs `,` $rhs attr-dict `:`
31                    // type($result) operation ::= `arith.maxui` $lhs `,` $rhs
32                    // attr-dict `:` type($result) operation ::= `arith.minsi`
33                    // $lhs `,` $rhs attr-dict `:` type($result) operation ::=
34                    // `arith.minui` $lhs `,` $rhs attr-dict `:` type($result)
35                    // operation ::= `arith.shli` $lhs `,` $rhs attr-dict `:`
36                    // type($result) operation ::= `arith.shrsi` $lhs `,` $rhs
37                    // attr-dict `:` type($result) operation ::= `arith.shrui`
38                    // $lhs `,` $rhs attr-dict `:` type($result)
39                    seq(choice('arith.addi', 'arith.subi', 'arith.divsi',
40                               'arith.divui', 'arith.ceildivsi',
41                               'arith.ceildivui', 'arith.floordivsi',
42                               'arith.remsi', 'arith.remui', 'arith.muli',
43                               'arith.mulsi_extended', 'arith.mului_extended',
44                               'arith.andi', 'arith.ori', 'arith.xori',
45                               'arith.maxsi', 'arith.maxui', 'arith.minsi',
46                               'arith.minui', 'arith.shli', 'arith.shrsi',
47                               'arith.shrui'),
48                        field('lhs', $.value_use), ',',
49                        field('rhs', $.value_use),
50                        field('attributes', optional($.attribute)),
51                        field('return', $._type_annotation)),
52
53                    // operation ::= `arith.addui_extended` $lhs `,` $rhs
54                    // attr-dict `:` type($sum)
55                    //                `,` type($overflow)
56                    seq('arith.addui_extended', field('lhs', $.value_use), ',',
57                        field('rhs', $.value_use),
58                        field('attributes', optional($.attribute)),
59                        field('return', seq(':', $.type, ',', $.type))),
60
61                    // operation ::= `arith.addf` $lhs `,` $rhs (`fastmath` ``
62                    // $fastmath^)?
63                    //                attr-dict `:` type($result)
64                    // operation ::= `arith.divf` $lhs `,` $rhs (`fastmath` ``
65                    // $fastmath^)?
66                    //                attr-dict `:` type($result)
67                    // operation ::= `arith.maximumf` $lhs `,` $rhs (`fastmath` ``
68                    // $fastmath^)?
69                    //                attr-dict `:` type($result)
70                    // operation ::= `arith.minimumf` $lhs `,` $rhs (`fastmath` ``
71                    // $fastmath^)?
72                    //                attr-dict `:` type($result)
73                    // operation ::= `arith.mulf` $lhs `,` $rhs (`fastmath` ``
74                    // $fastmath^)?
75                    //                attr-dict `:` type($result)
76                    // operation ::= `arith.remf` $lhs `,` $rhs (`fastmath` ``
77                    // $fastmath^)?
78                    //                attr-dict `:` type($result)
79                    // operation ::= `arith.subf` $lhs `,` $rhs (`fastmath` ``
80                    // $fastmath^)?
81                    //                attr-dict `:` type($result)
82                    seq(choice('arith.addf', 'arith.divf', 'arith.maximumf',
83                               'arith.minimumf', 'arith.mulf', 'arith.remf',
84                               'arith.subf'),
85                        field('lhs', $.value_use), ',',
86                        field('rhs', $.value_use),
87                        field('fastmath', optional($.fastmath_attr)),
88                        field('attributes', optional($.attribute)),
89                        field('return', $._type_annotation)),
90
91                    // operation ::= `arith.negf` $operand (`fastmath` ``
92                    // $fastmath^)?
93                    //                attr-dict `:` type($result)
94                    seq(choice('arith.negf'), field('operand', $.value_use),
95                        field('fastmath', optional($.fastmath_attr)),
96                        field('attributes', optional($.attribute)),
97                        field('return', $._type_annotation)),
98
99                    // operation ::= `arith.cmpi` $predicate `,` $lhs `,` $rhs
100                    // attr-dict `:` type($lhs) operation ::= `arith.cmpf`
101                    // $predicate `,` $lhs `,` $rhs attr-dict `:` type($lhs)
102                    seq(choice('arith.cmpi', 'arith.cmpf'),
103                        field('predicate',
104                              choice('eq', 'ne', 'oeq', 'olt', 'ole', 'ogt',
105                                     'oge', 'slt', 'sle', 'sgt', 'sge', 'ult',
106                                     'ule', 'ugt', 'uge', $.string_literal)),
107                        ',', field('lhs', $.value_use), ',',
108                        field('rhs', $.value_use),
109                        field('attributes', optional($.attribute)),
110                        field('return', $._type_annotation)),
111
112                    // operation ::= `arith.extf` $in attr-dict `:` type($in)
113                    // `to` type($out) operation ::= `arith.extsi` $in attr-dict
114                    // `:` type($in) `to` type($out) operation ::= `arith.extui`
115                    // $in attr-dict `:` type($in) `to` type($out) operation ::=
116                    // `arith.fptosi` $in attr-dict `:` type($in) `to`
117                    // type($out) operation ::= `arith.fptoui` $in attr-dict `:`
118                    // type($in) `to` type($out) operation ::=
119                    // `arith.index_cast` $in attr-dict `:` type($in) `to`
120                    // type($out) operation ::= `arith.index_castui` $in
121                    // attr-dict `:` type($in) `to` type($out) operation ::=
122                    // `arith.sitofp` $in attr-dict `:` type($in) `to`
123                    // type($out) operation ::= `arith.uitofp` $in attr-dict `:`
124                    // type($in) `to` type($out) operation ::= `arith.bitcast`
125                    // $in attr-dict `:` type($in) `to` type($out) operation ::=
126                    // `arith.truncf` $in attr-dict `:` type($in) `to`
127                    // type($out) operation ::= `arith.trunci` $in attr-dict `:`
128                    // type($in) `to` type($out)
129                    seq(choice('arith.extf', 'arith.extsi', 'arith.extui',
130                               'arith.fptosi', 'arith.fptoui',
131                               'arith.index_cast', 'arith.index_castui',
132                               'arith.sitofp', 'arith.uitofp', 'arith.bitcast',
133                               'arith.truncf', 'arith.trunci'),
134                        field('in', $.value_use),
135                        field('attributes', optional($.attribute)),
136                        field('return', $._type_annotation)),
137
138                    seq('arith.select', field('cond', $.value_use), ',',
139                        field('trueblk', $.value_use), ',',
140                        field('falseblk', $.value_use),
141                        field('return', $._type_annotation)))
142}
143