blob: a9966af4b586cd6f19eb7d1741602b69a0955520 [file] [log] [blame]
Dan Gohman10e730a2015-06-29 23:51:55 +00001// WebAssemblyInstrSIMD.td - WebAssembly SIMD codegen support -*- tablegen -*-//
2//
Chandler Carruth2946cd72019-01-19 08:50:56 +00003// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
Dan Gohman10e730a2015-06-29 23:51:55 +00006//
7//===----------------------------------------------------------------------===//
JF Bastien5ca0bac2015-07-10 18:23:10 +00008///
9/// \file
Adrian Prantl5f8f34e42018-05-01 15:54:18 +000010/// WebAssembly SIMD operand code-gen constructs.
JF Bastien5ca0bac2015-07-10 18:23:10 +000011///
Dan Gohman10e730a2015-06-29 23:51:55 +000012//===----------------------------------------------------------------------===//
13
Heejin Ahnd9a6de32018-10-09 22:23:39 +000014// Instructions requiring HasSIMD128 and the simd128 prefix byte
15multiclass SIMD_I<dag oops_r, dag iops_r, dag oops_s, dag iops_s,
16 list<dag> pattern_r, string asmstr_r = "",
17 string asmstr_s = "", bits<32> simdop = -1> {
18 defm "" : I<oops_r, iops_r, oops_s, iops_s, pattern_r, asmstr_r, asmstr_s,
19 !or(0xfd00, !and(0xff, simdop))>,
20 Requires<[HasSIMD128]>;
21}
22
Thomas Lively0ff82ac2018-10-13 07:09:10 +000023defm "" : ARGUMENT<V128, v16i8>;
24defm "" : ARGUMENT<V128, v8i16>;
25defm "" : ARGUMENT<V128, v4i32>;
26defm "" : ARGUMENT<V128, v2i64>;
27defm "" : ARGUMENT<V128, v4f32>;
28defm "" : ARGUMENT<V128, v2f64>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +000029
30// Constrained immediate argument types
Thomas Lively22442922018-08-21 21:03:18 +000031foreach SIZE = [8, 16] in
Thomas Livelyffde98d2018-10-13 16:58:03 +000032def ImmI#SIZE : ImmLeaf<i32,
Thomas Lively9a484382019-01-31 23:22:39 +000033 "return -(1 << ("#SIZE#" - 1)) <= Imm && Imm < (1 << ("#SIZE#" - 1));"
Thomas Livelyffde98d2018-10-13 16:58:03 +000034>;
Heejin Ahna0fd9c32018-08-14 18:53:27 +000035foreach SIZE = [2, 4, 8, 16, 32] in
36def LaneIdx#SIZE : ImmLeaf<i32, "return 0 <= Imm && Imm < "#SIZE#";">;
Derek Schuff51ed1312018-08-07 21:24:01 +000037
Heejin Ahnd9a6de32018-10-09 22:23:39 +000038//===----------------------------------------------------------------------===//
Thomas Lively4ddd2252018-11-09 01:49:19 +000039// Load and store
40//===----------------------------------------------------------------------===//
41
42// Load: v128.load
Thomas Livelya9b3d1f2019-09-25 00:15:59 +000043let mayLoad = 1, UseNamedOperandTable = 1 in
44defm LOAD_V128 :
45 SIMD_I<(outs V128:$dst), (ins P2Align:$p2align, offset32_op:$off, I32:$addr),
46 (outs), (ins P2Align:$p2align, offset32_op:$off), [],
47 "v128.load\t$dst, ${off}(${addr})$p2align",
48 "v128.load\t$off$p2align", 0>;
Thomas Lively4ddd2252018-11-09 01:49:19 +000049
50// Def load and store patterns from WebAssemblyInstrMemory.td for vector types
Thomas Livelya9b3d1f2019-09-25 00:15:59 +000051foreach vec_t = [v16i8, v8i16, v4i32, v2i64, v4f32, v2f64] in {
52def : LoadPatNoOffset<vec_t, load, LOAD_V128>;
53def : LoadPatImmOff<vec_t, load, regPlusImm, LOAD_V128>;
54def : LoadPatImmOff<vec_t, load, or_is_add, LOAD_V128>;
55def : LoadPatOffsetOnly<vec_t, load, LOAD_V128>;
56def : LoadPatGlobalAddrOffOnly<vec_t, load, LOAD_V128>;
Thomas Lively4ddd2252018-11-09 01:49:19 +000057}
58
Thomas Lively99d3dd22019-09-23 20:42:12 +000059// vNxM.load_splat
60multiclass SIMDLoadSplat<string vec, bits<32> simdop> {
61 let mayLoad = 1, UseNamedOperandTable = 1,
62 Predicates = [HasUnimplementedSIMD128] in
63 defm LOAD_SPLAT_#vec :
64 SIMD_I<(outs V128:$dst), (ins P2Align:$p2align, offset32_op:$off, I32:$addr),
65 (outs), (ins P2Align:$p2align, offset32_op:$off), [],
66 vec#".load_splat\t$dst, ${off}(${addr})$p2align",
67 vec#".load_splat\t$off$p2align", simdop>;
68}
69
70defm "" : SIMDLoadSplat<"v8x16", 194>;
71defm "" : SIMDLoadSplat<"v16x8", 195>;
72defm "" : SIMDLoadSplat<"v32x4", 196>;
73defm "" : SIMDLoadSplat<"v64x2", 197>;
74
75def wasm_load_splat_t : SDTypeProfile<1, 1, []>;
76def wasm_load_splat : SDNode<"WebAssemblyISD::LOAD_SPLAT", wasm_load_splat_t>;
77
78foreach args = [["v16i8", "i32", "extloadi8"], ["v8i16", "i32", "extloadi16"],
79 ["v4i32", "i32", "load"], ["v2i64", "i64", "load"],
80 ["v4f32", "f32", "load"], ["v2f64", "f64", "load"]] in
81def load_splat_#args[0] :
82 PatFrag<(ops node:$addr), (wasm_load_splat
83 (!cast<ValueType>(args[1]) (!cast<PatFrag>(args[2]) node:$addr)))>;
84
85let Predicates = [HasUnimplementedSIMD128] in
86foreach args = [["v16i8", "v8x16"], ["v8i16", "v16x8"], ["v4i32", "v32x4"],
87 ["v2i64", "v64x2"], ["v4f32", "v32x4"], ["v2f64", "v64x2"]] in {
88def : LoadPatNoOffset<!cast<ValueType>(args[0]),
89 !cast<PatFrag>("load_splat_"#args[0]),
90 !cast<NI>("LOAD_SPLAT_"#args[1])>;
91def : LoadPatImmOff<!cast<ValueType>(args[0]),
92 !cast<PatFrag>("load_splat_"#args[0]),
93 regPlusImm,
94 !cast<NI>("LOAD_SPLAT_"#args[1])>;
95def : LoadPatImmOff<!cast<ValueType>(args[0]),
96 !cast<PatFrag>("load_splat_"#args[0]),
97 or_is_add,
98 !cast<NI>("LOAD_SPLAT_"#args[1])>;
99def : LoadPatOffsetOnly<!cast<ValueType>(args[0]),
100 !cast<PatFrag>("load_splat_"#args[0]),
101 !cast<NI>("LOAD_SPLAT_"#args[1])>;
102def : LoadPatGlobalAddrOffOnly<!cast<ValueType>(args[0]),
103 !cast<PatFrag>("load_splat_"#args[0]),
104 !cast<NI>("LOAD_SPLAT_"#args[1])>;
105}
106
Thomas Lively81125f72019-09-27 02:06:50 +0000107// Load and extend
108multiclass SIMDLoadExtend<ValueType vec_t, string name, bits<32> simdop> {
109 let mayLoad = 1, UseNamedOperandTable = 1,
110 Predicates = [HasUnimplementedSIMD128] in {
111 defm LOAD_EXTEND_S_#vec_t :
112 SIMD_I<(outs V128:$dst), (ins P2Align:$p2align, offset32_op:$off, I32:$addr),
113 (outs), (ins P2Align:$p2align, offset32_op:$off), [],
114 name#"_s\t$dst, ${off}(${addr})$p2align",
115 name#"_s\t$off$p2align", simdop>;
116 defm LOAD_EXTEND_U_#vec_t :
117 SIMD_I<(outs V128:$dst), (ins P2Align:$p2align, offset32_op:$off, I32:$addr),
118 (outs), (ins P2Align:$p2align, offset32_op:$off), [],
119 name#"_u\t$dst, ${off}(${addr})$p2align",
120 name#"_u\t$off$p2align", !add(simdop, 1)>;
121 }
122}
123
124defm "" : SIMDLoadExtend<v8i16, "i16x8.load8x8", 210>;
125defm "" : SIMDLoadExtend<v4i32, "i32x4.load16x4", 212>;
126defm "" : SIMDLoadExtend<v2i64, "i64x2.load32x2", 214>;
127
128let Predicates = [HasUnimplementedSIMD128] in
129foreach types = [[v8i16, i8], [v4i32, i16], [v2i64, i32]] in
130foreach exts = [["sextloadv", "_S"],
131 ["zextloadv", "_U"],
132 ["extloadv", "_U"]] in {
133def : LoadPatNoOffset<types[0], !cast<PatFrag>(exts[0]#types[1]),
134 !cast<NI>("LOAD_EXTEND"#exts[1]#"_"#types[0])>;
135def : LoadPatImmOff<types[0], !cast<PatFrag>(exts[0]#types[1]), regPlusImm,
136 !cast<NI>("LOAD_EXTEND"#exts[1]#"_"#types[0])>;
137def : LoadPatImmOff<types[0], !cast<PatFrag>(exts[0]#types[1]), or_is_add,
138 !cast<NI>("LOAD_EXTEND"#exts[1]#"_"#types[0])>;
139def : LoadPatOffsetOnly<types[0], !cast<PatFrag>(exts[0]#types[1]),
140 !cast<NI>("LOAD_EXTEND"#exts[1]#"_"#types[0])>;
141def : LoadPatGlobalAddrOffOnly<types[0], !cast<PatFrag>(exts[0]#types[1]),
142 !cast<NI>("LOAD_EXTEND"#exts[1]#"_"#types[0])>;
143}
144
145
Thomas Lively4ddd2252018-11-09 01:49:19 +0000146// Store: v128.store
Thomas Livelya9b3d1f2019-09-25 00:15:59 +0000147let mayStore = 1, UseNamedOperandTable = 1 in
148defm STORE_V128 :
149 SIMD_I<(outs), (ins P2Align:$p2align, offset32_op:$off, I32:$addr, V128:$vec),
150 (outs), (ins P2Align:$p2align, offset32_op:$off), [],
151 "v128.store\t${off}(${addr})$p2align, $vec",
152 "v128.store\t$off$p2align", 1>;
Thomas Lively4ddd2252018-11-09 01:49:19 +0000153
154foreach vec_t = [v16i8, v8i16, v4i32, v2i64, v4f32, v2f64] in {
Thomas Lively4ddd2252018-11-09 01:49:19 +0000155// Def load and store patterns from WebAssemblyInstrMemory.td for vector types
Thomas Livelya9b3d1f2019-09-25 00:15:59 +0000156def : StorePatNoOffset<vec_t, store, STORE_V128>;
157def : StorePatImmOff<vec_t, store, regPlusImm, STORE_V128>;
158def : StorePatImmOff<vec_t, store, or_is_add, STORE_V128>;
159def : StorePatOffsetOnly<vec_t, store, STORE_V128>;
160def : StorePatGlobalAddrOffOnly<vec_t, store, STORE_V128>;
Thomas Lively4ddd2252018-11-09 01:49:19 +0000161}
162
163//===----------------------------------------------------------------------===//
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000164// Constructing SIMD values
165//===----------------------------------------------------------------------===//
Thomas Lively9075cd62018-10-03 00:19:39 +0000166
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000167// Constant: v128.const
Thomas Lively22442922018-08-21 21:03:18 +0000168multiclass ConstVec<ValueType vec_t, dag ops, dag pat, string args> {
Thomas Lively8dbf29af2018-12-20 02:10:22 +0000169 let isMoveImm = 1, isReMaterializable = 1,
Thomas Livelya9b3d1f2019-09-25 00:15:59 +0000170 Predicates = [HasUnimplementedSIMD128] in
Thomas Lively22442922018-08-21 21:03:18 +0000171 defm CONST_V128_#vec_t : SIMD_I<(outs V128:$dst), ops, (outs), ops,
172 [(set V128:$dst, (vec_t pat))],
173 "v128.const\t$dst, "#args,
Thomas Lively299d2142018-11-09 01:45:56 +0000174 "v128.const\t"#args, 2>;
Thomas Lively22442922018-08-21 21:03:18 +0000175}
Thomas Lively123c3bb2018-08-23 00:43:47 +0000176
Thomas Lively22442922018-08-21 21:03:18 +0000177defm "" : ConstVec<v16i8,
178 (ins vec_i8imm_op:$i0, vec_i8imm_op:$i1,
179 vec_i8imm_op:$i2, vec_i8imm_op:$i3,
180 vec_i8imm_op:$i4, vec_i8imm_op:$i5,
181 vec_i8imm_op:$i6, vec_i8imm_op:$i7,
182 vec_i8imm_op:$i8, vec_i8imm_op:$i9,
183 vec_i8imm_op:$iA, vec_i8imm_op:$iB,
184 vec_i8imm_op:$iC, vec_i8imm_op:$iD,
185 vec_i8imm_op:$iE, vec_i8imm_op:$iF),
186 (build_vector ImmI8:$i0, ImmI8:$i1, ImmI8:$i2, ImmI8:$i3,
187 ImmI8:$i4, ImmI8:$i5, ImmI8:$i6, ImmI8:$i7,
188 ImmI8:$i8, ImmI8:$i9, ImmI8:$iA, ImmI8:$iB,
189 ImmI8:$iC, ImmI8:$iD, ImmI8:$iE, ImmI8:$iF),
190 !strconcat("$i0, $i1, $i2, $i3, $i4, $i5, $i6, $i7, ",
191 "$i8, $i9, $iA, $iB, $iC, $iD, $iE, $iF")>;
192defm "" : ConstVec<v8i16,
193 (ins vec_i16imm_op:$i0, vec_i16imm_op:$i1,
194 vec_i16imm_op:$i2, vec_i16imm_op:$i3,
195 vec_i16imm_op:$i4, vec_i16imm_op:$i5,
196 vec_i16imm_op:$i6, vec_i16imm_op:$i7),
197 (build_vector
198 ImmI16:$i0, ImmI16:$i1, ImmI16:$i2, ImmI16:$i3,
199 ImmI16:$i4, ImmI16:$i5, ImmI16:$i6, ImmI16:$i7),
200 "$i0, $i1, $i2, $i3, $i4, $i5, $i6, $i7">;
Heejin Ahn43675872019-02-06 00:17:03 +0000201let IsCanonical = 1 in
Thomas Lively22442922018-08-21 21:03:18 +0000202defm "" : ConstVec<v4i32,
Heejin Ahne37ba2c2019-02-05 01:59:49 +0000203 (ins vec_i32imm_op:$i0, vec_i32imm_op:$i1,
Thomas Lively22442922018-08-21 21:03:18 +0000204 vec_i32imm_op:$i2, vec_i32imm_op:$i3),
Heejin Ahne37ba2c2019-02-05 01:59:49 +0000205 (build_vector (i32 imm:$i0), (i32 imm:$i1),
206 (i32 imm:$i2), (i32 imm:$i3)),
207 "$i0, $i1, $i2, $i3">;
Thomas Lively22442922018-08-21 21:03:18 +0000208defm "" : ConstVec<v2i64,
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000209 (ins vec_i64imm_op:$i0, vec_i64imm_op:$i1),
210 (build_vector (i64 imm:$i0), (i64 imm:$i1)),
211 "$i0, $i1">;
Thomas Lively22442922018-08-21 21:03:18 +0000212defm "" : ConstVec<v4f32,
213 (ins f32imm_op:$i0, f32imm_op:$i1,
214 f32imm_op:$i2, f32imm_op:$i3),
215 (build_vector (f32 fpimm:$i0), (f32 fpimm:$i1),
216 (f32 fpimm:$i2), (f32 fpimm:$i3)),
217 "$i0, $i1, $i2, $i3">;
218defm "" : ConstVec<v2f64,
219 (ins f64imm_op:$i0, f64imm_op:$i1),
220 (build_vector (f64 fpimm:$i0), (f64 fpimm:$i1)),
221 "$i0, $i1">;
Thomas Livelyc1742572018-08-23 00:48:37 +0000222
Thomas Lively4ddd2252018-11-09 01:49:19 +0000223// Shuffle lanes: shuffle
224defm SHUFFLE :
225 SIMD_I<(outs V128:$dst),
226 (ins V128:$x, V128:$y,
227 vec_i8imm_op:$m0, vec_i8imm_op:$m1,
228 vec_i8imm_op:$m2, vec_i8imm_op:$m3,
229 vec_i8imm_op:$m4, vec_i8imm_op:$m5,
230 vec_i8imm_op:$m6, vec_i8imm_op:$m7,
231 vec_i8imm_op:$m8, vec_i8imm_op:$m9,
232 vec_i8imm_op:$mA, vec_i8imm_op:$mB,
233 vec_i8imm_op:$mC, vec_i8imm_op:$mD,
234 vec_i8imm_op:$mE, vec_i8imm_op:$mF),
235 (outs),
236 (ins
237 vec_i8imm_op:$m0, vec_i8imm_op:$m1,
238 vec_i8imm_op:$m2, vec_i8imm_op:$m3,
239 vec_i8imm_op:$m4, vec_i8imm_op:$m5,
240 vec_i8imm_op:$m6, vec_i8imm_op:$m7,
241 vec_i8imm_op:$m8, vec_i8imm_op:$m9,
242 vec_i8imm_op:$mA, vec_i8imm_op:$mB,
243 vec_i8imm_op:$mC, vec_i8imm_op:$mD,
244 vec_i8imm_op:$mE, vec_i8imm_op:$mF),
245 [],
246 "v8x16.shuffle\t$dst, $x, $y, "#
247 "$m0, $m1, $m2, $m3, $m4, $m5, $m6, $m7, "#
248 "$m8, $m9, $mA, $mB, $mC, $mD, $mE, $mF",
249 "v8x16.shuffle\t"#
250 "$m0, $m1, $m2, $m3, $m4, $m5, $m6, $m7, "#
251 "$m8, $m9, $mA, $mB, $mC, $mD, $mE, $mF",
252 3>;
253
254// Shuffles after custom lowering
255def wasm_shuffle_t : SDTypeProfile<1, 18, []>;
256def wasm_shuffle : SDNode<"WebAssemblyISD::SHUFFLE", wasm_shuffle_t>;
257foreach vec_t = [v16i8, v8i16, v4i32, v2i64, v4f32, v2f64] in {
258def : Pat<(vec_t (wasm_shuffle (vec_t V128:$x), (vec_t V128:$y),
259 (i32 LaneIdx32:$m0), (i32 LaneIdx32:$m1),
260 (i32 LaneIdx32:$m2), (i32 LaneIdx32:$m3),
261 (i32 LaneIdx32:$m4), (i32 LaneIdx32:$m5),
262 (i32 LaneIdx32:$m6), (i32 LaneIdx32:$m7),
263 (i32 LaneIdx32:$m8), (i32 LaneIdx32:$m9),
264 (i32 LaneIdx32:$mA), (i32 LaneIdx32:$mB),
265 (i32 LaneIdx32:$mC), (i32 LaneIdx32:$mD),
266 (i32 LaneIdx32:$mE), (i32 LaneIdx32:$mF))),
267 (vec_t (SHUFFLE (vec_t V128:$x), (vec_t V128:$y),
268 (i32 LaneIdx32:$m0), (i32 LaneIdx32:$m1),
269 (i32 LaneIdx32:$m2), (i32 LaneIdx32:$m3),
270 (i32 LaneIdx32:$m4), (i32 LaneIdx32:$m5),
271 (i32 LaneIdx32:$m6), (i32 LaneIdx32:$m7),
272 (i32 LaneIdx32:$m8), (i32 LaneIdx32:$m9),
273 (i32 LaneIdx32:$mA), (i32 LaneIdx32:$mB),
274 (i32 LaneIdx32:$mC), (i32 LaneIdx32:$mD),
275 (i32 LaneIdx32:$mE), (i32 LaneIdx32:$mF)))>;
276}
277
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000278// Create vector with identical lanes: splat
279def splat2 : PatFrag<(ops node:$x), (build_vector node:$x, node:$x)>;
280def splat4 : PatFrag<(ops node:$x), (build_vector
281 node:$x, node:$x, node:$x, node:$x)>;
282def splat8 : PatFrag<(ops node:$x), (build_vector
283 node:$x, node:$x, node:$x, node:$x,
284 node:$x, node:$x, node:$x, node:$x)>;
285def splat16 : PatFrag<(ops node:$x), (build_vector
286 node:$x, node:$x, node:$x, node:$x,
287 node:$x, node:$x, node:$x, node:$x,
288 node:$x, node:$x, node:$x, node:$x,
289 node:$x, node:$x, node:$x, node:$x)>;
290
291multiclass Splat<ValueType vec_t, string vec, WebAssemblyRegClass reg_t,
292 PatFrag splat_pat, bits<32> simdop> {
293 // Prefer splats over v128.const for const splats (65 is lowest that works)
294 let AddedComplexity = 65 in
295 defm SPLAT_#vec_t : SIMD_I<(outs V128:$dst), (ins reg_t:$x), (outs), (ins),
296 [(set (vec_t V128:$dst), (splat_pat reg_t:$x))],
297 vec#".splat\t$dst, $x", vec#".splat", simdop>;
298}
299
Thomas Lively299d2142018-11-09 01:45:56 +0000300defm "" : Splat<v16i8, "i8x16", I32, splat16, 4>;
301defm "" : Splat<v8i16, "i16x8", I32, splat8, 8>;
302defm "" : Splat<v4i32, "i32x4", I32, splat4, 12>;
303defm "" : Splat<v2i64, "i64x2", I64, splat2, 15>;
304defm "" : Splat<v4f32, "f32x4", F32, splat4, 18>;
305defm "" : Splat<v2f64, "f64x2", F64, splat2, 21>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000306
Thomas Lively74c12ce2019-01-29 23:44:48 +0000307// scalar_to_vector leaves high lanes undefined, so can be a splat
308class ScalarSplatPat<ValueType vec_t, ValueType lane_t,
309 WebAssemblyRegClass reg_t> :
310 Pat<(vec_t (scalar_to_vector (lane_t reg_t:$x))),
311 (!cast<Instruction>("SPLAT_"#vec_t) reg_t:$x)>;
312
313def : ScalarSplatPat<v16i8, i32, I32>;
314def : ScalarSplatPat<v8i16, i32, I32>;
315def : ScalarSplatPat<v4i32, i32, I32>;
316def : ScalarSplatPat<v2i64, i64, I64>;
317def : ScalarSplatPat<v4f32, f32, F32>;
318def : ScalarSplatPat<v2f64, f64, F64>;
319
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000320//===----------------------------------------------------------------------===//
321// Accessing lanes
322//===----------------------------------------------------------------------===//
323
324// Extract lane as a scalar: extract_lane / extract_lane_s / extract_lane_u
325multiclass ExtractLane<ValueType vec_t, string vec, ImmLeaf imm_t,
326 WebAssemblyRegClass reg_t, bits<32> simdop,
327 string suffix = "", SDNode extract = vector_extract> {
328 defm EXTRACT_LANE_#vec_t#suffix :
329 SIMD_I<(outs reg_t:$dst), (ins V128:$vec, vec_i8imm_op:$idx),
330 (outs), (ins vec_i8imm_op:$idx),
331 [(set reg_t:$dst, (extract (vec_t V128:$vec), (i32 imm_t:$idx)))],
332 vec#".extract_lane"#suffix#"\t$dst, $vec, $idx",
333 vec#".extract_lane"#suffix#"\t$idx", simdop>;
334}
335
336multiclass ExtractPat<ValueType lane_t, int mask> {
337 def _s : PatFrag<(ops node:$vec, node:$idx),
338 (i32 (sext_inreg
339 (i32 (vector_extract
340 node:$vec,
341 node:$idx
342 )),
343 lane_t
344 ))>;
345 def _u : PatFrag<(ops node:$vec, node:$idx),
346 (i32 (and
347 (i32 (vector_extract
348 node:$vec,
349 node:$idx
350 )),
351 (i32 mask)
352 ))>;
353}
354
355defm extract_i8x16 : ExtractPat<i8, 0xff>;
356defm extract_i16x8 : ExtractPat<i16, 0xffff>;
357
358multiclass ExtractLaneExtended<string sign, bits<32> baseInst> {
359 defm "" : ExtractLane<v16i8, "i8x16", LaneIdx16, I32, baseInst, sign,
360 !cast<PatFrag>("extract_i8x16"#sign)>;
Thomas Lively299d2142018-11-09 01:45:56 +0000361 defm "" : ExtractLane<v8i16, "i16x8", LaneIdx8, I32, !add(baseInst, 4), sign,
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000362 !cast<PatFrag>("extract_i16x8"#sign)>;
Thomas Livelyd183d8c2018-08-30 21:36:48 +0000363}
364
Thomas Lively299d2142018-11-09 01:45:56 +0000365defm "" : ExtractLaneExtended<"_s", 5>;
Thomas Livelya9b3d1f2019-09-25 00:15:59 +0000366let Predicates = [HasUnimplementedSIMD128] in
Thomas Lively299d2142018-11-09 01:45:56 +0000367defm "" : ExtractLaneExtended<"_u", 6>;
Thomas Lively5222cb62018-08-15 18:15:18 +0000368defm "" : ExtractLane<v4i32, "i32x4", LaneIdx4, I32, 13>;
Thomas Lively299d2142018-11-09 01:45:56 +0000369defm "" : ExtractLane<v2i64, "i64x2", LaneIdx2, I64, 16>;
370defm "" : ExtractLane<v4f32, "f32x4", LaneIdx4, F32, 19>;
371defm "" : ExtractLane<v2f64, "f64x2", LaneIdx2, F64, 22>;
Thomas Livelyc1742572018-08-23 00:48:37 +0000372
Thomas Lively8dbf29af2018-12-20 02:10:22 +0000373// It would be more conventional to use unsigned extracts, but v8
374// doesn't implement them yet
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000375def : Pat<(i32 (vector_extract (v16i8 V128:$vec), (i32 LaneIdx16:$idx))),
Thomas Lively8dbf29af2018-12-20 02:10:22 +0000376 (EXTRACT_LANE_v16i8_s V128:$vec, (i32 LaneIdx16:$idx))>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000377def : Pat<(i32 (vector_extract (v8i16 V128:$vec), (i32 LaneIdx8:$idx))),
Thomas Lively8dbf29af2018-12-20 02:10:22 +0000378 (EXTRACT_LANE_v8i16_s V128:$vec, (i32 LaneIdx8:$idx))>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000379
Thomas Lively11a332d02018-10-19 19:08:06 +0000380// Lower undef lane indices to zero
381def : Pat<(and (i32 (vector_extract (v16i8 V128:$vec), undef)), (i32 0xff)),
382 (EXTRACT_LANE_v16i8_u V128:$vec, 0)>;
383def : Pat<(and (i32 (vector_extract (v8i16 V128:$vec), undef)), (i32 0xffff)),
384 (EXTRACT_LANE_v8i16_u V128:$vec, 0)>;
385def : Pat<(i32 (vector_extract (v16i8 V128:$vec), undef)),
386 (EXTRACT_LANE_v16i8_u V128:$vec, 0)>;
387def : Pat<(i32 (vector_extract (v8i16 V128:$vec), undef)),
388 (EXTRACT_LANE_v8i16_u V128:$vec, 0)>;
389def : Pat<(sext_inreg (i32 (vector_extract (v16i8 V128:$vec), undef)), i8),
390 (EXTRACT_LANE_v16i8_s V128:$vec, 0)>;
391def : Pat<(sext_inreg (i32 (vector_extract (v8i16 V128:$vec), undef)), i16),
392 (EXTRACT_LANE_v8i16_s V128:$vec, 0)>;
393def : Pat<(vector_extract (v4i32 V128:$vec), undef),
394 (EXTRACT_LANE_v4i32 V128:$vec, 0)>;
395def : Pat<(vector_extract (v2i64 V128:$vec), undef),
396 (EXTRACT_LANE_v2i64 V128:$vec, 0)>;
397def : Pat<(vector_extract (v4f32 V128:$vec), undef),
398 (EXTRACT_LANE_v4f32 V128:$vec, 0)>;
399def : Pat<(vector_extract (v2f64 V128:$vec), undef),
400 (EXTRACT_LANE_v2f64 V128:$vec, 0)>;
401
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000402// Replace lane value: replace_lane
403multiclass ReplaceLane<ValueType vec_t, string vec, ImmLeaf imm_t,
404 WebAssemblyRegClass reg_t, ValueType lane_t,
405 bits<32> simdop> {
406 defm REPLACE_LANE_#vec_t :
407 SIMD_I<(outs V128:$dst), (ins V128:$vec, vec_i8imm_op:$idx, reg_t:$x),
408 (outs), (ins vec_i8imm_op:$idx),
409 [(set V128:$dst, (vector_insert
410 (vec_t V128:$vec), (lane_t reg_t:$x), (i32 imm_t:$idx)))],
411 vec#".replace_lane\t$dst, $vec, $idx, $x",
412 vec#".replace_lane\t$idx", simdop>;
413}
414
Thomas Lively299d2142018-11-09 01:45:56 +0000415defm "" : ReplaceLane<v16i8, "i8x16", LaneIdx16, I32, i32, 7>;
416defm "" : ReplaceLane<v8i16, "i16x8", LaneIdx8, I32, i32, 11>;
417defm "" : ReplaceLane<v4i32, "i32x4", LaneIdx4, I32, i32, 14>;
418defm "" : ReplaceLane<v2i64, "i64x2", LaneIdx2, I64, i64, 17>;
419defm "" : ReplaceLane<v4f32, "f32x4", LaneIdx4, F32, f32, 20>;
420defm "" : ReplaceLane<v2f64, "f64x2", LaneIdx2, F64, f64, 23>;
Thomas Livelyc1742572018-08-23 00:48:37 +0000421
Thomas Lively11a332d02018-10-19 19:08:06 +0000422// Lower undef lane indices to zero
423def : Pat<(vector_insert (v16i8 V128:$vec), I32:$x, undef),
424 (REPLACE_LANE_v16i8 V128:$vec, 0, I32:$x)>;
425def : Pat<(vector_insert (v8i16 V128:$vec), I32:$x, undef),
426 (REPLACE_LANE_v8i16 V128:$vec, 0, I32:$x)>;
427def : Pat<(vector_insert (v4i32 V128:$vec), I32:$x, undef),
428 (REPLACE_LANE_v4i32 V128:$vec, 0, I32:$x)>;
429def : Pat<(vector_insert (v2i64 V128:$vec), I64:$x, undef),
430 (REPLACE_LANE_v2i64 V128:$vec, 0, I64:$x)>;
431def : Pat<(vector_insert (v4f32 V128:$vec), F32:$x, undef),
432 (REPLACE_LANE_v4f32 V128:$vec, 0, F32:$x)>;
433def : Pat<(vector_insert (v2f64 V128:$vec), F64:$x, undef),
434 (REPLACE_LANE_v2f64 V128:$vec, 0, F64:$x)>;
435
Thomas Lively4ddd2252018-11-09 01:49:19 +0000436//===----------------------------------------------------------------------===//
437// Comparisons
438//===----------------------------------------------------------------------===//
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000439
Thomas Lively4ddd2252018-11-09 01:49:19 +0000440multiclass SIMDCondition<ValueType vec_t, ValueType out_t, string vec,
441 string name, CondCode cond, bits<32> simdop> {
442 defm _#vec_t :
443 SIMD_I<(outs V128:$dst), (ins V128:$lhs, V128:$rhs), (outs), (ins),
444 [(set (out_t V128:$dst),
445 (setcc (vec_t V128:$lhs), (vec_t V128:$rhs), cond)
446 )],
447 vec#"."#name#"\t$dst, $lhs, $rhs", vec#"."#name, simdop>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000448}
449
Thomas Lively4ddd2252018-11-09 01:49:19 +0000450multiclass SIMDConditionInt<string name, CondCode cond, bits<32> baseInst> {
451 defm "" : SIMDCondition<v16i8, v16i8, "i8x16", name, cond, baseInst>;
452 defm "" : SIMDCondition<v8i16, v8i16, "i16x8", name, cond,
453 !add(baseInst, 10)>;
454 defm "" : SIMDCondition<v4i32, v4i32, "i32x4", name, cond,
455 !add(baseInst, 20)>;
456}
457
458multiclass SIMDConditionFP<string name, CondCode cond, bits<32> baseInst> {
459 defm "" : SIMDCondition<v4f32, v4i32, "f32x4", name, cond, baseInst>;
460 defm "" : SIMDCondition<v2f64, v2i64, "f64x2", name, cond,
461 !add(baseInst, 6)>;
462}
463
464// Equality: eq
465let isCommutable = 1 in {
466defm EQ : SIMDConditionInt<"eq", SETEQ, 24>;
467defm EQ : SIMDConditionFP<"eq", SETOEQ, 64>;
468} // isCommutable = 1
469
470// Non-equality: ne
471let isCommutable = 1 in {
472defm NE : SIMDConditionInt<"ne", SETNE, 25>;
473defm NE : SIMDConditionFP<"ne", SETUNE, 65>;
474} // isCommutable = 1
475
476// Less than: lt_s / lt_u / lt
477defm LT_S : SIMDConditionInt<"lt_s", SETLT, 26>;
478defm LT_U : SIMDConditionInt<"lt_u", SETULT, 27>;
479defm LT : SIMDConditionFP<"lt", SETOLT, 66>;
480
481// Greater than: gt_s / gt_u / gt
482defm GT_S : SIMDConditionInt<"gt_s", SETGT, 28>;
483defm GT_U : SIMDConditionInt<"gt_u", SETUGT, 29>;
484defm GT : SIMDConditionFP<"gt", SETOGT, 67>;
485
486// Less than or equal: le_s / le_u / le
487defm LE_S : SIMDConditionInt<"le_s", SETLE, 30>;
488defm LE_U : SIMDConditionInt<"le_u", SETULE, 31>;
489defm LE : SIMDConditionFP<"le", SETOLE, 68>;
490
491// Greater than or equal: ge_s / ge_u / ge
492defm GE_S : SIMDConditionInt<"ge_s", SETGE, 32>;
493defm GE_U : SIMDConditionInt<"ge_u", SETUGE, 33>;
494defm GE : SIMDConditionFP<"ge", SETOGE, 69>;
495
496// Lower float comparisons that don't care about NaN to standard WebAssembly
Thomas Lively0200d622019-03-19 00:55:34 +0000497// float comparisons. These instructions are generated with nnan and in the
498// target-independent expansion of unordered comparisons and ordered ne.
499foreach nodes = [[seteq, EQ_v4f32], [setne, NE_v4f32], [setlt, LT_v4f32],
500 [setgt, GT_v4f32], [setle, LE_v4f32], [setge, GE_v4f32]] in
501def : Pat<(v4i32 (nodes[0] (v4f32 V128:$lhs), (v4f32 V128:$rhs))),
502 (v4i32 (nodes[1] (v4f32 V128:$lhs), (v4f32 V128:$rhs)))>;
503
504foreach nodes = [[seteq, EQ_v2f64], [setne, NE_v2f64], [setlt, LT_v2f64],
505 [setgt, GT_v2f64], [setle, LE_v2f64], [setge, GE_v2f64]] in
506def : Pat<(v2i64 (nodes[0] (v2f64 V128:$lhs), (v2f64 V128:$rhs))),
507 (v2i64 (nodes[1] (v2f64 V128:$lhs), (v2f64 V128:$rhs)))>;
508
Thomas Lively4ddd2252018-11-09 01:49:19 +0000509
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000510//===----------------------------------------------------------------------===//
Thomas Lively4ddd2252018-11-09 01:49:19 +0000511// Bitwise operations
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000512//===----------------------------------------------------------------------===//
513
514multiclass SIMDBinary<ValueType vec_t, string vec, SDNode node, string name,
515 bits<32> simdop> {
516 defm _#vec_t : SIMD_I<(outs V128:$dst), (ins V128:$lhs, V128:$rhs),
517 (outs), (ins),
Thomas Lively7fa7e6a2018-10-11 00:49:24 +0000518 [(set (vec_t V128:$dst),
519 (node (vec_t V128:$lhs), (vec_t V128:$rhs))
520 )],
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000521 vec#"."#name#"\t$dst, $lhs, $rhs", vec#"."#name,
522 simdop>;
523}
524
Thomas Lively4ddd2252018-11-09 01:49:19 +0000525multiclass SIMDBitwise<SDNode node, string name, bits<32> simdop> {
526 defm "" : SIMDBinary<v16i8, "v128", node, name, simdop>;
527 defm "" : SIMDBinary<v8i16, "v128", node, name, simdop>;
528 defm "" : SIMDBinary<v4i32, "v128", node, name, simdop>;
529 defm "" : SIMDBinary<v2i64, "v128", node, name, simdop>;
Thomas Lively299d2142018-11-09 01:45:56 +0000530}
531
532multiclass SIMDUnary<ValueType vec_t, string vec, SDNode node, string name,
533 bits<32> simdop> {
534 defm _#vec_t : SIMD_I<(outs V128:$dst), (ins V128:$vec), (outs), (ins),
535 [(set (vec_t V128:$dst),
536 (vec_t (node (vec_t V128:$vec)))
537 )],
538 vec#"."#name#"\t$dst, $vec", vec#"."#name, simdop>;
539}
540
Thomas Lively4ddd2252018-11-09 01:49:19 +0000541// Bitwise logic: v128.not
542foreach vec_t = [v16i8, v8i16, v4i32, v2i64] in
Thomas Lively77b33c82018-11-15 03:38:59 +0000543defm NOT: SIMDUnary<vec_t, "v128", vnot, "not", 76>;
544
545// Bitwise logic: v128.and / v128.or / v128.xor
546let isCommutable = 1 in {
547defm AND : SIMDBitwise<and, "and", 77>;
548defm OR : SIMDBitwise<or, "or", 78>;
549defm XOR : SIMDBitwise<xor, "xor", 79>;
550} // isCommutable = 1
Thomas Lively4ddd2252018-11-09 01:49:19 +0000551
Thomas Lively3fcdd252019-09-27 02:11:40 +0000552// Bitwise logic: v128.andnot
553def andnot : PatFrag<(ops node:$left, node:$right), (and $left, (vnot $right))>;
554let Predicates = [HasUnimplementedSIMD128] in
555defm ANDNOT : SIMDBitwise<andnot, "andnot", 216>;
556
Thomas Lively4ddd2252018-11-09 01:49:19 +0000557// Bitwise select: v128.bitselect
558foreach vec_t = [v16i8, v8i16, v4i32, v2i64, v4f32, v2f64] in
559 defm BITSELECT_#vec_t :
560 SIMD_I<(outs V128:$dst), (ins V128:$v1, V128:$v2, V128:$c), (outs), (ins),
561 [(set (vec_t V128:$dst),
562 (vec_t (int_wasm_bitselect
Thomas Livelyedb54b22019-01-09 18:13:11 +0000563 (vec_t V128:$v1), (vec_t V128:$v2), (vec_t V128:$c)
Thomas Lively4ddd2252018-11-09 01:49:19 +0000564 ))
565 )],
566 "v128.bitselect\t$dst, $v1, $v2, $c", "v128.bitselect", 80>;
567
568// Bitselect is equivalent to (c & v1) | (~c & v2)
569foreach vec_t = [v16i8, v8i16, v4i32, v2i64] in
570 def : Pat<(vec_t (or (and (vec_t V128:$c), (vec_t V128:$v1)),
571 (and (vnot V128:$c), (vec_t V128:$v2)))),
572 (!cast<Instruction>("BITSELECT_"#vec_t)
573 V128:$v1, V128:$v2, V128:$c)>;
574
575//===----------------------------------------------------------------------===//
576// Integer unary arithmetic
577//===----------------------------------------------------------------------===//
578
Thomas Lively299d2142018-11-09 01:45:56 +0000579multiclass SIMDUnaryInt<SDNode node, string name, bits<32> baseInst> {
580 defm "" : SIMDUnary<v16i8, "i8x16", node, name, baseInst>;
581 defm "" : SIMDUnary<v8i16, "i16x8", node, name, !add(baseInst, 17)>;
582 defm "" : SIMDUnary<v4i32, "i32x4", node, name, !add(baseInst, 34)>;
583 defm "" : SIMDUnary<v2i64, "i64x2", node, name, !add(baseInst, 51)>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000584}
585
Thomas Lively4ddd2252018-11-09 01:49:19 +0000586multiclass SIMDReduceVec<ValueType vec_t, string vec, SDNode op, string name,
587 bits<32> simdop> {
588 defm _#vec_t : SIMD_I<(outs I32:$dst), (ins V128:$vec), (outs), (ins),
589 [(set I32:$dst, (i32 (op (vec_t V128:$vec))))],
590 vec#"."#name#"\t$dst, $vec", vec#"."#name, simdop>;
591}
592
593multiclass SIMDReduce<SDNode op, string name, bits<32> baseInst> {
594 defm "" : SIMDReduceVec<v16i8, "i8x16", op, name, baseInst>;
595 defm "" : SIMDReduceVec<v8i16, "i16x8", op, name, !add(baseInst, 17)>;
596 defm "" : SIMDReduceVec<v4i32, "i32x4", op, name, !add(baseInst, 34)>;
597 defm "" : SIMDReduceVec<v2i64, "i64x2", op, name, !add(baseInst, 51)>;
598}
599
Thomas Lively108e98e2018-10-10 01:09:09 +0000600// Integer vector negation
601def ivneg : PatFrag<(ops node:$in), (sub immAllZerosV, node:$in)>;
602
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000603// Integer negation: neg
Thomas Lively299d2142018-11-09 01:45:56 +0000604defm NEG : SIMDUnaryInt<ivneg, "neg", 81>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000605
Thomas Lively4ddd2252018-11-09 01:49:19 +0000606// Any lane true: any_true
607defm ANYTRUE : SIMDReduce<int_wasm_anytrue, "any_true", 82>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000608
Thomas Lively4ddd2252018-11-09 01:49:19 +0000609// All lanes true: all_true
610defm ALLTRUE : SIMDReduce<int_wasm_alltrue, "all_true", 83>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000611
Thomas Lively18857472019-06-19 00:02:13 +0000612// Reductions already return 0 or 1, so and 1, setne 0, and seteq 1
613// can be folded out
614foreach reduction =
615 [["int_wasm_anytrue", "ANYTRUE"], ["int_wasm_alltrue", "ALLTRUE"]] in
616foreach ty = [v16i8, v8i16, v4i32, v2i64] in {
617def : Pat<(i32 (and
618 (i32 (!cast<Intrinsic>(reduction[0]) (ty V128:$x))),
619 (i32 1)
620 )),
621 (i32 (!cast<NI>(reduction[1]#"_"#ty) (ty V128:$x)))>;
622def : Pat<(i32 (setne
623 (i32 (!cast<Intrinsic>(reduction[0]) (ty V128:$x))),
624 (i32 0)
625 )),
626 (i32 (!cast<NI>(reduction[1]#"_"#ty) (ty V128:$x)))>;
627def : Pat<(i32 (seteq
628 (i32 (!cast<Intrinsic>(reduction[0]) (ty V128:$x))),
629 (i32 1)
630 )),
631 (i32 (!cast<NI>(reduction[1]#"_"#ty) (ty V128:$x)))>;
632}
633
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000634//===----------------------------------------------------------------------===//
635// Bit shifts
636//===----------------------------------------------------------------------===//
637
638multiclass SIMDShift<ValueType vec_t, string vec, SDNode node, dag shift_vec,
639 string name, bits<32> simdop> {
640 defm _#vec_t : SIMD_I<(outs V128:$dst), (ins V128:$vec, I32:$x),
641 (outs), (ins),
642 [(set (vec_t V128:$dst),
643 (node V128:$vec, (vec_t shift_vec)))],
644 vec#"."#name#"\t$dst, $vec, $x", vec#"."#name, simdop>;
645}
646
Thomas Lively299d2142018-11-09 01:45:56 +0000647multiclass SIMDShiftInt<SDNode node, string name, bits<32> baseInst> {
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000648 defm "" : SIMDShift<v16i8, "i8x16", node, (splat16 I32:$x), name, baseInst>;
649 defm "" : SIMDShift<v8i16, "i16x8", node, (splat8 I32:$x), name,
Thomas Lively299d2142018-11-09 01:45:56 +0000650 !add(baseInst, 17)>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000651 defm "" : SIMDShift<v4i32, "i32x4", node, (splat4 I32:$x), name,
Thomas Lively299d2142018-11-09 01:45:56 +0000652 !add(baseInst, 34)>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000653 defm "" : SIMDShift<v2i64, "i64x2", node, (splat2 (i64 (zext I32:$x))),
Thomas Lively299d2142018-11-09 01:45:56 +0000654 name, !add(baseInst, 51)>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000655}
656
657// Left shift by scalar: shl
Thomas Lively299d2142018-11-09 01:45:56 +0000658defm SHL : SIMDShiftInt<shl, "shl", 84>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000659
660// Right shift by scalar: shr_s / shr_u
Thomas Lively299d2142018-11-09 01:45:56 +0000661defm SHR_S : SIMDShiftInt<sra, "shr_s", 85>;
662defm SHR_U : SIMDShiftInt<srl, "shr_u", 86>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000663
Thomas Lively7663e0c2019-06-26 16:19:59 +0000664// Truncate i64 shift operands to i32s, except if they are already i32s
665foreach shifts = [[shl, SHL_v2i64], [sra, SHR_S_v2i64], [srl, SHR_U_v2i64]] in {
666def : Pat<(v2i64 (shifts[0]
667 (v2i64 V128:$vec),
668 (v2i64 (splat2 (i64 (sext I32:$x))))
669 )),
670 (v2i64 (shifts[1] (v2i64 V128:$vec), (i32 I32:$x)))>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000671def : Pat<(v2i64 (shifts[0] (v2i64 V128:$vec), (v2i64 (splat2 I64:$x)))),
672 (v2i64 (shifts[1] (v2i64 V128:$vec), (I32_WRAP_I64 I64:$x)))>;
Thomas Lively7663e0c2019-06-26 16:19:59 +0000673}
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000674
Thomas Lively55735d52018-10-20 01:31:18 +0000675// 2xi64 shifts with constant shift amounts are custom lowered to avoid wrapping
676def wasm_shift_t : SDTypeProfile<1, 2,
677 [SDTCisVec<0>, SDTCisSameAs<0, 1>, SDTCisVT<2, i32>]
678>;
679def wasm_shl : SDNode<"WebAssemblyISD::VEC_SHL", wasm_shift_t>;
680def wasm_shr_s : SDNode<"WebAssemblyISD::VEC_SHR_S", wasm_shift_t>;
681def wasm_shr_u : SDNode<"WebAssemblyISD::VEC_SHR_U", wasm_shift_t>;
682foreach shifts = [[wasm_shl, SHL_v2i64],
683 [wasm_shr_s, SHR_S_v2i64],
684 [wasm_shr_u, SHR_U_v2i64]] in
685def : Pat<(v2i64 (shifts[0] (v2i64 V128:$vec), I32:$x)),
686 (v2i64 (shifts[1] (v2i64 V128:$vec), I32:$x))>;
687
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000688//===----------------------------------------------------------------------===//
Thomas Lively4ddd2252018-11-09 01:49:19 +0000689// Integer binary arithmetic
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000690//===----------------------------------------------------------------------===//
691
Thomas Lively4ddd2252018-11-09 01:49:19 +0000692multiclass SIMDBinaryIntSmall<SDNode node, string name, bits<32> baseInst> {
693 defm "" : SIMDBinary<v16i8, "i8x16", node, name, baseInst>;
694 defm "" : SIMDBinary<v8i16, "i16x8", node, name, !add(baseInst, 17)>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000695}
696
Thomas Lively4ddd2252018-11-09 01:49:19 +0000697multiclass SIMDBinaryIntNoI64x2<SDNode node, string name, bits<32> baseInst> {
698 defm "" : SIMDBinaryIntSmall<node, name, baseInst>;
699 defm "" : SIMDBinary<v4i32, "i32x4", node, name, !add(baseInst, 34)>;
700}
701
702multiclass SIMDBinaryInt<SDNode node, string name, bits<32> baseInst> {
703 defm "" : SIMDBinaryIntNoI64x2<node, name, baseInst>;
704 defm "" : SIMDBinary<v2i64, "i64x2", node, name, !add(baseInst, 51)>;
705}
706
707// Integer addition: add / add_saturate_s / add_saturate_u
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000708let isCommutable = 1 in {
Thomas Lively4ddd2252018-11-09 01:49:19 +0000709defm ADD : SIMDBinaryInt<add, "add", 87>;
710defm ADD_SAT_S : SIMDBinaryIntSmall<saddsat, "add_saturate_s", 88>;
711defm ADD_SAT_U : SIMDBinaryIntSmall<uaddsat, "add_saturate_u", 89>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000712} // isCommutable = 1
713
Thomas Lively4ddd2252018-11-09 01:49:19 +0000714// Integer subtraction: sub / sub_saturate_s / sub_saturate_u
715defm SUB : SIMDBinaryInt<sub, "sub", 90>;
716defm SUB_SAT_S :
717 SIMDBinaryIntSmall<int_wasm_sub_saturate_signed, "sub_saturate_s", 91>;
718defm SUB_SAT_U :
719 SIMDBinaryIntSmall<int_wasm_sub_saturate_unsigned, "sub_saturate_u", 92>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000720
Thomas Lively4ddd2252018-11-09 01:49:19 +0000721// Integer multiplication: mul
722defm MUL : SIMDBinaryIntNoI64x2<mul, "mul", 93>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000723
724//===----------------------------------------------------------------------===//
Thomas Lively4ddd2252018-11-09 01:49:19 +0000725// Floating-point unary arithmetic
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000726//===----------------------------------------------------------------------===//
727
Thomas Lively299d2142018-11-09 01:45:56 +0000728multiclass SIMDUnaryFP<SDNode node, string name, bits<32> baseInst> {
729 defm "" : SIMDUnary<v4f32, "f32x4", node, name, baseInst>;
730 defm "" : SIMDUnary<v2f64, "f64x2", node, name, !add(baseInst, 11)>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000731}
732
Thomas Lively299d2142018-11-09 01:45:56 +0000733// Absolute value: abs
734defm ABS : SIMDUnaryFP<fabs, "abs", 149>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000735
Thomas Lively4ddd2252018-11-09 01:49:19 +0000736// Negation: neg
737defm NEG : SIMDUnaryFP<fneg, "neg", 150>;
738
739// Square root: sqrt
Thomas Livelya9b3d1f2019-09-25 00:15:59 +0000740let Predicates = [HasUnimplementedSIMD128] in
Thomas Lively4ddd2252018-11-09 01:49:19 +0000741defm SQRT : SIMDUnaryFP<fsqrt, "sqrt", 151>;
742
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000743//===----------------------------------------------------------------------===//
Thomas Lively4ddd2252018-11-09 01:49:19 +0000744// Floating-point binary arithmetic
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000745//===----------------------------------------------------------------------===//
746
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000747multiclass SIMDBinaryFP<SDNode node, string name, bits<32> baseInst> {
748 defm "" : SIMDBinary<v4f32, "f32x4", node, name, baseInst>;
Thomas Lively299d2142018-11-09 01:45:56 +0000749 defm "" : SIMDBinary<v2f64, "f64x2", node, name, !add(baseInst, 11)>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000750}
751
752// Addition: add
753let isCommutable = 1 in
Thomas Lively299d2142018-11-09 01:45:56 +0000754defm ADD : SIMDBinaryFP<fadd, "add", 154>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000755
756// Subtraction: sub
Thomas Lively299d2142018-11-09 01:45:56 +0000757defm SUB : SIMDBinaryFP<fsub, "sub", 155>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000758
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000759// Multiplication: mul
760let isCommutable = 1 in
Thomas Lively299d2142018-11-09 01:45:56 +0000761defm MUL : SIMDBinaryFP<fmul, "mul", 156>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000762
Thomas Lively4ddd2252018-11-09 01:49:19 +0000763// Division: div
Thomas Livelya9b3d1f2019-09-25 00:15:59 +0000764let Predicates = [HasUnimplementedSIMD128] in
Thomas Lively4ddd2252018-11-09 01:49:19 +0000765defm DIV : SIMDBinaryFP<fdiv, "div", 157>;
766
767// NaN-propagating minimum: min
768defm MIN : SIMDBinaryFP<fminimum, "min", 158>;
769
770// NaN-propagating maximum: max
771defm MAX : SIMDBinaryFP<fmaximum, "max", 159>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000772
773//===----------------------------------------------------------------------===//
774// Conversions
775//===----------------------------------------------------------------------===//
776
777multiclass SIMDConvert<ValueType vec_t, ValueType arg_t, SDNode op,
778 string name, bits<32> simdop> {
779 defm op#_#vec_t#_#arg_t :
780 SIMD_I<(outs V128:$dst), (ins V128:$vec), (outs), (ins),
781 [(set (vec_t V128:$dst), (vec_t (op (arg_t V128:$vec))))],
782 name#"\t$dst, $vec", name, simdop>;
783}
784
Thomas Lively6a87dda2019-01-08 06:25:55 +0000785// Integer to floating point: convert
786defm "" : SIMDConvert<v4f32, v4i32, sint_to_fp, "f32x4.convert_i32x4_s", 175>;
787defm "" : SIMDConvert<v4f32, v4i32, uint_to_fp, "f32x4.convert_i32x4_u", 176>;
788defm "" : SIMDConvert<v2f64, v2i64, sint_to_fp, "f64x2.convert_i64x2_s", 177>;
789defm "" : SIMDConvert<v2f64, v2i64, uint_to_fp, "f64x2.convert_i64x2_u", 178>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000790
Thomas Lively6a87dda2019-01-08 06:25:55 +0000791// Floating point to integer with saturation: trunc_sat
792defm "" : SIMDConvert<v4i32, v4f32, fp_to_sint, "i32x4.trunc_sat_f32x4_s", 171>;
793defm "" : SIMDConvert<v4i32, v4f32, fp_to_uint, "i32x4.trunc_sat_f32x4_u", 172>;
794defm "" : SIMDConvert<v2i64, v2f64, fp_to_sint, "i64x2.trunc_sat_f64x2_s", 173>;
795defm "" : SIMDConvert<v2i64, v2f64, fp_to_uint, "i64x2.trunc_sat_f64x2_u", 174>;
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000796
Thomas Livelyae530c52019-09-13 22:54:41 +0000797// Widening operations
798multiclass SIMDWiden<ValueType vec_t, string vec, ValueType arg_t, string arg,
799 bits<32> baseInst> {
800 defm "" : SIMDConvert<vec_t, arg_t, int_wasm_widen_low_signed,
801 vec#".widen_low_"#arg#"_s", baseInst>;
802 defm "" : SIMDConvert<vec_t, arg_t, int_wasm_widen_high_signed,
803 vec#".widen_high_"#arg#"_s", !add(baseInst, 1)>;
804 defm "" : SIMDConvert<vec_t, arg_t, int_wasm_widen_low_unsigned,
805 vec#".widen_low_"#arg#"_u", !add(baseInst, 2)>;
806 defm "" : SIMDConvert<vec_t, arg_t, int_wasm_widen_high_unsigned,
807 vec#".widen_high_"#arg#"_u", !add(baseInst, 3)>;
808}
809
810defm "" : SIMDWiden<v8i16, "i16x8", v16i8, "i8x16", 202>;
811defm "" : SIMDWiden<v4i32, "i32x4", v8i16, "i16x8", 206>;
812
813// Narrowing operations
814multiclass SIMDNarrow<ValueType vec_t, string vec, ValueType arg_t, string arg,
815 bits<32> baseInst> {
816 defm NARROW_S_#vec_t :
817 SIMD_I<(outs V128:$dst), (ins V128:$low, V128:$high), (outs), (ins),
818 [(set (vec_t V128:$dst), (vec_t (int_wasm_narrow_signed
819 (arg_t V128:$low), (arg_t V128:$high))))],
820 vec#".narrow_"#arg#"_s\t$dst, $low, $high", vec#".narrow_"#arg#"_s",
821 baseInst>;
822 defm NARROW_U_#vec_t :
823 SIMD_I<(outs V128:$dst), (ins V128:$low, V128:$high), (outs), (ins),
824 [(set (vec_t V128:$dst), (vec_t (int_wasm_narrow_unsigned
825 (arg_t V128:$low), (arg_t V128:$high))))],
826 vec#".narrow_"#arg#"_u\t$dst, $low, $high", vec#".narrow_"#arg#"_u",
827 !add(baseInst, 1)>;
828}
829
830defm "" : SIMDNarrow<v16i8, "i8x16", v8i16, "i16x8", 198>;
831defm "" : SIMDNarrow<v8i16, "i16x8", v4i32, "i32x4", 200>;
832
Thomas Lively2ebacb12018-10-11 00:01:25 +0000833// Lower llvm.wasm.trunc.saturate.* to saturating instructions
834def : Pat<(v4i32 (int_wasm_trunc_saturate_signed (v4f32 V128:$src))),
835 (fp_to_sint_v4i32_v4f32 (v4f32 V128:$src))>;
836def : Pat<(v4i32 (int_wasm_trunc_saturate_unsigned (v4f32 V128:$src))),
837 (fp_to_uint_v4i32_v4f32 (v4f32 V128:$src))>;
838def : Pat<(v2i64 (int_wasm_trunc_saturate_signed (v2f64 V128:$src))),
839 (fp_to_sint_v2i64_v2f64 (v2f64 V128:$src))>;
840def : Pat<(v2i64 (int_wasm_trunc_saturate_unsigned (v2f64 V128:$src))),
841 (fp_to_uint_v2i64_v2f64 (v2f64 V128:$src))>;
842
Heejin Ahnd9a6de32018-10-09 22:23:39 +0000843// Bitcasts are nops
844// Matching bitcast t1 to t1 causes strange errors, so avoid repeating types
845foreach t1 = [v16i8, v8i16, v4i32, v2i64, v4f32, v2f64] in
846foreach t2 = !foldl(
847 []<ValueType>, [v16i8, v8i16, v4i32, v2i64, v4f32, v2f64],
848 acc, cur, !if(!eq(!cast<string>(t1), !cast<string>(cur)),
849 acc, !listconcat(acc, [cur])
850 )
851) in
852def : Pat<(t1 (bitconvert (t2 V128:$v))), (t1 V128:$v)>;
Thomas Livelyd0d93172019-08-31 00:12:29 +0000853
854//===----------------------------------------------------------------------===//
855// Quasi-Fused Multiply- Add and Subtract (QFMA/QFMS)
856//===----------------------------------------------------------------------===//
Thomas Livelya9b3d1f2019-09-25 00:15:59 +0000857
Thomas Livelyd0d93172019-08-31 00:12:29 +0000858multiclass SIMDQFM<ValueType vec_t, string vec, bits<32> baseInst> {
859 defm QFMA_#vec_t :
860 SIMD_I<(outs V128:$dst), (ins V128:$a, V128:$b, V128:$c),
861 (outs), (ins),
862 [(set (vec_t V128:$dst),
863 (int_wasm_qfma (vec_t V128:$a), (vec_t V128:$b), (vec_t V128:$c)))],
864 vec#".qfma\t$dst, $a, $b, $c", vec#".qfma", baseInst>;
865 defm QFMS_#vec_t :
866 SIMD_I<(outs V128:$dst), (ins V128:$a, V128:$b, V128:$c),
867 (outs), (ins),
868 [(set (vec_t V128:$dst),
869 (int_wasm_qfms (vec_t V128:$a), (vec_t V128:$b), (vec_t V128:$c)))],
870 vec#".qfms\t$dst, $a, $b, $c", vec#".qfms", !add(baseInst, 1)>;
871}
872
873defm "" : SIMDQFM<v4f32, "f32x4", 0x98>;
874defm "" : SIMDQFM<v2f64, "f64x2", 0xa3>;