1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
// RUN: mlir-opt %s | mlir-opt | FileCheck %s
// CHECK-LABEL: @binary_ops
// CHECK-SAME: %[[A:.*]]: index, %[[B:.*]]: index
func.func @binary_ops(%a: index, %b: index) {
// CHECK-NEXT: index.add %[[A]], %[[B]]
%0 = index.add %a, %b
// CHECK-NEXT: index.sub %[[A]], %[[B]]
%1 = index.sub %a, %b
// CHECK-NEXT: index.mul %[[A]], %[[B]]
%2 = index.mul %a, %b
// CHECK-NEXT: index.divs %[[A]], %[[B]]
%3 = index.divs %a, %b
// CHECK-NEXT: index.divu %[[A]], %[[B]]
%4 = index.divu %a, %b
// CHECK-NEXT: index.ceildivs %[[A]], %[[B]]
%5 = index.ceildivs %a, %b
// CHECK-NEXT: index.ceildivu %[[A]], %[[B]]
%6 = index.ceildivu %a, %b
// CHECK-NEXT: index.floordivs %[[A]], %[[B]]
%7 = index.floordivs %a, %b
// CHECK-NEXT: index.rems %[[A]], %[[B]]
%8 = index.rems %a, %b
// CHECK-NEXT: index.remu %[[A]], %[[B]]
%9 = index.remu %a, %b
// CHECK-NEXT: index.maxs %[[A]], %[[B]]
%10 = index.maxs %a, %b
// CHECK-NEXT: index.maxu %[[A]], %[[B]]
%11 = index.maxu %a, %b
// CHECK-NEXT: index.mins %[[A]], %[[B]]
%12 = index.mins %a, %b
// CHECK-NEXT: index.minu %[[A]], %[[B]]
%13 = index.minu %a, %b
// CHECK-NEXT: index.shl %[[A]], %[[B]]
%14 = index.shl %a, %b
// CHECK-NEXT: index.shrs %[[A]], %[[B]]
%15 = index.shrs %a, %b
// CHECK-NEXT: index.shru %[[A]], %[[B]]
%16 = index.shru %a, %b
// CHECK-NEXT: index.and %[[A]], %[[B]]
%17 = index.and %a, %b
// CHECK-NEXT: index.or %[[A]], %[[B]]
%18 = index.or %a, %b
// CHECK-NEXT: index.xor %[[A]], %[[B]]
%19 = index.xor %a, %b
return
}
// CHECK-LABEL: @cmp_op
// CHECK-SAME: %[[A:.*]]: index, %[[B:.*]]: index
func.func @cmp_op(%a: index, %b: index) {
// CHECK-NEXT: index.cmp eq(%[[A]], %[[B]])
%0 = index.cmp eq(%a, %b)
// CHECK-NEXT: index.cmp ne(%[[A]], %[[B]])
%1 = index.cmp ne(%a, %b)
// CHECK-NEXT: index.cmp slt(%[[A]], %[[B]])
%2 = index.cmp slt(%a, %b)
// CHECK-NEXT: index.cmp sle(%[[A]], %[[B]])
%3 = index.cmp sle(%a, %b)
// CHECK-NEXT: index.cmp sgt(%[[A]], %[[B]])
%4 = index.cmp sgt(%a, %b)
// CHECK-NEXT: index.cmp sge(%[[A]], %[[B]])
%5 = index.cmp sge(%a, %b)
// CHECK-NEXT: index.cmp ult(%[[A]], %[[B]])
%6 = index.cmp ult(%a, %b)
// CHECK-NEXT: index.cmp ule(%[[A]], %[[B]])
%7 = index.cmp ule(%a, %b)
// CHECK-NEXT: index.cmp ugt(%[[A]], %[[B]])
%8 = index.cmp ugt(%a, %b)
// CHECK-NEXT: index.cmp uge(%[[A]], %[[B]])
%9 = index.cmp uge(%a, %b)
return
}
// CHECK-LABEL: @sizeof_op
func.func @sizeof_op() {
// CHECK: index.sizeof
%0 = index.sizeof
return
}
// CHECK-LABEL: @constant_op
func.func @constant_op() {
// CHECK-NEXT: %idx0 = index.constant 0
%0 = index.constant 0
// CHECK-NEXT: %idx1 = index.constant 1
%1 = index.constant 1
// CHECK-NEXT: %idx42 = index.constant 42
%2 = index.constant 42
return
}
// CHECK-LABEL: @bool_constant_op
func.func @bool_constant_op() {
// CHECK-NEXT: %true = index.bool.constant true
%0 = index.bool.constant true
// CHECK-NEXT: %false = index.bool.constant false
%1 = index.bool.constant false
return
}
// CHECK-LABEL: @cast_op
// CHECK-SAME: %[[A:.*]]: index, %[[B:.*]]: i32, %[[C:.*]]: i64
func.func @cast_op(%a: index, %b: i32, %c: i64) {
// CHECK-NEXT: index.casts %[[A]] : index to i64
%0 = index.casts %a : index to i64
// CHECK-NEXT: index.casts %[[B]] : i32 to index
%1 = index.casts %b : i32 to index
// CHECK-NEXT: index.casts %[[C]] : i64 to index
%2 = index.casts %c : i64 to index
// CHECK-NEXT: index.castu %[[A]] : index to i64
%3 = index.castu %a : index to i64
// CHECK-NEXT: index.castu %[[B]] : i32 to index
%4 = index.castu %b : i32 to index
// CHECK-NEXT: index.castu %[[C]] : i64 to index
%5 = index.castu %c : i64 to index
return
}
|
