4
4
// spv.FAdd
5
5
//===----------------------------------------------------------------------===//
6
6
7
- func @fadd_scalar (%arg: f32 ) -> f32 {
7
+ func.func @fadd_scalar (%arg: f32 ) -> f32 {
8
8
// CHECK: spv.FAdd
9
9
%0 = spv.FAdd %arg , %arg : f32
10
10
return %0 : f32
@@ -16,7 +16,7 @@ func @fadd_scalar(%arg: f32) -> f32 {
16
16
// spv.FDiv
17
17
//===----------------------------------------------------------------------===//
18
18
19
- func @fdiv_scalar (%arg: f32 ) -> f32 {
19
+ func.func @fdiv_scalar (%arg: f32 ) -> f32 {
20
20
// CHECK: spv.FDiv
21
21
%0 = spv.FDiv %arg , %arg : f32
22
22
return %0 : f32
@@ -28,7 +28,7 @@ func @fdiv_scalar(%arg: f32) -> f32 {
28
28
// spv.FMod
29
29
//===----------------------------------------------------------------------===//
30
30
31
- func @fmod_scalar (%arg: f32 ) -> f32 {
31
+ func.func @fmod_scalar (%arg: f32 ) -> f32 {
32
32
// CHECK: spv.FMod
33
33
%0 = spv.FMod %arg , %arg : f32
34
34
return %0 : f32
@@ -40,37 +40,37 @@ func @fmod_scalar(%arg: f32) -> f32 {
40
40
// spv.FMul
41
41
//===----------------------------------------------------------------------===//
42
42
43
- func @fmul_scalar (%arg: f32 ) -> f32 {
43
+ func.func @fmul_scalar (%arg: f32 ) -> f32 {
44
44
// CHECK: spv.FMul
45
45
%0 = spv.FMul %arg , %arg : f32
46
46
return %0 : f32
47
47
}
48
48
49
- func @fmul_vector (%arg: vector <4 xf32 >) -> vector <4 xf32 > {
49
+ func.func @fmul_vector (%arg: vector <4 xf32 >) -> vector <4 xf32 > {
50
50
// CHECK: spv.FMul
51
51
%0 = spv.FMul %arg , %arg : vector <4 xf32 >
52
52
return %0 : vector <4 xf32 >
53
53
}
54
54
55
55
// -----
56
56
57
- func @fmul_i32 (%arg: i32 ) -> i32 {
57
+ func.func @fmul_i32 (%arg: i32 ) -> i32 {
58
58
// expected-error @+1 {{operand #0 must be 16/32/64-bit float or vector of 16/32/64-bit float values}}
59
59
%0 = spv.FMul %arg , %arg : i32
60
60
return %0 : i32
61
61
}
62
62
63
63
// -----
64
64
65
- func @fmul_bf16 (%arg: bf16 ) -> bf16 {
65
+ func.func @fmul_bf16 (%arg: bf16 ) -> bf16 {
66
66
// expected-error @+1 {{operand #0 must be 16/32/64-bit float or vector of 16/32/64-bit float values}}
67
67
%0 = spv.FMul %arg , %arg : bf16
68
68
return %0 : bf16
69
69
}
70
70
71
71
// -----
72
72
73
- func @fmul_tensor (%arg: tensor <4 xf32 >) -> tensor <4 xf32 > {
73
+ func.func @fmul_tensor (%arg: tensor <4 xf32 >) -> tensor <4 xf32 > {
74
74
// expected-error @+1 {{operand #0 must be 16/32/64-bit float or vector of 16/32/64-bit float values}}
75
75
%0 = spv.FMul %arg , %arg : tensor <4 xf32 >
76
76
return %0 : tensor <4 xf32 >
@@ -82,7 +82,7 @@ func @fmul_tensor(%arg: tensor<4xf32>) -> tensor<4xf32> {
82
82
// spv.FNegate
83
83
//===----------------------------------------------------------------------===//
84
84
85
- func @fnegate_scalar (%arg: f32 ) -> f32 {
85
+ func.func @fnegate_scalar (%arg: f32 ) -> f32 {
86
86
// CHECK: spv.FNegate
87
87
%0 = spv.FNegate %arg : f32
88
88
return %0 : f32
@@ -94,7 +94,7 @@ func @fnegate_scalar(%arg: f32) -> f32 {
94
94
// spv.FRem
95
95
//===----------------------------------------------------------------------===//
96
96
97
- func @frem_scalar (%arg: f32 ) -> f32 {
97
+ func.func @frem_scalar (%arg: f32 ) -> f32 {
98
98
// CHECK: spv.FRem
99
99
%0 = spv.FRem %arg , %arg : f32
100
100
return %0 : f32
@@ -106,7 +106,7 @@ func @frem_scalar(%arg: f32) -> f32 {
106
106
// spv.FSub
107
107
//===----------------------------------------------------------------------===//
108
108
109
- func @fsub_scalar (%arg: f32 ) -> f32 {
109
+ func.func @fsub_scalar (%arg: f32 ) -> f32 {
110
110
// CHECK: spv.FSub
111
111
%0 = spv.FSub %arg , %arg : f32
112
112
return %0 : f32
@@ -118,7 +118,7 @@ func @fsub_scalar(%arg: f32) -> f32 {
118
118
// spv.IAdd
119
119
//===----------------------------------------------------------------------===//
120
120
121
- func @iadd_scalar (%arg: i32 ) -> i32 {
121
+ func.func @iadd_scalar (%arg: i32 ) -> i32 {
122
122
// CHECK: spv.IAdd
123
123
%0 = spv.IAdd %arg , %arg : i32
124
124
return %0 : i32
@@ -130,7 +130,7 @@ func @iadd_scalar(%arg: i32) -> i32 {
130
130
// spv.IMul
131
131
//===----------------------------------------------------------------------===//
132
132
133
- func @imul_scalar (%arg: i32 ) -> i32 {
133
+ func.func @imul_scalar (%arg: i32 ) -> i32 {
134
134
// CHECK: spv.IMul
135
135
%0 = spv.IMul %arg , %arg : i32
136
136
return %0 : i32
@@ -142,7 +142,7 @@ func @imul_scalar(%arg: i32) -> i32 {
142
142
// spv.ISub
143
143
//===----------------------------------------------------------------------===//
144
144
145
- func @isub_scalar (%arg: i32 ) -> i32 {
145
+ func.func @isub_scalar (%arg: i32 ) -> i32 {
146
146
// CHECK: spv.ISub
147
147
%0 = spv.ISub %arg , %arg : i32
148
148
return %0 : i32
@@ -154,7 +154,7 @@ func @isub_scalar(%arg: i32) -> i32 {
154
154
// spv.SDiv
155
155
//===----------------------------------------------------------------------===//
156
156
157
- func @sdiv_scalar (%arg: i32 ) -> i32 {
157
+ func.func @sdiv_scalar (%arg: i32 ) -> i32 {
158
158
// CHECK: spv.SDiv
159
159
%0 = spv.SDiv %arg , %arg : i32
160
160
return %0 : i32
@@ -166,7 +166,7 @@ func @sdiv_scalar(%arg: i32) -> i32 {
166
166
// spv.SMod
167
167
//===----------------------------------------------------------------------===//
168
168
169
- func @smod_scalar (%arg: i32 ) -> i32 {
169
+ func.func @smod_scalar (%arg: i32 ) -> i32 {
170
170
// CHECK: spv.SMod
171
171
%0 = spv.SMod %arg , %arg : i32
172
172
return %0 : i32
@@ -178,7 +178,7 @@ func @smod_scalar(%arg: i32) -> i32 {
178
178
// spv.SNegate
179
179
//===----------------------------------------------------------------------===//
180
180
181
- func @snegate_scalar (%arg: i32 ) -> i32 {
181
+ func.func @snegate_scalar (%arg: i32 ) -> i32 {
182
182
// CHECK: spv.SNegate
183
183
%0 = spv.SNegate %arg : i32
184
184
return %0 : i32
@@ -189,7 +189,7 @@ func @snegate_scalar(%arg: i32) -> i32 {
189
189
// spv.SRem
190
190
//===----------------------------------------------------------------------===//
191
191
192
- func @srem_scalar (%arg: i32 ) -> i32 {
192
+ func.func @srem_scalar (%arg: i32 ) -> i32 {
193
193
// CHECK: spv.SRem
194
194
%0 = spv.SRem %arg , %arg : i32
195
195
return %0 : i32
@@ -201,7 +201,7 @@ func @srem_scalar(%arg: i32) -> i32 {
201
201
// spv.UDiv
202
202
//===----------------------------------------------------------------------===//
203
203
204
- func @udiv_scalar (%arg: i32 ) -> i32 {
204
+ func.func @udiv_scalar (%arg: i32 ) -> i32 {
205
205
// CHECK: spv.UDiv
206
206
%0 = spv.UDiv %arg , %arg : i32
207
207
return %0 : i32
@@ -213,7 +213,7 @@ func @udiv_scalar(%arg: i32) -> i32 {
213
213
// spv.UMod
214
214
//===----------------------------------------------------------------------===//
215
215
216
- func @umod_scalar (%arg: i32 ) -> i32 {
216
+ func.func @umod_scalar (%arg: i32 ) -> i32 {
217
217
// CHECK: spv.UMod
218
218
%0 = spv.UMod %arg , %arg : i32
219
219
return %0 : i32
@@ -224,23 +224,23 @@ func @umod_scalar(%arg: i32) -> i32 {
224
224
// spv.VectorTimesScalar
225
225
//===----------------------------------------------------------------------===//
226
226
227
- func @vector_times_scalar (%vector: vector <4 xf32 >, %scalar: f32 ) -> vector <4 xf32 > {
227
+ func.func @vector_times_scalar (%vector: vector <4 xf32 >, %scalar: f32 ) -> vector <4 xf32 > {
228
228
// CHECK: spv.VectorTimesScalar %{{.+}}, %{{.+}} : (vector<4xf32>, f32) -> vector<4xf32>
229
229
%0 = spv.VectorTimesScalar %vector , %scalar : (vector <4 xf32 >, f32 ) -> vector <4 xf32 >
230
230
return %0 : vector <4 xf32 >
231
231
}
232
232
233
233
// -----
234
234
235
- func @vector_times_scalar (%vector: vector <4 xf32 >, %scalar: f16 ) -> vector <4 xf32 > {
235
+ func.func @vector_times_scalar (%vector: vector <4 xf32 >, %scalar: f16 ) -> vector <4 xf32 > {
236
236
// expected-error @+1 {{scalar operand and result element type match}}
237
237
%0 = spv.VectorTimesScalar %vector , %scalar : (vector <4 xf32 >, f16 ) -> vector <4 xf32 >
238
238
return %0 : vector <4 xf32 >
239
239
}
240
240
241
241
// -----
242
242
243
- func @vector_times_scalar (%vector: vector <4 xf32 >, %scalar: f32 ) -> vector <3 xf32 > {
243
+ func.func @vector_times_scalar (%vector: vector <4 xf32 >, %scalar: f32 ) -> vector <3 xf32 > {
244
244
// expected-error @+1 {{vector operand and result type mismatch}}
245
245
%0 = spv.VectorTimesScalar %vector , %scalar : (vector <4 xf32 >, f32 ) -> vector <3 xf32 >
246
246
return %0 : vector <3 xf32 >
0 commit comments